def test_single_usage(self):
user = AnonymousUser()
creator = OrderCreator()
product = factories.ProductFactory(stockrecords__num_in_stock=1000)
# Make the order creator a bit more slow too reliable trigger
# concurrency issues
org_create_order_model = OrderCreator.create_order_model
def new_create_order_model(*args, **kwargs):
time.sleep(0.5)
return org_create_order_model(creator, *args, **kwargs)
creator.create_order_model = new_create_order_model
# Start 5 threads to place an order concurrently
def worker():
order_number = threading.current_thread().name
basket = factories.BasketFactory()
basket.add_product(product)
place_order(creator,
basket=basket,
order_number=order_number,
user=user)
exceptions = run_concurrently(worker, num_threads=5)
assert all(isinstance(x, ValueError) for x in exceptions), exceptions
assert len(exceptions) == 0
assert Order.objects.count() == 5
stockrecord = product.stockrecords.first()
assert stockrecord.num_allocated == 5
def test_single_usage(self):
user = AnonymousUser()
creator = OrderCreator()
product = factories.ProductFactory(stockrecords__num_in_stock=1000)
# Make the order creator a bit more slow too reliable trigger
# concurrency issues
org_create_order_model = OrderCreator.create_order_model
def new_create_order_model(*args, **kwargs):
time.sleep(0.5)
return org_create_order_model(creator, *args, **kwargs)
creator.create_order_model = new_create_order_model
# Start 5 threads to place an order concurrently
def worker():
order_number = threading.current_thread().name
basket = factories.BasketFactory()
basket.add_product(product)
place_order(
creator, basket=basket, order_number=order_number, user=user)
exceptions = run_concurrently(worker, num_threads=5)
assert all(isinstance(x, ValueError) for x in exceptions), exceptions
assert len(exceptions) == 0
assert Order.objects.count() == 5
stockrecord = product.stockrecords.first()
assert stockrecord.num_allocated == 5
def test_voucher_single_usage(self):
user = AnonymousUser()
creator = OrderCreator()
product = factories.ProductFactory()
voucher = factories.VoucherFactory(usage=Voucher.SINGLE_USE)
voucher.offers.add(factories.create_offer(offer_type='Voucher'))
# Make the order creator a bit more slow too reliable trigger
# concurrency issues
org_create_order_model = OrderCreator.create_order_model
def new_create_order_model(*args, **kwargs):
time.sleep(0.5)
return org_create_order_model(creator, *args, **kwargs)
creator.create_order_model = new_create_order_model
org_record_voucher_usage = OrderCreator.record_voucher_usage
def record_voucher_usage(*args, **kwargs):
time.sleep(0.5)
return org_record_voucher_usage(creator, *args, **kwargs)
creator.record_voucher_usage = record_voucher_usage
# Start 5 threads to place an order concurrently
def worker():
order_number = threading.current_thread().name
basket = factories.BasketFactory()
basket.add_product(product)
basket.vouchers.add(voucher)
surcharges = SurchargeApplicator().get_applicable_surcharges(
basket)
place_order(creator,
surcharges=surcharges,
basket=basket,
order_number=order_number,
user=user)
exceptions = run_concurrently(worker, num_threads=5)
voucher.refresh_from_db()
assert all(isinstance(x, ValueError) for x in exceptions), exceptions
assert len(exceptions) == 4
assert voucher.applications.count() == 1
assert Order.objects.count() == 1
def test_concurrent_invoice_creation(self):
num_threads = 5
creator = InvoiceCreator()
org_create_invoice = InvoiceCreator.create_invoice
def new_create_invoice(order, **kwargs):
time.sleep(0.5)
return org_create_invoice(creator, order, **kwargs)
def worker():
order_number = threading.current_thread().name
order = create_order(number=order_number, user=self.user)
new_create_invoice(order)
exceptions = run_concurrently(worker, num_threads=num_threads)
assert len(exceptions) == 0
assert Invoice.objects.count() == 5
def test_voucher_single_usage(self):
user = AnonymousUser()
creator = OrderCreator()
product = factories.ProductFactory()
voucher = factories.VoucherFactory(usage=Voucher.SINGLE_USE)
voucher.offers.add(factories.create_offer(offer_type='Voucher'))
# Make the order creator a bit more slow too reliable trigger
# concurrency issues
org_create_order_model = OrderCreator.create_order_model
def new_create_order_model(*args, **kwargs):
time.sleep(0.5)
return org_create_order_model(creator, *args, **kwargs)
creator.create_order_model = new_create_order_model
org_record_voucher_usage = OrderCreator.record_voucher_usage
def record_voucher_usage(*args, **kwargs):
time.sleep(0.5)
return org_record_voucher_usage(creator, *args, **kwargs)
creator.record_voucher_usage = record_voucher_usage
# Start 5 threads to place an order concurrently
def worker():
order_number = threading.current_thread().name
basket = factories.BasketFactory()
basket.add_product(product)
basket.vouchers.add(voucher)
place_order(
creator, basket=basket, order_number=order_number, user=user)
exceptions = run_concurrently(worker, num_threads=5)
voucher.refresh_from_db()
assert all(isinstance(x, ValueError) for x in exceptions), exceptions
assert len(exceptions) == 4
assert voucher.applications.count() == 1
assert Order.objects.count() == 1