def setUp(self):
self.cm = CustomerMiddleware()
self.factory = RequestFactory()
User = get_user_model()
self.lisa = User.objects.create(**self.credentials['lisa'])
self.bart = User.objects.create(**self.credentials['bart'])
# bart should already have a Customer account
Customer.objects.create(user=self.bart)
def setUp(self):
super(CustomerTest, self).setUp()
User = get_user_model()
# Lisa has a Django account, but no Customer account yet
self.lisa = User.objects.create(**self.USERS['lisa'])
# Bart has a Django account and a Customer account
self.bart = User.objects.create(**self.USERS['bart'])
Customer.objects.create(user=self.bart).save()
self.factory = RequestFactory()
self.customer_middleware = CustomerMiddleware()
def middleware_process_request(self, request, sessionid=None):
if sessionid:
request.COOKIES[settings.SESSION_COOKIE_NAME] = sessionid
SessionMiddleware().process_request(request)
AuthenticationMiddleware().process_request(request)
CustomerMiddleware().process_request(request)
