def test_should_search_posts(self):
yql_query = 'SELECT * FROM meme.search(5) WHERE query = "a sample query"'
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.search("a sample query", 5)
assert len(posts) == 1
assert posts[0].guid == "123"
def test_should_search_posts(self):
yql_mock = Mock()
yql_query = 'SELECT * FROM meme.search(0,10) WHERE query="a sample query"'
query_result = Mock()
query_result.rows = {'guid':'123', 'pubid':'123',
'type':'post', 'caption':'blah', 'content':'blah',
'comment':'blah', 'url':'http://meme.yahoo.com/p/123',
'timestamp':'1234567890', 'repost_count':'12345'}
query_result.count = 1
when(yql_mock).execute(yql_query).thenReturn(query_result)
repository = PostRepository()
repository.yql = yql_mock
posts = repository.search('a sample query', 10)
assert len(posts) == 1
assert posts[0].guid == '123'
