symposion_app Files · vendor/registrasion/registrasion/tests/test_helpers.py

Files @ 252697b842c0

Branch filter:

Location: symposion_app/vendor/registrasion/registrasion/tests/test_helpers.py

252697b842c0 947 B text/x-python Show Annotation Show as Raw Download as Raw

Joel Addison

Update to Django 2.2

Upgrade site and modules to Django 2.2. Remove and replace obsolete
functionality with current equivalents. Update requirements to latest
versions where possible. Remove unused dependencies.

import datetime

from registrasion.models import commerce

from registrasion.tests.controller_helpers import TestingCartController
from registrasion.tests.controller_helpers import TestingCreditNoteController
from registrasion.tests.controller_helpers import TestingInvoiceController


class TestHelperMixin(object):

    def _invoice_containing_prod_1(self, qty=1):
        cart = TestingCartController.for_user(self.USER_1)
        cart.add_to_cart(self.PROD_1, qty)

        return TestingInvoiceController.for_cart(self.reget(cart.cart))

    def _manual_invoice(self, value=1):
        items = [("Item", value)]
        due = datetime.timedelta(hours=1)
        inv = TestingInvoiceController.manual_invoice(self.USER_1, due, items)

        return TestingInvoiceController(inv)

    def _credit_note_for_invoice(self, invoice):
        note = commerce.CreditNote.objects.get(invoice=invoice)
        return TestingCreditNoteController(note)