spec/jcukeforker/runner_spec.rb in jcukeforker-0.2.4 vs spec/jcukeforker/runner_spec.rb in jcukeforker-0.2.5

@@ -17,17 +17,14 @@
 
         mock_task_manager = double(TaskManager, :update => nil)
         mock_status_server = double(StatusServer)
         mock_tasks = Array.new(2) { |n| double("Worker-#{n}") }
 
-        TaskManager.should_receive(:new).and_return mock_task_manager
+        TaskManager.should_receive(:new).with(features, {format: format, out: out, extra_args: []}).and_return mock_task_manager
         StatusServer.should_receive(:new).with('6333').and_return mock_status_server
 
         mock_status_server.should_receive(:add_observer).with listeners.first
         mock_status_server.should_receive(:add_observer).with mock_task_manager
-
-        mock_task_manager.should_receive(:add).with(feature: features[0], format: format, out: out, extra_args: [])
-        mock_task_manager.should_receive(:add).with(feature: features[1], format: format, out: out, extra_args: [])
 
         Runner.create(features,
           :max    => max,
           :notify => listeners,
           :format => format,