def _get_agents(self): with session_scope(): res = {} agents = agent_dao.all() for agent in agents: res.update(self._format_agent_data(agent)) return res
def test_all(self): agent1 = self._insert_agent(self.agent1_number) agent2 = self._insert_agent(self.agent2_number) expected = [agent1, agent2] result = agent_dao.all() assert_that(result, equal_to(expected))
def test_all_multi_tenant(self): tenant = self.add_tenant() agent1 = self.add_agent(number=self.agent1_number, tenant_uuid=self.default_tenant.uuid) agent2 = self.add_agent(number=self.agent2_number, tenant_uuid=tenant.uuid) expected_tenant1 = [agent1] result = agent_dao.all(tenant_uuids=[self.default_tenant.uuid]) assert_that(result, equal_to(expected_tenant1)) expected_tenant2 = [agent2] result = agent_dao.all(tenant_uuids=[tenant.uuid]) assert_that(result, equal_to(expected_tenant2)) expected_all = [agent1, agent2] result = agent_dao.all( tenant_uuids=[self.default_tenant.uuid, tenant.uuid]) assert_that(result, equal_to(expected_all))
def _get_agents(self): res = {} agents = agent_dao.all() for agent in agents: res.update(self._format_agent_data(agent)) return res
def test_del_agent(self): agent_dao.del_agent(self._insert_agent().id) self.assertTrue(agent_dao.all() == [])
def test_all_empty(self): result = agent_dao.all() self.assertEqual([], result)
def test_all_empty(self): result = agent_dao.all() assert_that(result, empty())