graph_spec.rb in build-graph-0.3.5

- old
+ new

@@ -120,15 +120,17 @@
 		end
 		
 		def update!
 			group = Process::Group.new
 			
-			super do |walker, node|
+			walker = super do |walker, node|
 				Task.new(self, walker, node, group)
 			end
 			
 			group.wait
+			
+			return walker
 		end
 	end
 	
 	include Build::Files
 	
@@ -202,10 +204,11 @@
 				process program_path, Paths::NONE do
 					run("./" + program_path.relative_path, chdir: program_path.root)
 				end
 			end
 			
-			controller.update!
+			walker = controller.update!
+			expect(walker).to be_kind_of Build::Graph::Walker
 			
 			expect(program_path).to be_exist
 			expect(code_glob.first.mtime).to be <= program_path.mtime
 		end
 	end