spec/onstomp/connections_spec.rb in onstomp-1.0.0pre1 vs spec/onstomp/connections_spec.rb in onstomp-1.0.0

@@ -61,9 +61,13 @@
         ::TCPSocket.stub(:new => nil)
         ::OpenSSL::SSL::SSLSocket.stub(:new => nil)
         Connections::PROTOCOL_VERSIONS['1.0'] = stomp_1_0_class
         Connections::PROTOCOL_VERSIONS['1.1'] = stomp_1_1_class
       end
+      after(:each) do
+        Connections::PROTOCOL_VERSIONS['1.0'] = Connections::Stomp_1_0
+        Connections::PROTOCOL_VERSIONS['1.1'] = Connections::Stomp_1_1
+      end
       
       describe "TCP connections" do
         before(:each) do
           ::TCPSocket.should_receive(:new).with('host.domain.tld', 10101).and_return(tcp_socket)
           stomp_1_0_class.should_receive(:new).with(tcp_socket, client).and_return(stomp_1_0)