WMGZON/tests/unit/database/test_users.py

import pytest
import sqlite3
from controllers.database.user import UserController
from models.users.customer import Customer
from models.users.seller import Seller

customer = Customer(
    "testcustomer",
    "Password1",
    "firstname",
    "lastname",
    "test@test",
    "123456789"
)

seller = Seller(
    "testseller",
    "Password1",
    "firstname",
    "lastname",
    "seller@seller",
    "987654321"
)

# Tests a new user can be created


def test_create_user():
    db = UserController()
    db.create(customer)

# Tests the database maintains integrity when we try
# and add a user with the same details


def test_duplicate_user():
    db = UserController()
    with pytest.raises(sqlite3.IntegrityError):
        db.create(customer)

# Test we the same user details get returned from the database


def test_read_user():
    db = UserController()

    # Test the same user is returned
    user = db.read(customer.username)
    assert isinstance(user, Customer)

    # Update the ID on the item as database assigns new id
    customer.id = user.id
    assert user.__dict__ == customer.__dict__


# Tests a new seller can be created
def test_create_seller():
    db = UserController()
    db.create(seller)

# Test that the same seller details get returned from the database


def test_read_seller():
    db = UserController()

    # Test the same user is returned
    user = db.read(seller.username)
    assert isinstance(user, Seller)

    # Update the ID on the item as database assigns new id
    seller.id = user.id
    user.store = ""
    assert user.__dict__ == seller.__dict__