def test_recv_data_with_empty_data(self):
"""Test case to test _recv_data with empty data."""
fake_socket = base_test.SocketClass(None, None, None, '')
with mock.patch.object(socket, 'socket', return_value=fake_socket):
with mock.patch.object(ovsdb_writer.LOG, 'warning'):
ovsdb_conf = base_test.FakeConf()
l2gw_obj = ovsdb_writer.OVSDBWriter(cfg.CONF.ovsdb, ovsdb_conf)
result = l2gw_obj._recv_data(mock.ANY)
self.assertIsNone(result)
def _open_connection(self, ovsdb_identifier):
ovsdb_fd = None
gateway = self.gateways.get(ovsdb_identifier)
try:
ovsdb_fd = ovsdb_writer.OVSDBWriter(self.conf.ovsdb, gateway)
yield ovsdb_fd
finally:
if ovsdb_fd:
ovsdb_fd.disconnect()
def test_recv_data_with_empty_data(self):
"""Test case to test _recv_data with empty data."""
fake_socket = base_test.SocketClass(None, None, None, '')
with contextlib.nested(
mock.patch.object(socket, 'socket', return_value=fake_socket),
mock.patch.object(ovsdb_writer.LOG,
'warning')) as (fake_sock, fake_warn):
ovsdb_conf = base_test.FakeConf()
l2gw_obj = ovsdb_writer.OVSDBWriter(cfg.CONF.ovsdb, ovsdb_conf)
result = l2gw_obj._recv_data()
self.assertEqual(None, result)
def test_recv_data(self):
"""Test case to test _recv_data with a valid data."""
fake_data = {"fake_key": "fake_value"}
fake_socket = base_test.SocketClass(
None, None, None,
base_test.FakeDecodeClass(jsonutils.dumps(fake_data)))
with mock.patch.object(socket, 'socket', return_value=fake_socket):
ovsdb_conf = base_test.FakeConf()
l2gw_obj = ovsdb_writer.OVSDBWriter(cfg.CONF.ovsdb, ovsdb_conf)
result = l2gw_obj._recv_data()
self.assertEqual(jsonutils.dumps(fake_data), result)
def test_recv_data_with_socket_error(self):
"""Test case to test _recv_data with socket error."""
fake_socket = base_test.SocketClass(None, None, socket.error)
with mock.patch.object(socket, 'socket', return_value=fake_socket):
with mock.patch.object(ovsdb_writer.LOG, 'warning') as fake_warn:
ovsdb_conf = base_test.FakeConf()
l2gw_obj = ovsdb_writer.OVSDBWriter(cfg.CONF.ovsdb, ovsdb_conf)
result = l2gw_obj._recv_data(mock.ANY)
self.assertIsNone(result)
fake_warn.assert_called_with(
"Did not receive any reply from the OVSDB server")
def setUp(self):
super(TestOVSDBWriter, self).setUp()
self.conf = mock.patch.object(conf, 'L2GatewayConfig').start()
config.register_ovsdb_opts_helper(cfg.CONF)
cfg.CONF.set_override('max_connection_retries', 0, 'ovsdb')
self.sock = mock.patch('socket.socket').start()
self.ssl_sock = mock.patch.object(ssl, 'wrap_socket').start()
self.op_id = 'abcd'
self.l2gw_ovsdb = ovsdb_writer.OVSDBWriter(mock.Mock(), self.conf)
self.fake_message = {'id': self.op_id, 'fake_key': 'fake_value'}
self.l2gw_ovsdb.responses = [self.fake_message]
def test_recv_data_with_socket_error(self):
"""Test case to test _recv_data with socket error."""
fake_socket = base_test.SocketClass(None, None, socket.error)
with contextlib.nested(
mock.patch.object(socket, 'socket', return_value=fake_socket),
mock.patch.object(ovsdb_writer.LOG,
'warning')) as (fake_sock, fake_warn):
ovsdb_conf = base_test.FakeConf()
l2gw_obj = ovsdb_writer.OVSDBWriter(cfg.CONF.ovsdb, ovsdb_conf)
result = l2gw_obj._recv_data()
self.assertEqual(None, result)
fake_warn.assert_called_with(
_LW("Did not receive any reply from the OVSDB "
"server"))
