Ross
27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
from http import HTTPStatus
|
|
|
|
from generic.models import CidUser, CidUserGroup, UserUserGroup
|
|
from django.contrib.auth.models import User
|
|
|
|
def AssertNotFound(client, url):
|
|
"""Helper to quickly test for urls that should return a 404 (NOT FOUND) error"""
|
|
response = client.get(url)
|
|
assert response.status_code == HTTPStatus.NOT_FOUND, f"Response content: {response.content}"
|
|
|
|
|
|
def create_cid_user_and_groups(db):
|
|
group1 = CidUserGroup.objects.create(name="Group1")
|
|
group2 = CidUserGroup.objects.create(name="Group2")
|
|
|
|
cid_user_1000 = CidUser.objects.create(cid=1000, passcode="ABCD", group=group1)
|
|
cid_user_1001 = CidUser.objects.create(cid=1001, passcode="EFGH", group=group1)
|
|
cid_user_2001 = CidUser.objects.create(cid=2001, passcode="EFGH", group=group2)
|
|
|
|
assert cid_user_1000.cid == 1000
|
|
assert cid_user_1001.cid == 1001
|
|
|
|
# Create UserUsers
|
|
user1 = User.objects.create_user(username="user1", email="test@email.com", password="test1234")
|
|
usergroup1 = UserUserGroup.objects.create(name="Group1")
|
|
usergroup1.users.add(user1)
|
|
usergroup2 = UserUserGroup.objects.create(name="Group2") |