def test_bankcard_card_correct_save(self): # issue #1486 user_klass = get_user_model() user = user_klass.objects.create_user('_', '*****@*****.**', 'pwd') end = datetime.date(day=1, month=1, year=2010) bankcard = Bankcard.objects.create(user=user, number="5500000000000004", expiry_date=end) saved_bankcard = Bankcard.objects.get(id=bankcard.id) self.assertEqual('Mastercard', saved_bankcard.card_type)
from django.conf import settings from django.test import TestCase from django.urls import reverse from django.utils.six.moves import http_client from django.utils.translation import ugettext from wshop.apps.basket import reports from wshop.apps.basket.models import Basket from wshop.apps.partner import strategy from wshop.core.compat import get_user_model, unquote_cookie from wshop.test import factories from wshop.test.basket import add_product from wshop.test.factories import create_product from wshop.test.testcases import WebTestCase User = get_user_model() class TestBasketMerging(TestCase): def setUp(self): self.product = create_product(num_in_stock=10) self.user_basket = Basket() self.user_basket.strategy = strategy.Default() add_product(self.user_basket, product=self.product) self.cookie_basket = Basket() self.cookie_basket.strategy = strategy.Default() add_product(self.cookie_basket, quantity=2, product=self.product) self.user_basket.merge(self.cookie_basket, add_quantities=False) def test_cookie_basket_has_status_set(self): self.assertEqual(Basket.MERGED, self.cookie_basket.status)
class Meta: model = get_user_model()
def setUp(self): self.user_klass = get_user_model()