import unittest
from pdc_config import TestConfig
from app import create_app, db_mgr


class BaseTestCase(unittest.TestCase):
    def setUp(self):
        # configure data base
        self.app = create_app(TestConfig)
        self.app_context = self.app.app_context()
        self.app_context.push()

    def tearDown(self):
        self.app_context.pop()

    def test_always_true(self):
        self.assertTrue(True)


class DbMgrTestCase(BaseTestCase):

    def test_agents(self):
        all_agents = db_mgr.agents()
        print(len(all_agents))
        self.assertEqual(548, len(all_agents))

    def test_charges_by_agent(self):
        all_charges = db_mgr.charges_by_agent(355)
        self.assertEqual(6, len(all_charges))