Files
@ c4c8a7ab8296
Branch filter:
Location: symposion_app/registrasion/tests/controller_helpers.py - annotation
c4c8a7ab8296
1.7 KiB
text/x-python
Tidies up the admin interface for flags
eab1deff7776 6b10a0a7e427 563355485435 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 760996588372 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 eab1deff7776 2f77f5bb23ee 2f77f5bb23ee 2f77f5bb23ee 2f77f5bb23ee 563355485435 563355485435 563355485435 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 2fbe78909086 2fbe78909086 38cdb8aa6330 38cdb8aa6330 ae8f39381f4a 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 6b10a0a7e427 | from registrasion.controllers.cart import CartController
from registrasion.controllers.credit_note import CreditNoteController
from registrasion.controllers.invoice import InvoiceController
from registrasion import models as rego
from django.core.exceptions import ObjectDoesNotExist
class TestingCartController(CartController):
def set_quantity(self, product, quantity, batched=False):
''' Sets the _quantity_ of the given _product_ in the cart to the given
_quantity_. '''
self.set_quantities(((product, quantity),))
def add_to_cart(self, product, quantity):
''' Adds _quantity_ of the given _product_ to the cart. Raises
ValidationError if constraints are violated.'''
try:
product_item = rego.ProductItem.objects.get(
cart=self.cart,
product=product)
old_quantity = product_item.quantity
except ObjectDoesNotExist:
old_quantity = 0
self.set_quantity(product, old_quantity + quantity)
def next_cart(self):
self.cart.active = False
self.cart.save()
class TestingInvoiceController(InvoiceController):
def pay(self, reference, amount):
''' Testing method for simulating an invoice paymenht by the given
amount. '''
self.validate_allowed_to_pay()
''' Adds a payment '''
rego.ManualPayment.objects.create(
invoice=self.invoice,
reference=reference,
amount=amount,
)
self.update_status()
class TestingCreditNoteController(CreditNoteController):
def refund(self):
rego.CreditNoteRefund.objects.create(
parent=self.credit_note,
reference="Whoops."
)
|