def test__add_query_with_raw(self):
"""Test proper handling of mapping with raw field."""
view = ElasticListAPIView()
view.Meta.model = MagicMock()
view.Meta.model.field_has_raw.return_value = True
expectation = {'query': {'match': {'param1.raw': 'value1'}}}
elasticfilter = ElasticFilter()
queryset = Search()
# pylint: disable=W0212
result = elasticfilter._add_query('param1', 'value1', view, queryset)
self.assertTrue(view.Meta.model.field_has_raw.called)
self.assertEqual(result.to_dict(), expectation)
def test__add_query_no_raw(self):
"""Test proper handling of mapping without raw field."""
view = ElasticListAPIView()
view.Meta.model = MagicMock()
view.Meta.model.field_has_raw.return_value = False
expectation = {"query": {"match": {"param1": "value1"}}}
elasticfilter = ElasticFilter()
queryset = Search()
# pylint: disable=W0212
result = elasticfilter._add_query("param1", "value1", view, queryset)
self.assertTrue(view.Meta.model.field_has_raw.called)
self.assertEqual(result.to_dict(), expectation)
def test__add_query_no_raw(self):
"""Test proper handling of mapping without raw field."""
view = ElasticListAPIView()
view.Meta.model = MagicMock()
view.Meta.model.field_has_raw.return_value = False
expectation = {'query': {'match': {'param1': 'value1'}}}
elasticfilter = ElasticFilter()
queryset = Search()
# pylint: disable=W0212
result = elasticfilter._add_query('param1', 'value1', view, queryset)
self.assertTrue(view.Meta.model.field_has_raw.called)
self.assertEqual(result.to_dict(), expectation)
