def test_add_and_get_users(session_factory): repository = SqlAlchemyRepository(session_factory) assert repository.get_users() == [] repository.add_user(User("Brian Marshall", "123")) repository.add_user(User("Mark Tremonti", "456")) repository.add_user(User("Mark Tremonti", "456")) assert repository.get_user("Brian Marshall") == User( "Brian Marshall", "123") assert repository.get_user("Scott Phillips") is None assert repository.get_users() == [ User("Brian Marshall", "123"), User("Mark Tremonti", "456") ]
def test_can_retrieve_an_movie_and_add_a_comment_to_it(session_factory): repo = SqlAlchemyRepository(session_factory) # Fetch Movie and User. movie = repo.get_movie(5) author = repo.get_user('thorke') # Create a new Comment, connecting it to the Movie and User. comment = make_comment('First death in Australia', author, movie) movie_fetched = repo.get_movie(5) author_fetched = repo.get_user('thorke') assert comment in movie_fetched.comments assert comment in author_fetched.comments
def test_repository_can_add_a_comment(session_factory): repo = SqlAlchemyRepository(session_factory) user = repo.get_user('thorke') movie = repo.get_movie(2) comment = make_comment("Trump's onto it!", user, movie) repo.add_comment(comment) assert comment in repo.get_comments()
def test_repository_can_add_a_user(session_factory): repo = SqlAlchemyRepository(session_factory) user = User('Dave', '123456789') repo.add_user(user) repo.add_user(User('Martin', '123456789')) user2 = repo.get_user('Dave') assert user2 == user and user2 is user
def test_repository_does_not_retrieve_a_non_existent_user(session_factory): repo = SqlAlchemyRepository(session_factory) user = repo.get_user('Sam') assert user is None
def test_repository_can_retrieve_a_user(session_factory): repo = SqlAlchemyRepository(session_factory) user = repo.get_user('fmercury') assert user == User('fmercury', '8734gfe2058v')