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_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)
