spec/unit/connection_adapters_spec.rb in websocket-rails-0.4.9 vs spec/unit/connection_adapters_spec.rb in websocket-rails-0.5.0

@@ -41,10 +41,22 @@
         end
 
         it "should create a new DataStore::Connection instance" do
           subject.data_store.should be_a DataStore::Connection
         end
+
+        before do
+          WebsocketRails.config.stub(:user_identifier).and_return(:name)
+          WebsocketRails::DelegationController.any_instance
+            .stub_chain(:current_user, :name)
+            .and_return('Frank')
+          subject
+        end
+
+        it "adds itself to the UserManager Hash" do
+          WebsocketRails.users['Frank'].should == subject
+        end
       end
 
       describe "#on_open" do
         it "should dispatch an on_open event" do
           on_open_event = double('event').as_null_object
@@ -67,10 +79,15 @@
           on_close_event = double('event')
           Event.should_receive(:new_on_close).and_return(on_close_event)
           dispatcher.should_receive(:dispatch).with(on_close_event)
           subject.on_close("data")
         end
+
+        it "removes itself from the global UserMnaager" do
+          subject.on_close
+          WebsocketRails.users[subject.id].nil?.should == true
+        end
       end
 
       describe "#on_error" do
         it "should dispatch an on_error event" do
           subject.stub(:on_close)
@@ -85,12 +102,39 @@
           subject.should_receive(:on_close).with(data)
           subject.on_error("test_data")
         end
       end
 
+      describe "#send_message" do
+        before do
+          Event.any_instance.stub(:trigger)
+        end
+        after do
+          subject.send_message :message, "some_data"
+        end
+
+        it "creates and triggers a new event" do
+          Event.any_instance.should_receive(:trigger)
+        end
+
+        it "sets it's user identifier on the event" do
+          subject.stub(:user_identifier).and_return(:some_name_or_id)
+          Event.should_receive(:new) do |name, options|
+            options[:user_id].should == :some_name_or_id
+          end.and_call_original
+        end
+
+        it "sets the connection property of the event correctly" do
+          subject.stub(:user_identifier).and_return(:some_name_or_id)
+          Event.should_receive(:new) do |name, options|
+            options[:connection].should == subject
+          end.and_call_original
+        end
+      end
+
       describe "#send" do
         it "should raise a NotImplementedError exception" do
-          expect { subject.send :message }.to raise_exception( NotImplementedError )
+          expect { subject.send :message }.to raise_exception(NotImplementedError)
         end
       end
 
       describe "#enqueue" do
         it "should add the event to the queue" do