def main():
q = request.args.get('q')
if not q:
abort(400)
try:
recommendation = SearchRecommendation(q, request)
response = recommendation.do_search(q)
except CacheMissError:
return jsonify({}), 202
except Exception as e:
if app.config['DEBUG']:
return jsonify({e.__class__.__name__: e.args}), 500
return jsonify({}), 500
else:
return jsonify(response), 200
def main():
q = request.args.get('q')
if not q:
abort(400)
try:
recommendation = SearchRecommendation(q, request)
response = recommendation.do_search(q)
except CacheMissError:
return jsonify({}), 202
except Exception as e:
if app.config['DEBUG']:
return jsonify({e.__class__.__name__: e.args}), 500
return jsonify({}), 500
else:
return jsonify(response), 200
def setUp(self):
self.instance = SearchRecommendation('', None)
class TestSearchRecommendation(TestCase):
def setUp(self):
self.instance = SearchRecommendation('', None)
def tearDown(self):
mock_memcached.flush_all()
def test_get_suggestion_engine(self):
engine = self.instance.get_suggestion_engine()
ok_(issubclass(engine, BaseSuggestionEngine))
def test_get_query_engine(self):
engine = self.instance.get_query_engine()
ok_(issubclass(engine, BaseQueryEngine))
@patch('app.search.classification.domain.DomainClassifier.is_match')
def test_get_classifiers(self, mock_match):
mock_match.return_value = True
classifiers = self.instance.get_classifiers(
{'url': 'http://%s/' % DOMAIN})
eq_(len(classifiers), 1)
ok_(isinstance(classifiers[0], DomainClassifier))
return classifiers
@patch(('app.search.recommendation.SearchRecommendation.get_suggestion_eng'
'ine'))
@patch('app.search.suggest.bing.BingSuggestionEngine.search')
def test_get_suggestions(self, mock_bing, mock_suggestion_engine):
mock_bing.return_value = BING_RESULTS
mock_suggestion_engine.return_value = BingSuggestionEngine
eq_(self.instance.get_suggestions(BING_QUERY), BING_RESULTS)
eq_(mock_bing.call_count, 1)
def test_get_top_suggestion(self):
eq_(self.instance.get_top_suggestion(BING_RESULTS), BING_RESULTS[0])
@patch('app.search.recommendation.SearchRecommendation.get_query_engine')
@patch('app.search.query.yahoo.YahooQueryEngine.search')
def test_do_query(self, mock_yahoo, mock_query_engine):
response = YAHOO_RESPONSE['bossresponse']['web']['results'][0]
mock_yahoo.return_value = response
mock_query_engine.return_value = YahooQueryEngine
eq_(self.instance.do_query(YAHOO_QUERY), response)
eq_(mock_yahoo.call_count, 1)
@patch('app.search.recommendation.SearchRecommendation.get_classifiers')
@patch('app.search.recommendation.SearchRecommendation.do_query')
@patch('app.search.recommendation.SearchRecommendation.get_top_suggestion')
@patch('app.search.recommendation.SearchRecommendation.get_suggestions')
@patch('app.memorize.memcached', mock_memcached)
def test_do_search_get_recommendation(self, mock_suggestions,
mock_top_suggestion, mock_result,
mock_classifiers):
suggestions = BING_RESULTS
top_suggestion = BING_RESULTS[0]
result = YAHOO_RESPONSE['bossresponse']['web']['results'][0]
classifiers = [DomainClassifier(result)]
mock_suggestions.return_value = suggestions
mock_top_suggestion.return_value = top_suggestion
mock_result.return_value = result
mock_classifiers.return_value = classifiers
with self.assertRaises(CacheMissError):
self.instance.do_search(QUERY)
search = self.instance.do_search(QUERY)
ok_(search.from_cache)
ok_(all([k in search for k in ['enhancements', 'query', 'result']]))
eq_(list(search['enhancements'].keys()), [c.type for c in classifiers])
eq_(search['enhancements']['domain'], classifiers[0].enhance())
eq_(search['query']['completed'], top_suggestion)
eq_(search['query']['original'], QUERY)
eq_(search['result'], result)
def setUp(self):
self.instance = SearchRecommendation('')
class TestSearchRecommendation(TestCase):
def setUp(self):
self.instance = SearchRecommendation('')
def tearDown(self):
mock_memcached.flush_all()
def test_get_suggestion_engine(self):
engine = self.instance.get_suggestion_engine()
ok_(issubclass(engine, BaseSuggestionEngine))
def test_get_query_engine(self):
engine = self.instance.get_query_engine()
ok_(issubclass(engine, BaseQueryEngine))
@patch('app.search.classification.domain.DomainClassifier.is_match')
def test_get_classifiers(self, mock_match):
mock_match.return_value = True
classifiers = self.instance.get_classifiers({
'url': 'http://%s/' % DOMAIN
})
eq_(len(classifiers), 1)
ok_(isinstance(classifiers[0], DomainClassifier))
return classifiers
@patch(('app.search.recommendation.SearchRecommendation.get_suggestion_eng'
'ine'))
@patch('app.search.suggest.bing.BingSuggestionEngine.search')
def test_get_suggestions(self, mock_bing, mock_suggestion_engine):
mock_bing.return_value = BING_RESULTS
mock_suggestion_engine.return_value = BingSuggestionEngine
eq_(self.instance.get_suggestions(BING_QUERY), BING_RESULTS)
eq_(mock_bing.call_count, 1)
def test_get_top_suggestion(self):
eq_(self.instance.get_top_suggestion(BING_RESULTS), BING_RESULTS[0])
@patch('app.search.recommendation.SearchRecommendation.get_query_engine')
@patch('app.search.query.yahoo.YahooQueryEngine.search')
def test_do_query(self, mock_yahoo, mock_query_engine):
response = YAHOO_RESPONSE['bossresponse']['web']['results'][0]
mock_yahoo.return_value = response
mock_query_engine.return_value = YahooQueryEngine
eq_(self.instance.do_query(YAHOO_QUERY), response)
eq_(mock_yahoo.call_count, 1)
@patch('app.search.recommendation.SearchRecommendation.get_classifiers')
@patch('app.search.recommendation.SearchRecommendation.do_query')
@patch('app.search.recommendation.SearchRecommendation.get_top_suggestion')
@patch('app.search.recommendation.SearchRecommendation.get_suggestions')
@patch('app.memorize.memcached', mock_memcached)
def test_do_search_get_recommendation(self, mock_suggestions,
mock_top_suggestion, mock_result,
mock_classifiers):
suggestions = BING_RESULTS
top_suggestion = BING_RESULTS[0]
result = YAHOO_RESPONSE['bossresponse']['web']['results'][0]
classifiers = [DomainClassifier(result)]
mock_suggestions.return_value = suggestions
mock_top_suggestion.return_value = top_suggestion
mock_result.return_value = result
mock_classifiers.return_value = classifiers
search = self.instance.do_search(QUERY)
ok_(all([k in search for k in ['enhancements', 'query', 'result']]))
eq_(list(search['enhancements'].keys()), [c.type for c in classifiers])
eq_(search['enhancements']['domain'], classifiers[0].enhance())
eq_(search['query']['completed'], top_suggestion)
eq_(search['query']['original'], QUERY)
eq_(search['result'], result)
