def test_get_existing_movie(): create_test_tables() db = MySQLDatabase(schema='qmdb_test', env='test') movie = db.get_movie(1234) assert movie.title == 'The Matrix' assert movie.year == 1999 remove_test_tables(db)
def test_get_non_existent_movie(): create_test_tables() db = MySQLDatabase(schema='qmdb_test', env='test') with pytest.raises(MovieNotInDatabaseError) as e_info: movie = db.get_movie(12345) remove_test_tables(db)