99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# ITerativ GmbH
|
|
# http://www.iterativ.ch/
|
|
#
|
|
# Copyright (c) 2019 ITerativ GmbH. All rights reserved.
|
|
#
|
|
# Created on 2019-10-02
|
|
# @author: chrigu <christian.cueni@iterativ.ch>
|
|
import json
|
|
from unittest.mock import patch
|
|
from django.contrib.sessions.middleware import SessionMiddleware
|
|
from django.test import TestCase, RequestFactory
|
|
from graphene.test import Client
|
|
|
|
from api.schema_public import schema
|
|
from core.factories import UserFactory
|
|
from core.hep_client import HepClient
|
|
from registration.factories import LicenseTypeFactory, LicenseFactory
|
|
from users.models import Role
|
|
|
|
FAKE_TOKEN = 'abcd12345!'
|
|
|
|
with open('orders.json', 'r') as file:
|
|
order_data = file.read()
|
|
|
|
ORDERS = json.loads(order_data)
|
|
|
|
|
|
class PasswordResetTests(TestCase):
|
|
def setUp(self):
|
|
self.user = UserFactory(username='aschi@iterativ.ch', email='aschi@iterativ.ch')
|
|
self.teacher_role = Role.objects.create(key=Role.objects.TEACHER_KEY, name="Teacher Role")
|
|
self.teacher_license_type = LicenseTypeFactory(for_role=self.teacher_role)
|
|
|
|
request = RequestFactory().post('/')
|
|
|
|
# adding session
|
|
middleware = SessionMiddleware()
|
|
middleware.process_request(request)
|
|
request.session.save()
|
|
self.client = Client(schema=schema, context_value=request)
|
|
|
|
def make_login_mutation(self, username, password):
|
|
mutation = '''
|
|
mutation Login($input: LoginInput!){
|
|
login(input: $input) {
|
|
success
|
|
errors {
|
|
field
|
|
}
|
|
}
|
|
}
|
|
'''
|
|
|
|
return self.client.execute(mutation, variables={
|
|
'input': {
|
|
'usernameInput': username,
|
|
'passwordInput': password
|
|
}
|
|
})
|
|
|
|
@patch.object(HepClient, 'customer_token', return_value={'token': FAKE_TOKEN})
|
|
@patch.object(HepClient, 'customer_orders', return_value=ORDERS)
|
|
def test_user_can_login(self, orders_mock, token_mock):
|
|
result = self.make_login_mutation(self.user.email, 'test123')
|
|
self.assertTrue(result.get('data').get('login').get('success'))
|
|
self.assertTrue(self.user.is_authenticated)
|
|
|
|
def test_user_cannot_login_with_invalid_password(self):
|
|
password = 'test123'
|
|
self.user.set_password(password)
|
|
self.user.save()
|
|
|
|
result = self.make_login_mutation(self.user.email, 'test1234')
|
|
self.assertFalse(result.get('data').get('login').get('success'))
|
|
|
|
def test_user_with_active_license_can_login(self):
|
|
password = 'test123'
|
|
self.user.set_password(password)
|
|
self.user.save()
|
|
|
|
LicenseFactory(license_type=self.teacher_license_type, licensee=self.user)
|
|
|
|
result = self.make_login_mutation(self.user.email, password)
|
|
self.assertTrue(result.get('data').get('login').get('success'))
|
|
|
|
def test_user_with_inactive_license_cannot_login(self):
|
|
password = 'test123'
|
|
self.user.set_password(password)
|
|
self.user.save()
|
|
|
|
self.teacher_license_type.active = False
|
|
self.teacher_license_type.save()
|
|
LicenseFactory(license_type=self.teacher_license_type, licensee=self.user)
|
|
|
|
result = self.make_login_mutation(self.user.email, password)
|
|
self.assertFalse(result.get('data').get('login').get('success'))
|