def test_num_results(self, mock_search):
mock_search.return_value = [
SearchBackend._format_result(
SearchResult('rhel', 'Red Hat Enterprise Linux')),
SearchBackend._format_result(SearchResult('foo', 'Foo')),
SearchBackend._format_result(SearchResult('bar', 'Bar')),
]
response = self.test_client.get('/v1/search?q=rhel')
data = json.loads(response.data)
self.assertEqual(data['num_results'], 3)
def test_num_results(self, mock_search):
mock_search.return_value = [
SearchBackend._format_result(SearchResult('rhel', 'Red Hat Enterprise Linux',
**SearchResult.result_defaults)),
SearchBackend._format_result(SearchResult('foo', 'Foo',
**SearchResult.result_defaults)),
SearchBackend._format_result(SearchResult('bar', 'Bar',
**SearchResult.result_defaults)),
]
response = self.test_client.get('/v1/search?q=rhel')
data = json.loads(response.data)
self.assertEqual(data['num_results'], 3)
def test_with_results(self, mock_search):
mock_search.return_value = [
SearchBackend._format_result(SearchResult('rhel', 'Red Hat Enterprise Linux',
**SearchResult.result_defaults)),
]
response = self.test_client.get('/v1/search?q=rhel')
data = json.loads(response.data)
self.assertDictEqual(data, {
'query': 'rhel',
'num_results': 1,
'results': mock_search.return_value
})
def test_with_results(self, mock_search):
mock_search.return_value = [
SearchBackend._format_result(
SearchResult('rhel', 'Red Hat Enterprise Linux')),
]
response = self.test_client.get('/v1/search?q=rhel')
data = json.loads(response.data)
self.assertDictEqual(data, {
'query': 'rhel',
'num_results': 1,
'results': mock_search.return_value
})
