def get(self):
qParser = reqparse.RequestParser()
qParser.add_argument("fromDate", type=dateInput, default=None)
qParser.add_argument("toDate", type=dateInput, default=None)
qParser.add_argument("doctor_id", type=int, default=None)
qParser.add_argument("service_id", type=int, default=None)
qParser.add_argument("state", type=str, default=None)
args = qParser.parse_args()
fromDate = args.get("fromDate")
toDate = args.get("toDate")
doctor_id = args.get("doctor_id")
service_id = args.get("service_id")
state = args.get("state")
report = GroupedSessionsSummary()
report.filterDateRange(fromDate, toDate)
report.filterByDoctor(doctor_id)
report.filterByService(service_id)
report.filterByState(state)
return report.result()
def test_withDefaultDate_report(self):
report = GroupedSessionsSummary()
report.filterDateRange()
result = report.result()
print(result)
expected = {'count': 5}
self.assertEquals(expected, result)
def test_done_sessions_count_report(self):
report = GroupedSessionsSummary()
# report.filterDateRange(datetime.date.today(), datetime.datetime.now() + datetime.timedelta(days=1) )
report.filterDateRange()
result = report.result()
totalSessions = 0
for item in result['details']:
self.assertTrue(item["service_id"] is not None)
self.assertTrue(item["stateCount"] is not None)
# self.assertEquals(item["stateCount"])
totalSessions += item["stateCount"]
self.assertTrue(item["doctor_id"] is not None)
self.assertTrue(item["serviceName"] is not None)
self.assertTrue(item["doctorName"] is not None)
self.assertTrue(item["state"] is not None)
self.assertEquals(item["state"], 'done')
self.assertEquals(totalSessions, result['totalNumberOfSessions'])
def test_with_doctor_id_report(self):
dr1_test = Doctor.query.filter(Doctor.name == 'dr1_test').one()
report = GroupedSessionsSummary()
report.filterDateRange(datetime.date.today(), datetime.datetime.now() + datetime.timedelta(days=1))
report.filterByDoctor(dr1_test.id)
result = report.result()
# print(result)
expected = {'count': 7}
self.assertEquals(expected, result)
