def seeding(index: int, single: bool = False, save: bool = True, **kwargs) -> models.QuerySet:
from apps.order.utils import OrderUtils
order = OrderUtils.seeding(1, True)
if index == 0:
raise Exception('Indext must be start with 1.')
def get_data(i: int) -> dict:
data = {
'order': order.id,
'title': "title{}".format(i),
'url': "url{}".format(i),
'quantity': i,
'unit_price': 10.5 + i
}
_order = kwargs.get('order', None)
if _order is not None:
data['order'] = _order.pk
if save is False:
return data
instance = OrderItemBaseSr(data=data)
instance.is_valid(raise_exception=True)
instance = instance.save()
return instance
def get_list_data(index):
return [get_data(i) for i in range(1, index + 1)]
return get_data(index) if single is True else get_list_data(index)
def setUp(self):
order = OrderUtils.seeding(1, True)
self.items = BolUtils.seeding(3)
self.ids = [self.items[0].pk, self.items[1].pk]
for item in self.items:
item.order = order
item.save()
def setUp(self):
order = OrderUtils.seeding(1, True)
self.items = BolUtils.seeding(2)
self.ids = [self.items[0].pk, self.items[1].pk]
self.items[0].order = order
self.items[0].save()
def test_normal_case(self):
order = OrderUtils.seeding(1, True)
staff = StaffUtils.seeding(1, True)
TransactionUtils.deposit(order, staff)
self.assertEqual(Transaction.objects.count(), 1)
TransactionUtils.undeposit(order)
self.assertEqual(Transaction.objects.count(), 0)
def test_normal_order_with_register(self):
item = BolUtils.seeding(1, True)
order = OrderUtils.seeding(1, True)
item.order = order
item.insurance = True
item.cny_insurance_value = 50
item.save()
self.assertEqual(BolUtils.cal_insurance_fee(item), 0)
def test_match(self):
purchase_code = '123456'
order = OrderUtils.seeding(1, True)
order.purchase_code = purchase_code
order.save()
item = Bol.objects.create(uid='test', purchase_code=purchase_code)
self.assertEqual(item.order, order)
def setUp(self):
self.customer = CustomerUtils.seeding(1, True)
self.staff = StaffUtils.seeding(1, True)
self.order = OrderUtils.seeding(1, True, customer=self.customer)
self.order.status = Status.NEW
OrderItemUtils.seeding(1, True, order=self.order)
self.bol = BolUtils.seeding(1, True, order=self.order)
self.bol.mass = 5
self.bol.save()
OrderUtils.force_cal(self.order)
def test_normal_case(self):
order = OrderUtils.seeding(1, True)
data = {
'order': order.id,
'title': "title1",
'url': "url1",
'shop_link': "shop_link1",
'shop_nick': "shop_nick1",
'site': "site1",
'quantity': 3,
'unit_price': 50.5
}
order_item = OrderItemBaseSr(data=data)
order_item.is_valid(raise_exception=True)
order_item.save()
self.assertEqual(order_item.data['price'], 151.5)
def test_normal_case(self):
order = OrderUtils.seeding(1, True)
OrderItemUtils.seeding(2, order=order)
staff = StaffUtils.seeding(1, True)
TransactionUtils.recharge(100000000, MoneyType.CASH, order.customer, staff)
self.assertEqual(Transaction.objects.count(), 1)
TransactionUtils.deposit(order, staff)
transaction = Transaction.objects.first()
self.assertEqual(Transaction.objects.count(), 2)
self.assertEqual(transaction.order, order)
self.assertEqual(transaction.staff, staff)
self.assertEqual(transaction.customer, order.customer)
self.assertEqual(transaction.amount, OrderUtils.get_deposit_amount(order))
self.assertEqual(transaction.type, Type.DEPOSIT)