def setUp(self):
with patch('voltha.adapters.adtran_olt.net.adtran_zmq.structlog.get_logger'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.zmq_factory'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.ZmqPairConnection', autospec=True):
# Create AdtranZmqClient instance for test
self.adtran_zmq_client = AdtranZmqClient('1.2.3.4', lambda x: x,
5657)
class TestAdtranZmqClient_send(TestCase):
"""
This class contains all methods to unit test AdtranZmqClient.send()
"""
def setUp(self):
with patch('voltha.adapters.adtran_olt.net.adtran_zmq.structlog.get_logger'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.zmq_factory'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.ZmqPairConnection', autospec=True):
# Create AdtranZmqClient instance for test
self.adtran_zmq_client = AdtranZmqClient('1.2.3.4', lambda x: x,
5657)
# Test send() with normal data
def test_send_normal_data(self):
self.adtran_zmq_client.send("data")
self.adtran_zmq_client._socket.send.assert_called_once_with('data')
# Test send() with bad data (force exception)
def test_send_bad_data(self):
# Cause exception to occur
self.adtran_zmq_client._socket.send.side_effect = ValueError
# _socket.send in AdtranZmqClient.send() already mocked out via mock_zmq_pair_connection
self.adtran_zmq_client.send("cause exception")
self.adtran_zmq_client._socket.send.assert_called_once_with(
'cause exception')
self.adtran_zmq_client.log.exception.assert_called_once()
class TestAdtranZmqClient_setup_curve_security(TestCase):
"""
This class contains all methods to unit test AdtranZmqClient.setup_curve_security()
"""
def setUp(self):
with patch('voltha.adapters.adtran_olt.net.adtran_zmq.structlog.get_logger'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.zmq_factory'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.ZmqPairConnection', autospec=True):
# Create AdtranZmqClient instance for test
self.adtran_zmq_client = AdtranZmqClient('1.2.3.4', lambda x: x,
5657)
# Test setup_curve_security() -- not much to test, just creating line coverage
def test_setup_curve_security(self):
with self.assertRaises(NotImplementedError):
self.adtran_zmq_client.setup_curve_security()
class TestAdtranZmqClient_setup_plain_security(TestCase):
"""
This class contains all methods to unit test AdtranZmqClient.setup_plain_security()
"""
def setUp(self):
with patch('voltha.adapters.adtran_olt.net.adtran_zmq.structlog.get_logger'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.zmq_factory'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.ZmqPairConnection', autospec=True):
# Create AdtranZmqClient instance for test
self.adtran_zmq_client = AdtranZmqClient('1.2.3.4', lambda x: x,
5657)
# Test setup_plain_security() including verifying calls to addCallbacks()
# Omitting coverage for the methods inside of setup_plain_security() for now
def test_setup_plain_security(self, mk_twisted_zmq_authenticator):
deferred = self.adtran_zmq_client.setup_plain_security('user', 'pswd')
self.adtran_zmq_client.auth.start.assert_called_once()
# Test that addCallbacks was called twice
self.assertEqual(deferred.addCallbacks.call_count, 2)
# Test that both params used in each call is a function
# First, the call to d.addCallbacks(configure_plain, config_failure)
self.assertTrue(callable(
deferred.addCallbacks.call_args_list[0][0][0]))
self.assertTrue(callable(
deferred.addCallbacks.call_args_list[0][0][1]))
# Second, the call to d.addCallbacks(add_endoints, endpoint_failure)
self.assertTrue(callable(
deferred.addCallbacks.call_args_list[1][0][0]))
self.assertTrue(callable(
deferred.addCallbacks.call_args_list[1][0][1]))
class TestAdtranZmqClient_shutdown(TestCase):
"""
This class contains all methods to unit test AdtranZmqClient.shutdown()
"""
def setUp(self):
with patch('voltha.adapters.adtran_olt.net.adtran_zmq.structlog.get_logger'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.zmq_factory'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.ZmqPairConnection', autospec=True):
# Create AdtranZmqClient instance for test
self.adtran_zmq_client = AdtranZmqClient('1.2.3.4', lambda x: x,
5657)
# Test shutdown() and verifying that the socket call to shutdown() is made
def test_shutdown(self):
# _socket.shutdown in AdtranZmqClient.shutdown() already mocked out via mock_zmq_pair_connection
self.adtran_zmq_client.shutdown()
self.adtran_zmq_client._socket.shutdown.assert_called_once()
def setUp(self):
with patch('voltha.adapters.adtran_olt.net.adtran_zmq.structlog.get_logger'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.zmq_factory'), \
patch('voltha.adapters.adtran_olt.net.adtran_zmq.ZmqPairConnection', autospec=True) as mk_zmq_pair_conn:
# Save mock instance ID for comparison later
self.zmq_pair_conn_mock_id = mk_zmq_pair_conn.return_value
# Create AdtranZmqClient instance for test
self.adtran_zmq_client = AdtranZmqClient('1.2.3.4', lambda x: x,
5657)
def test_rx_nop(self):
# rx_nop() is a static method
AdtranZmqClient.rx_nop(None)