task_dispatcher_spec.rb in kamerling-0.0.3

- old
+ new
@@ -1,23 +1,46 @@
 require_relative '../spec_helper'
+require_relative '../../lib/kamerling/addr'
+require_relative '../../lib/kamerling/client'
+require_relative '../../lib/kamerling/dispatch'
+require_relative '../../lib/kamerling/message'
+require_relative '../../lib/kamerling/net_dispatcher'
+require_relative '../../lib/kamerling/project'
+require_relative '../../lib/kamerling/repos'
+require_relative '../../lib/kamerling/task'
+require_relative '../../lib/kamerling/task_dispatcher'
+require_relative '../../lib/kamerling/uuid'
 
-module Kamerling describe TaskDispatcher do
-  describe '#dispatch' do
-    it 'dispatches tasks to free clients and marks them as busy' do
-      addr    = fake :addr
-      client  = fake :client, addr: addr, uuid: UUID['16B client  UUID']
-      project = fake :project, uuid: UUID['16B project UUID']
-      task    = fake :task, data: 'data', uuid: UUID['16B task    UUID']
-      repos   = fake :repos, as: :class, projects: [project]
+module Kamerling
+  describe TaskDispatcher do
+    let(:addr)           { Addr.new                                      }
+    let(:client)         { Client.new(addr: addr)                        }
+    let(:net_dispatcher) { fake(:net_dispatcher, as: :class)             }
+    let(:project)        { Project.new                                   }
+    let(:repos)          { fake(:repos, as: :class, projects: [project]) }
+    let(:task)           { Task.new(data: 'data')                        }
+
+    before do
       stub(repos).next_task_for(project) { task }
       stub(repos).free_clients_for(project) { [client] }
-      net_dispatcher = fake :net_dispatcher, as: :class
+      td = TaskDispatcher.new(net_dispatcher: net_dispatcher, repos: repos)
+      td.dispatch_all
+    end
 
-      TaskDispatcher.new(net_dispatcher: net_dispatcher, repos: repos).dispatch
+    describe '#dispatch_all' do
+      it 'dispatches tasks to free clients' do
+        message = Message.build(client: client, payload: 'data',
+                                project: project, task: task, type: :DATA)
+        net_dispatcher.must_have_received :dispatch, [addr, message]
+      end
 
-      header = "DATA#{"\0" * 12}16B client  UUID16B project UUID16B task    "
-      net_dispatcher.must_have_received :dispatch, [addr, header + 'UUIDdata']
-      client.must_have_received :busy=, [true]
-      repos.must_have_received :<<, [client]
+      it 'marks clients as busy and persists the change' do
+        assert client.busy
+        repos.must_have_received :<<, [client]
+      end
+
+      it 'creates and stores a Dispatch object along the way' do
+        repos.must_have_received :<<, [any(Dispatch)]
+      end
     end
   end
-end end
+end