コード例 #1
0
ファイル: test_monthly_bill.py プロジェクト: dimagi/payments
    def setUp(self):
        for domain in Domain.get_all():
            domain.delete()

        start_date, _ = HQMonthlyBill.get_default_start_end()
        sms_date = start_date + datetime.timedelta(days=7)

        all_billables = SMSBillable.get_all()
        # all_billables contains duplicates; only delete each doc once
        for b_id in set(b._id for b in all_billables):
            SMSBillable.get_db().delete_doc(b_id)

        self.domain = Domain()
        self.domain.name = "domain_with_sms"
        self.domain.is_active = True
        self.domain.date_created = sms_date
        self.domain.save()

        # Incoming billables

        self.tropo_bill = TropoSMSBillable()
        self.tropo_bill.billable_date = sms_date
        self.tropo_bill.billable_amount = 2
        self.tropo_bill.conversion_rate = 1
        self.tropo_bill.dimagi_surcharge = 0.002
        self.tropo_bill.rate_id = "INCOMING_RATE_TROPO"
        self.tropo_bill.log_id = "INCOMING_LOG_TROPO"
        self.tropo_bill.domain = self.domain.name
        self.tropo_bill.direction = INCOMING
        self.tropo_bill.phone_number = "+15551234567"
        self.tropo_bill.tropo_id = "TROPO_ID"
        self.tropo_bill.save()

        # Outgoing billables

        self.mach_bill = MachSMSBillable()
        self.mach_bill.billable_date = sms_date
        self.mach_bill.contacted_mach_api = sms_date
        self.mach_bill.mach_delivered_date = sms_date
        self.mach_bill.billable_amount = 0.01
        self.mach_bill.conversion_rate = 1.2
        self.mach_bill.dimagi_surcharge = 0.002
        self.mach_bill.rate_id = "OUTGOING_MACH_RATE"
        self.mach_bill.log_id = "OUTGOING_MACH_LOG"
        self.mach_bill.domain = self.domain.name
        self.mach_bill.direction = OUTGOING
        self.mach_bill.phone_number = "+15551234567"
        self.mach_bill.mach_delivery_status = "delivered"
        self.mach_bill.mach_id = "MACH_MESSAGE_ID"
        self.mach_bill.save()

        self.unicel_bill = UnicelSMSBillable()
        self.unicel_bill.billable_date = sms_date
        self.unicel_bill.billable_amount = 2
        self.unicel_bill.conversion_rate = 1
        self.unicel_bill.dimagi_surcharge = 0.002
        self.unicel_bill.rate_id = "OUTGOING_UNICEL_RATE"
        self.unicel_bill.log_id = "OUTGOING_UNICEL_LOG"
        self.unicel_bill.domain = self.domain.name
        self.unicel_bill.direction = OUTGOING
        self.unicel_bill.phone_number = "+15551234567"
        self.unicel_bill.unicel_id = "UNICEL_ID"
        self.unicel_bill.save()
コード例 #2
0
ファイル: test_monthly_bill.py プロジェクト: dimagi/payments
class TestMonthlyBillFewUsers(TestCase):

    def setUp(self):
        for domain in Domain.get_all():
            domain.delete()

        start_date, _ = HQMonthlyBill.get_default_start_end()
        sms_date = start_date + datetime.timedelta(days=7)

        all_billables = SMSBillable.get_all()
        # all_billables contains duplicates; only delete each doc once
        for b_id in set(b._id for b in all_billables):
            SMSBillable.get_db().delete_doc(b_id)

        self.domain = Domain()
        self.domain.name = "domain_with_sms"
        self.domain.is_active = True
        self.domain.date_created = sms_date
        self.domain.save()

        # Incoming billables

        self.tropo_bill = TropoSMSBillable()
        self.tropo_bill.billable_date = sms_date
        self.tropo_bill.billable_amount = 2
        self.tropo_bill.conversion_rate = 1
        self.tropo_bill.dimagi_surcharge = 0.002
        self.tropo_bill.rate_id = "INCOMING_RATE_TROPO"
        self.tropo_bill.log_id = "INCOMING_LOG_TROPO"
        self.tropo_bill.domain = self.domain.name
        self.tropo_bill.direction = INCOMING
        self.tropo_bill.phone_number = "+15551234567"
        self.tropo_bill.tropo_id = "TROPO_ID"
        self.tropo_bill.save()

        # Outgoing billables

        self.mach_bill = MachSMSBillable()
        self.mach_bill.billable_date = sms_date
        self.mach_bill.contacted_mach_api = sms_date
        self.mach_bill.mach_delivered_date = sms_date
        self.mach_bill.billable_amount = 0.01
        self.mach_bill.conversion_rate = 1.2
        self.mach_bill.dimagi_surcharge = 0.002
        self.mach_bill.rate_id = "OUTGOING_MACH_RATE"
        self.mach_bill.log_id = "OUTGOING_MACH_LOG"
        self.mach_bill.domain = self.domain.name
        self.mach_bill.direction = OUTGOING
        self.mach_bill.phone_number = "+15551234567"
        self.mach_bill.mach_delivery_status = "delivered"
        self.mach_bill.mach_id = "MACH_MESSAGE_ID"
        self.mach_bill.save()

        self.unicel_bill = UnicelSMSBillable()
        self.unicel_bill.billable_date = sms_date
        self.unicel_bill.billable_amount = 2
        self.unicel_bill.conversion_rate = 1
        self.unicel_bill.dimagi_surcharge = 0.002
        self.unicel_bill.rate_id = "OUTGOING_UNICEL_RATE"
        self.unicel_bill.log_id = "OUTGOING_UNICEL_LOG"
        self.unicel_bill.domain = self.domain.name
        self.unicel_bill.direction = OUTGOING
        self.unicel_bill.phone_number = "+15551234567"
        self.unicel_bill.unicel_id = "UNICEL_ID"
        self.unicel_bill.save()


    def tearDown(self):
        self.mach_bill.delete()
        self.unicel_bill.delete()
        self.tropo_bill.delete()

        self.domain.delete()

    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.")