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__