Files
@ b320b227aae9
Branch filter:
Location: symposion_app/registrasion/tests/controller_helpers.py - annotation
b320b227aae9
1.5 KiB
text/x-python
Merge branch 'guided_registration_3'
eab1deff7776 563355485435 eab1deff7776 eab1deff7776 eab1deff7776 38cdb8aa6330 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 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 38cdb8aa6330 | from registrasion.controllers.cart import CartController
from registrasion.controllers.invoice import InvoiceController
from registrasion import models as rego
from django.core.exceptions import ObjectDoesNotExist
from django.core.exceptions import ValidationError
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 '''
payment = rego.ManualPayment.objects.create(
invoice=self.invoice,
reference=reference,
amount=amount,
)
self.update_status()
|