def test_loading_of_users(empty_session):
users = list()
users.append(("Andrew", "1234"))
users.append(("Cindy", "1111"))
insert_users(empty_session, users)
user1 = User("Andrew", "1234")
user2 = User("Cindy", "999")
user1.username = '******'
user1.password = '******'
user2.username = '******'
user2.password = '******'
assert empty_session.query(User).all() == [user1, user2]
def test_repository_can_retrieve_a_user(session_factory):
repo = SqlAlchemyRepository(session_factory)
user = repo.get_user('fmercury')
user2 = User('fmercury', '8734gfe2058v')
user2.username = '******'
user2.password = '******'
assert user == user2
def test_repository_can_add_a_user(session_factory):
repo = SqlAlchemyRepository(session_factory)
user = User('Dave', '123456789')
user.username = '******'
user.password = '******'
repo.add_user(user)
user2 = repo.get_user('Dave')
assert user2 == user
def add_user(username: str, password: str, repo: AbstractRepository):
# Check that the given username is available.
user = repo.get_user(username)
if user is not None:
raise NameNotUniqueException
# Encrypt password so that the database doesn't store passwords 'in the clear'.
password_hash = generate_password_hash(password)
# Create and store the new User, with password encrypted.
user = User(username, password_hash)
user.username = username
user.password = password_hash
repo.add_user(user)
def make_user():
user = User('Bob', '212434')
user.password = '******'
user.username = '******'
return user