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
Exemple #2
0
    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))
Exemple #3
0
    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))
Exemple #4
0
    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))
Exemple #5
0
 def _get_agents(self):
     res = {}
     agents = agent_dao.all()
     for agent in agents:
         res.update(self._format_agent_data(agent))
     return res
Exemple #6
0
 def test_del_agent(self):
     agent_dao.del_agent(self._insert_agent().id)
     self.assertTrue(agent_dao.all() == [])
Exemple #7
0
 def test_all_empty(self):
     result = agent_dao.all()
     self.assertEqual([], result)
Exemple #8
0
 def test_del_agent(self):
     agent_dao.del_agent(self._insert_agent().id)
     self.assertTrue(agent_dao.all() == [])
Exemple #9
0
 def test_all_empty(self):
     result = agent_dao.all()
     self.assertEqual([], result)
Exemple #10
0
 def test_all_empty(self):
     result = agent_dao.all()
     assert_that(result, empty())