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'