def search():
"""
Renders the search results page.
"""
context = helpers.get_context(request)
dbconnection = database.DatabaseConnection()
dbconnection.connect()
if context.get('query'):
query_index = database.build_mongo_index_query(input=context['query'])
#flash('query: '+str(mongo_query), category='text-info')
context['query_index'] = query_index
if query_index != None:
cursor, cursor_count = database.query_mongo(query=query_index, collection=settings.COLLECTION_INDEX, db=dbconnection)
context['cursor'] = cursor
context['cursor_count'] = cursor_count
query_pages, query_pages_hits = database.build_mongo_pages_query(input=cursor)
if (query_pages_hits != {}):
context['query_pages'] = query_pages
context['query_pages_hits'] = query_pages_hits
documents, documents_count = database.query_mongo(query=query_pages, collection=settings.COLLECTION_DOCUMENTS, db=dbconnection, sort=context['sort'], number_of_results=context['results'], order=context['order'])
context['documents'] = documents
context['documents_count'] = documents_count
if(context['documents_count'] == 0):
no_results()
else:
no_results()
else:
no_results()
return render_template("search.html", context=context)
def test_building_a_one_word_mongo_pages_query(self):
"""
Tests if a one word search will build the proper mongo pages query.
"""
self.test_input_cursor = {'word1': {'url1':[9], 'url2':[1], 'url3':[1,3,4], 'url4':[6]}}
self.expected_output = {'$or':[{'url':'url4'},{'url':'url1'},{'url':'url3'},{'url':'url2'}]}
self.expected_hits = {'url1':[9], 'url2':[1], 'url3':[1,3,4], 'url4':[6]}
self.result, self.hits = database.build_mongo_pages_query(input=self.test_input_cursor)
self.assertEqual(self.result, self.expected_output), "Building a page query for a one word search is broken."
self.assertEqual(self.hits, self.expected_hits), "Building a page word hits list for a one word search is broken."