from mock import patch from decimal import Decimal as D from django.contrib.auth.models import User from django.core.urlresolvers import reverse from oscar.test.helpers import create_product, create_order from oscar.test import ClientTestCase, WebTestCase from oscar.apps.basket.models import Basket class TestASignedInUser(WebTestCase): email = 'customer@example.com' password = 'cheeseshop' def setUp(self): self.user = User.objects.create_user('_', self.email, self.password) self.order = create_order(user=self.user) def tearDown(self): Basket.objects.all().delete() def test_can_view_their_profile(self): response = self.app.get(reverse('customer:summary'), user=self.user) self.assertEqual(200, response.status_code) self.assertTrue(self.email in response.content) def test_can_update_their_email_address_and_name(self): profile_form_page = self.app.get(reverse('customer:profile-update'), user=self.user) self.assertEqual(200, profile_form_page.status_code) form = profile_form_page.forms['profile_form'] form['email'] = 'new@example.com' form['first_name'] = 'Barry' form['last_name'] = 'Chuckle' response = form.submit() self.assertRedirects(response, reverse('customer:summary')) user = User.objects.get(id=self.user.id) self.assertEqual('new@example.com', user.email) self.assertEqual('Barry', user.first_name) self.assertEqual('Chuckle', user.last_name) def test_cant_update_their_email_address_if_it_already_exists(self): User.objects.create_user(username='testuser', email='new@example.com', password="somerandompassword") self.assertEquals(User.objects.count(), 2) profile_form_page = self.app.get(reverse('customer:profile-update'), user=self.user) self.assertEqual(200, profile_form_page.status_code) form = profile_form_page.forms['profile_form'] form['email'] = 'new@example.com' form['first_name'] = 'Barry' form['last_name'] = 'Chuckle' response = form.submit() user = User.objects.get(id=self.user.id) self.assertEqual(self.email, user.email) try: User.objects.get(email='new@example.com') except User.MultipleObjectsReturned: self.fail("email for user changed to existing one") self.assertContains(response, 'A user with this email address already exists') def test_can_change_their_password(self): password_form_page = self.app.get(reverse('customer:change-password'), user=self.user) self.assertEqual(200, password_form_page.status_code) form = password_form_page.forms['change_password_form'] form['old_password'] = self.password form['new_password1'] = 'bubblesgopop' form['new_password2'] = 'bubblesgopop' response = form.submit() self.assertRedirects(response, reverse('customer:summary')) def test_can_reorder_a_previous_order(self): order_history_page = self.app.get(reverse('customer:order-list'), user=self.user) form = order_history_page.forms['order_form_%d' % self.order.id] form.submit() basket = Basket.open.get(owner=self.user) self.assertEqual(1, basket.all_lines().count()) def test_can_reorder_a_previous_order_line(self): order_history_page = self.app.get( reverse('customer:order', kwargs={'order_number': self.order.number}), user=self.user) line = self.order.lines.all()[0] form = order_history_page.forms['line_form_%d' % line.id] form.submit() basket = Basket.open.get(owner=self.user) self.assertEqual(1, basket.all_lines().count()) def test_cannot_reorder_an_out_of_stock_product(self): product = self.order.lines.all()[0].product product.stockrecord.num_in_stock = 0 product.stockrecord.save() order_history_page = self.app.get(reverse('customer:order-list'), user=self.user) form = order_history_page.forms['order_form_%d' % self.order.id] form.submit() basket = Basket.open.get(owner=self.user) self.assertEqual(0, basket.all_lines().count()) class TestReorderingOrderLines(ClientTestCase): # TODO - rework this as a webtest @patch('django.conf.settings.OSCAR_MAX_BASKET_QUANTITY_THRESHOLD', 1) def test_cannot_reorder_when_basket_maximum_exceeded(self): order = create_order(user=self.user) line = order.lines.all()[0] Basket.objects.all().delete() #add a product product = create_product(price=D('12.00')) self.client.post(reverse('basket:add'), {'product_id': product.id, 'quantity': 1}) basket = Basket.objects.all()[0] self.assertEquals(len(basket.all_lines()), 1) #try to reorder a product self.client.post(reverse('customer:order', args=(order.number,)), {'order_id': order.pk, 'action': 'reorder'}) self.assertEqual(len(basket.all_lines()), 1) self.assertNotEqual(line.product.pk, product.pk) @patch('django.conf.settings.OSCAR_MAX_BASKET_QUANTITY_THRESHOLD', 1) def test_cannot_reorder_line_when_basket_maximum_exceeded(self): order = create_order(user=self.user) line = order.lines.all()[0] Basket.objects.all().delete() #add a product product = create_product(price=D('12.00')) self.client.post(reverse('basket:add'), {'product_id': product.id, 'quantity': 1}) basket = Basket.objects.all()[0] self.assertEquals(len(basket.all_lines()), 1) self.client.post(reverse('customer:order-line', args=(order.number, line.pk)), {'action': 'reorder'}) self.assertEquals(len(basket.all_lines()), 1) self.assertNotEqual(line.product.pk, product.pk)