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")