def load_users(data_path: str, repo: MemoryRepository):
users = dict()
for data_row in read_csv_file(os.path.join(data_path, 'users.csv')):
user = User(user_name=data_row[1],
password=generate_password_hash(data_row[2]))
repo.add_user(user)
users[int(data_row[0])] = user
return users
def load_users(data_path: str, repo: MovieRepo):
users = dict()
filename = os.path.join(data_path, "users.csv")
for data_row in read_csv_file(filename):
user = User(name=data_row[1],
password=generate_password_hash(data_row[2]))
repo.add_user(user)
users[data_row[0]] = user
return users
def load_users(data_path: str, repo: MainRepository):
users = dict()
for data_row in read_csv_file(data_path + '/users.csv'):
user = User(
user_name=data_row[1],
password=generate_password_hash(data_row[2])
)
repo.add_user(user)
users[data_row[0]] = user
return users
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)
repo.add_user(user)
def user():
return User('dbowie', '1234567890')
def test_repository_can_add_a_user(in_memory_repo):
user = User('Jane', '123456789')
in_memory_repo.add_user(user)
assert in_memory_repo.get_user('jane') is user
def test_repository_can_retrieve_a_user(in_memory_repo):
user = in_memory_repo.get_user('fmercury')
assert user == User('fmercury', '8734gfe2058v')
def test_repository_can_add_a_user(in_movie_repo):
user = User('Dave', '123456789')
in_movie_repo.add_user(user)
assert in_movie_repo.get_user('Dave') is user