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