def create_order(request, name, sku):
def can_create_publication_order(request, sku):
if Publication.objects.filter(pk=sku).exists():
publication = Publication.objects.get(pk=sku)
if publication.editor == request.user:
# when the item is created and only one (due to the rest of the process of create_order)
if not Item.objects.filter(name=name, sku=sku).exists():
return True
return False
def can_create_scientist_account_order(request, sku):
User = get_user_model()
if User.objects.filter(pk=sku).exists():
if int(request.user.pk) == int(sku):
# when the item is created and only one (due to the rest of the process of create_order)
if not Item.objects.filter(name=name, sku=sku).exists():
return True
return False
if name == PUBLICATION:
if not can_create_publication_order(request, sku):
return HttpResponseForbidden()
elif name == SCIENTIST_ACCOUNT:
if not can_create_scientist_account_order(request, sku):
return HttpResponseForbidden()
else:
return Http404()
item = Item(name=name, sku=int(sku))
item.save()
price = Price()
price.save()
order = Order(item=item, user=request.user, price=price)
order.save()
return redirect('detail_order', token=order.token)
def setUp(self):
User = get_user_model()
self.user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user.save()
self.item = Item(name="'scientist-account'", sku=self.user.pk)
self.item.save()
self.order = Order(status='new', user=self.user, item=self.item)
self.order.save()
self.url = reverse("address", kwargs={'token': self.order.token})
def setUp(self):
User = get_user_model()
self.user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user.save()
self.item = Item(name='scientist-account', sku=self.user.pk)
self.item.save()
self.order = Order(status='new', user=self.user, item=self.item)
self.order.save()
self.user2 = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user2.save()
self.item2 = Item(name='scientist-account', sku=self.user2.pk)
self.item2.save()
self.order2 = Order(status='new', user=self.user2, item=self.item2)
self.order2.save()
self.url = reverse('list_order')
class OrderOwnedTableViewTestCase(TestCase):
def setUp(self):
User = get_user_model()
self.user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user.save()
self.item = Item(name='scientist-account', sku=self.user.pk)
self.item.save()
self.order = Order(status='new', user=self.user, item=self.item)
self.order.save()
self.user2 = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user2.save()
self.item2 = Item(name='scientist-account', sku=self.user2.pk)
self.item2.save()
self.order2 = Order(status='new', user=self.user2, item=self.item2)
self.order2.save()
self.url = reverse('list_order')
def test_not_logged(self):
resp = self.client.get(self.url)
self.assertEqual(resp.status_code, 302)
def test_get_template(self):
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url)
self.assertEqual(resp.status_code, 200)
self.assertEqual(
resp.template_name,
['custompayment/order_list.html', 'custompayment/order_list.html'])
def test_get_queryset(self):
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url)
self.assertEqual(len(list(resp.context_data['object_list'])), 1)
def setUp(self):
User = get_user_model()
self.user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user.save()
self.item = Item(name='scientist-account', sku=self.user.pk)
self.item.save()
self.countrypayment = CountryPayment(country='FR',
pib_per_inhabitant='1000',
accepted=True)
self.countrypayment.save()
self.address = Address(scientist=self.user,
first_name='fname1',
last_name='lname1',
country='FR')
self.address.save()
self.price = Price()
self.price.save()
self.order = Order(status='new',
user=self.user,
item=self.item,
billing_address=self.address,
price=self.price)
self.order.save()
sdate = timezone.now() - datetime.timedelta(days=1)
edate = timezone.now() + datetime.timedelta(days=1)
self.discount = Discount(name='lol',
code='lol',
discount_for='scientist-account',
discount_type='percent',
discount_value='10',
starting_date=sdate,
ending_date=edate)
self.discount.save()
self.url = reverse("detail_order", kwargs={'token': self.order.token})
def test_str(self):
item = mock.Mock(spec=Item)
item.name = SCIENTIST_ACCOUNT
item.sku = 1
self.assertEqual(Item.__str__(item), 'Scientist account')
class BillingAddressUpdateTestCase(TestCase):
def setUp(self):
User = get_user_model()
self.user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user.save()
self.item = Item(name="'scientist-account'", sku=self.user.pk)
self.item.save()
self.order = Order(status='new', user=self.user, item=self.item)
self.order.save()
self.url = reverse("address", kwargs={'token': self.order.token})
def test_not_logged(self):
resp = self.client.get(self.url)
self.assertEqual(resp.status_code, 302)
def test_bad_user(self):
# test the user in the view to return 302 if not the same
User = get_user_model()
bad_user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
bad_user.save()
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 403)
def test_get_template(self):
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 200)
self.assertEqual(resp.template_name,
["custompayment/address_form.html"])
def test_create_address(self):
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 200)
self.assertIsInstance(resp.context_data['object'], Address)
def test_get_existing_address_not_linked(self):
address = Address(scientist=self.user,
first_name='fname1',
last_name='lname1')
address.save()
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 200)
self.assertIsInstance(resp.context_data['object'], Address)
self.assertEqual(resp.context_data['object'].first_name, 'fname1')
self.assertEqual(resp.context_data['object'].last_name, 'lname1')
def test_get_existing_address_linked(self):
address = Address(scientist=self.user,
first_name='fname1',
last_name='lname1')
address.save()
self.order.billing_address = address
self.order.save()
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 200)
self.assertIsInstance(resp.context_data['object'], Address)
self.assertEqual(resp.context_data['object'].first_name, 'fname1')
self.assertEqual(resp.context_data['object'].last_name, 'lname1')
def test_form_valid(self):
assert self.client.login(username="******", password="******")
resp = self.client.post(
self.url, {
"first_name": 'f1',
"last_name": 'f1',
"street_address_1": 'f1',
"city": 'f1',
"postal_code": 'f1',
"country": 'FR'
})
self.assertEqual(resp.status_code, 302)
self.assertEqual(resp._headers['location'][1],
'/checkout/' + self.order.token + '/')
def test_form_invalid(self):
assert self.client.login(username="******", password="******")
resp = self.client.post(
self.url, {
"last_name": 'f1',
"street_address_1": 'f1',
"city": 'f1',
"postal_code": 'f1',
"country": 'FR'
})
self.assertEqual(resp.status_code, 200)
self.assertEqual(resp.template_name,
["custompayment/address_form.html"])
class OrderDetailViewTestCase(TestCase):
def setUp(self):
User = get_user_model()
self.user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
self.user.save()
self.item = Item(name='scientist-account', sku=self.user.pk)
self.item.save()
self.countrypayment = CountryPayment(country='FR',
pib_per_inhabitant='1000',
accepted=True)
self.countrypayment.save()
self.address = Address(scientist=self.user,
first_name='fname1',
last_name='lname1',
country='FR')
self.address.save()
self.price = Price()
self.price.save()
self.order = Order(status='new',
user=self.user,
item=self.item,
billing_address=self.address,
price=self.price)
self.order.save()
sdate = timezone.now() - datetime.timedelta(days=1)
edate = timezone.now() + datetime.timedelta(days=1)
self.discount = Discount(name='lol',
code='lol',
discount_for='scientist-account',
discount_type='percent',
discount_value='10',
starting_date=sdate,
ending_date=edate)
self.discount.save()
self.url = reverse("detail_order", kwargs={'token': self.order.token})
def test_not_logged(self):
resp = self.client.get(self.url)
self.assertEqual(resp.status_code, 302)
def test_bad_user(self):
# test the user in the view to return 302 if not the same
User = get_user_model()
bad_user = User.objects.create_user(username="******",
password="******",
email="*****@*****.**")
bad_user.save()
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 403)
def test_get_template(self):
assert self.client.login(username="******", password="******")
resp = self.client.get(self.url, follow=True)
self.assertEqual(resp.status_code, 200)
self.assertEqual(resp.template_name,
['custompayment/order_detail.html'])
def test_get_404(self):
assert self.client.login(username="******", password="******")
url = reverse("detail_order", kwargs={'token': '43422'})
resp = self.client.get(url, follow=True)
self.assertEqual(resp.status_code, 404)
def test_fill_bad_discount_code(self):
assert self.client.login(username="******", password="******")
resp = self.client.post(self.url, {
'discount': '1234',
'submit': "Apply"
},
follow=True)
self.assertEqual(resp.status_code, 200)
self.assertIsNotNone(resp.context_data['form'].__dict__['_errors'])
def test_fill_good_discount_code(self):
assert self.client.login(username="******", password="******")
resp = self.client.post(self.url, {
'discount': 'lol',
'submit': "Apply"
},
follow=True)
self.assertEqual(resp.status_code, 200)
order_discount = Order.objects.get(token=self.order.token).discount
self.assertEqual(order_discount, self.discount)
def test_accept(self):
assert self.client.login(username="******", password="******")
resp = self.client.post(self.url, {'submit': "Accept"}, follow=True)
self.assertEqual(resp.status_code, 200)
self.assertEqual(resp.template_name, 'custompayment/payment.html')