Two db request wrappers

hitier
1 parent 9183e41e
Showing 2 changed files with 28 additions and 1 deletions Show diff stats
app/db_mgr.py
tests/backend_tests.py
@@ -0,0 +1,15 @@
+from app.models import db, Agent, Charge
+
+
+def agents():
+    all_agents = [a.firstname for a in Agent.query.all()]
+    return all_agents
+
+
+def charges_by_agent(agent_id):
+    # all_charges = db.session.query(Charge).all()
+    sql_txt = "select sum(charge_rate), p.name from charge inner join period p on charge.period_id = p.id\
+                where agent_id={} group by period_id order by p.name"\
+                .format(agent_id)
+    request = db.session.execute(sql_txt)
+    return request.fetchall()
 import unittest
 from pdc_config import TestConfig
-from app import create_app
+from app import create_app, db_mgr
  
  
 class BaseTestCase(unittest.TestCase):
@@ -15,3 +15,15 @@ class BaseTestCase(unittest.TestCase):
  
     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))