def testSMSBilling(self):
generate_monthly_bills()
last_bill = HQMonthlyBill.get_bills(self.domain.name).first()
if last_bill:
self.assertEqual(self.tropo_bill.total_billed, last_bill.incoming_sms_billed)
self.assertEqual(self.unicel_bill.total_billed + self.mach_bill.total_billed, last_bill.outgoing_sms_billed)
else:
raise Exception("Monthly Bill not successfully generated.")
def generate_bills(request):
status = "Last Month"
domain = request.GET.get('domain')
domain_status = domain or "all domains"
try:
start = dateutil.parser.parse(request.GET.get('start'))
end = dateutil.parser.parse(request.GET.get('end'))
end = end.replace(minute=59, hour=23, second=59, microsecond=999999)
date_range = [start, end]
status = "%s through %s" % (start, end)
except Exception as e:
date_range = None
generate_monthly_bills(billing_range=date_range, domain_name=domain)
return HttpResponse("Bills generated for %s on %s." % (status, domain_status))
