def test_should_get_only_one_meme_post(self): yql_query = 'SELECT * FROM meme.posts(1) WHERE owner_guid = "foo123bar"' yql_mock = Mock() when(yql_mock).execute(yql_query).thenReturn(self.single_query_result) post_repository = PostRepository() post_repository.yql = yql_mock posts = post_repository.get_by_meme("foo123bar", 1) assert len(posts) == 1 assert posts[0].guid == "123"
def test_should_get_meme_posts(self): yql_query = 'SELECT * FROM meme.posts(2) WHERE owner_guid = "foo123bar"' yql_mock = Mock() when(yql_mock).execute(yql_query).thenReturn(self.multiple_query_result) post_repository = PostRepository() post_repository.yql = yql_mock posts = post_repository.get_by_meme("foo123bar", 2) assert len(posts) == 2 assert posts[0].guid == "123" assert posts[1].guid == "456"