udp_spec.rb in kamerling-0.0.3

- old
+ new
@@ -1,44 +1,50 @@
+require 'socket'
 require_relative '../../spec_helper'
+require_relative '../../../lib/kamerling/addr'
+require_relative '../../../lib/kamerling/handler'
+require_relative '../../../lib/kamerling/message'
+require_relative '../../../lib/kamerling/server/udp'
 
-module Kamerling describe Server::UDP do
-  let(:addr) { Addr['localhost', 1979, :UDP] }
+module Kamerling
+  describe Server::UDP do
+    let(:addr) { Addr['localhost', 1979, :UDP] }
 
-  describe '#start' do
-    it 'listens on an UDP port and passes received inputs to the handler' do
-      server = Server::UDP.new addr: addr, handler: handler = fake(:handler)
-      server.start
-      foo = UDPSocket.new
-      bar = UDPSocket.new
-      foo.send 'foo', 0, *server.addr
-      bar.send 'bar', 0, *server.addr
-      foo_addr = Addr['127.0.0.1', foo.addr[1], :UDP]
-      bar_addr = Addr['127.0.0.1', bar.addr[1], :UDP]
-      run_all_threads
-      server.stop
-      handler.must_have_received :handle, ['foo', foo_addr]
-      handler.must_have_received :handle, ['bar', bar_addr]
-    end
+    describe '#start' do
+      it 'listens on an UDP port and passes received inputs to the handler' do
+        server = Server::UDP.new(addr: addr, handler: handler = fake(:handler))
+        server.start
+        foo = UDPSocket.new
+        bar = UDPSocket.new
+        3.times do
+          foo.send 'DATA', 0, *server.addr
+          bar.send 'PING', 0, *server.addr
+        end
+        run_all_threads
+        server.stop
+        handler.must_have_received :handle, [Message.parse('DATA'), any(Addr)]
+        handler.must_have_received :handle, [Message.parse('PING'), any(Addr)]
+      end
 
-    it 'doesn’t blow up on unknown inputs' do
-      server = Server::UDP.new addr: addr, handler: handler = fake(:handler)
-      server.start
-      stub(handler).handle('foo', any(Addr)) { fail Handler::UnknownInput }
-      UDPSocket.new.send 'foo', 0, *server.addr
-      run_all_threads
-      server.stop
+      it 'doesn’t blow up on unknown inputs' do
+        server = Server::UDP.new(addr: addr)
+        server.start
+        UDPSocket.new.send 'foo', 0, *server.addr
+        run_all_threads
+        server.stop
+      end
     end
-  end
 
-  describe '#stop' do
-    it 'closes the socket (and thus allows rebinding to it)' do
-      Server::UDP.new(addr: addr).start.stop
-      UDPSocket.new.tap { |socket| socket.bind(*addr) }.close
+    describe '#stop' do
+      it 'closes the socket (and thus allows rebinding to it)' do
+        Server::UDP.new(addr: addr).start.stop
+        UDPSocket.new.tap { |socket| socket.bind(*addr) }.close
+      end
     end
-  end
 
-  describe '#addr' do
-    it 'returns the server’s host + port as an UDP addr' do
-      Server::UDP.new(addr: addr).addr.must_equal addr
+    describe '#addr' do
+      it 'returns the server’s host + port as an UDP addr' do
+        Server::UDP.new(addr: addr).addr.must_equal addr
+      end
     end
   end
-end end
+end