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"
