Add tests for search results