class ElasticSearchBaseQueryCreationTests(TestCase):
MATCH_ALL = {'match_all': {}}
def setUp(self):
self.query_creator = ElasticSearchBaseQueryCreator()
def test_baseQueryCreation_textQueryProvided_baseQueryCreated(self):
TEXT = 'some text query'
proper_base_query = {
'bool': {
'should': [{
'match': {
'title': {
'query': TEXT,
'boost': 3,
'fuzziness': 1
}
}
}, {
'match': {
'dataSample': {
'query': TEXT,
'boost': 2
}
}
}, {
'match': {
'sourceUri': {
'query': TEXT,
}
}
}]
}
}
self.assertDictEqual(
proper_base_query,
self.query_creator.create_base_query({'query': TEXT}))
def test_baseQueryCreation_noQueryElement_matchAllReturned(self):
self.assertDictEqual(self.MATCH_ALL,
self.query_creator.create_base_query({}))
def test_baseQueryCreation_emptyQuery_matchAllReturned(self):
self.assertDictEqual(
self.MATCH_ALL, self.query_creator.create_base_query({'query':
''}))
def setUp(self):
self.query_creator = ElasticSearchBaseQueryCreator()
def test_baseQueryCreation_noQueryElement_matchAllReturned(self):
self.assertDictEqual(
self.MATCH_ALL,
ElasticSearchBaseQueryCreator.create_base_query({}))
def test_baseQueryCreation_emptyQuery_matchAllReturned(self):
self.assertDictEqual(
self.MATCH_ALL,
ElasticSearchBaseQueryCreator.create_base_query({'query': ''}))
def test_baseQueryCreation_textQueryProvided_baseQueryCreated(self):
TEXT = 'some text query'
self.assertDictEqual(
ElasticSearchBaseQueryCreator.render_es_query(TEXT),
ElasticSearchBaseQueryCreator.create_base_query({'query': TEXT}))
