fixed zeromq tests

josegonzalez · josegonzalez · commit e1fcd0033c99 · 2013-03-29T12:33:12.000-04:00
diff --git a/beaver/tests/test_zmq_transport.py b/beaver/tests/test_zmq_transport.py
@@ -20,14 +20,14 @@ def setUp(self):
         self.beaver_config = BeaverConfig(mock.Mock(config=None))
 
     def test_pub(self):
-        os.environ["ZEROMQ_ADDRESS"] = "tcp://localhost:2120"
+        self.beaver_config.set('zeromq_address', 'tcp://localhost:2120')
         transport = ZmqTransport(self.beaver_config, self.file_config)
         transport.interrupt()
         #assert not transport.zeromq_bind
 
     def test_bind(self):
-        self.beaver_config.mode = "bind"
-        os.environ["ZEROMQ_ADDRESS"] = "tcp://*:2120"
+        self.beaver_config.set('zeromq_bind', 'bind')
+        self.beaver_config.set('zeromq_address', 'tcp://localhost:2120')
         transport = ZmqTransport(self.beaver_config, self.file_config)
         #assert transport.zeromq_bind
 
