def setUp(self):
super(TestUpdateHealthDb, self).setUp()
conf = self.useFixture(oslo_fixture.Config(cfg.CONF))
conf.config(group="health_manager",
event_streamer_driver='queue_event_streamer')
session_patch = mock.patch('octavia.db.api.get_session')
self.addCleanup(session_patch.stop)
self.mock_session = session_patch.start()
self.session_mock = mock.MagicMock()
self.mock_session.return_value = self.session_mock
self.hm = update_db.UpdateHealthDb()
self.event_client = mock.MagicMock()
self.hm.event_streamer.client = self.event_client
self.amphora_repo = mock.MagicMock()
self.amphora_health_repo = mock.MagicMock()
self.listener_repo = mock.MagicMock()
self.loadbalancer_repo = mock.MagicMock()
self.member_repo = mock.MagicMock()
self.pool_repo = mock.MagicMock()
self.hm.amphora_repo = self.amphora_repo
fake_lb = mock.MagicMock()
self.hm.amphora_repo.get_lb_for_amphora.return_value = fake_lb
self.hm.amphora_health_repo = self.amphora_health_repo
self.hm.listener_repo = self.listener_repo
self.hm.listener_repo.count.return_value = 1
self.hm.loadbalancer_repo = self.loadbalancer_repo
self.hm.member_repo = self.member_repo
self.hm.pool_repo = self.pool_repo
def setUp(self):
super(TestUpdateHealthDb, self).setUp()
session_patch = mock.patch('octavia.db.api.get_session')
self.addCleanup(session_patch.stop)
self.mock_session = session_patch.start()
self.session_mock = mock.MagicMock()
self.mock_session.return_value = self.session_mock
self.hm = update_db.UpdateHealthDb()
self.amphora_repo = mock.MagicMock()
self.amphora_health_repo = mock.MagicMock()
self.listener_repo = mock.MagicMock()
self.loadbalancer_repo = mock.MagicMock()
self.member_repo = mock.MagicMock()
self.pool_repo = mock.MagicMock()
self.hm.amphora_repo = self.amphora_repo
self.hm.amphora_health_repo = self.amphora_health_repo
self.hm.listener_repo = self.listener_repo
self.hm.listener_repo.count.return_value = 1
self.hm.loadbalancer_repo = self.loadbalancer_repo
self.hm.member_repo = self.member_repo
self.hm.pool_repo = self.pool_repo