def test_listenermgr_delete(happy_path_driver):
mock_driver, mock_ctx = happy_path_driver
lstnr_mgr = dv2.ListenerManager(mock_driver)
fake_lstnr = FakeListener()
lstnr_mgr.delete(mock_ctx, fake_lstnr)
assert mock_driver.agent_rpc.delete_listener.call_args == \
mock.call(mock_ctx, fake_lstnr.to_dict(), {}, 'test_agent')
def test_listener_update_exception(mock_log, happy_path_driver):
mock_driver, mock_ctx = happy_path_driver
mock_driver.agent_rpc.update_listener.side_effect = Exception('test')
lstnr_mgr = dv2.ListenerManager(mock_driver)
fake_old_lstnr = FakeListener(id='old_listener')
fake_new_lstnr = FakeListener(id='new_listener')
with pytest.raises(Exception) as ex:
lstnr_mgr.update(mock_ctx, fake_old_lstnr, fake_new_lstnr)
assert 'test' == ex.value.message
assert mock_log.error.call_args == mock.call(
'Exception: listener update: test')
def test_listener_update(happy_path_driver):
mock_driver, mock_ctx = happy_path_driver
lstnr_mgr = dv2.ListenerManager(mock_driver)
fake_old_lstnr = FakeListener(id='old_listener')
fake_new_lstnr = FakeListener(id='new_listener')
lstnr_mgr.update(mock_ctx, fake_old_lstnr, fake_new_lstnr)
assert mock_driver.agent_rpc.update_listener.call_args == \
mock.call(
mock_ctx,
fake_old_lstnr.to_dict(),
fake_new_lstnr.to_dict(),
{},
'test_agent')