mirror of
https://github.com/graphql-python/graphene-django.git
synced 2025-07-13 17:52:19 +03:00
refactoring of elasticsearch filter test
This commit is contained in:
Alejandro Nunez Capote
parent
8692c82266
commit
721894fe1a
68
graphene_django/elasticsearch/tests/commons.py
Normal file
68
graphene_django/elasticsearch/tests/commons.py
Normal file
|
|
@ -0,0 +1,68 @@
|
|||
from datetime import datetime
|
||||
|
||||
from mock import mock
|
||||
|
||||
from graphene import Schema
|
||||
|
||||
from graphene_django.tests.models import Article, Reporter
|
||||
from graphene_django.elasticsearch.tests.filters import ESFilterQuery, ArticleDocument
|
||||
|
||||
|
||||
def fake_data():
|
||||
r1 = Reporter.objects.create(first_name="r1", last_name="r1", email="r1@test.com")
|
||||
a1 = Article.objects.create(
|
||||
headline="a1",
|
||||
pub_date=datetime.now(),
|
||||
pub_date_time=datetime.now(),
|
||||
reporter=r1,
|
||||
editor=r1,
|
||||
)
|
||||
a2 = Article.objects.create(
|
||||
headline="a2",
|
||||
pub_date=datetime.now(),
|
||||
pub_date_time=datetime.now(),
|
||||
reporter=r1,
|
||||
editor=r1,
|
||||
)
|
||||
return a1, a2
|
||||
|
||||
|
||||
def generate_query(field, query_str):
|
||||
query = """
|
||||
query {
|
||||
%s(%s) {
|
||||
edges {
|
||||
node {
|
||||
headline
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
""" % (field, query_str)
|
||||
return query
|
||||
|
||||
|
||||
def filter_generation(field, query_str, expected_arguments, method_to_mock="query"):
|
||||
a1, a2 = fake_data()
|
||||
|
||||
query = generate_query(field, query_str)
|
||||
|
||||
mock_count = mock.Mock(return_value=3)
|
||||
mock_slice = mock.Mock(return_value=mock.Mock(to_queryset=mock.Mock(
|
||||
return_value=Article.objects.filter(pk__in=[a1.id, a2.id])
|
||||
)))
|
||||
mock_query = mock.Mock(return_value=ArticleDocument.search())
|
||||
|
||||
with mock.patch('django_elasticsearch_dsl.search.Search.count', mock_count), \
|
||||
mock.patch('django_elasticsearch_dsl.search.Search.__getitem__', mock_slice), \
|
||||
mock.patch("elasticsearch_dsl.Search.%s" % method_to_mock, mock_query):
|
||||
schema = Schema(query=ESFilterQuery)
|
||||
result = schema.execute(query)
|
||||
|
||||
assert not result.errors
|
||||
|
||||
mock_query.assert_called_with(expected_arguments)
|
||||
|
||||
assert len(result.data[field]["edges"]) == 2
|
||||
assert result.data[field]["edges"][0]["node"]["headline"] == "a1"
|
||||
assert result.data[field]["edges"][1]["node"]["headline"] == "a2"
|
||||
|
|
@ -1,325 +0,0 @@
|
|||
from datetime import datetime
|
||||
|
||||
import pytest
|
||||
from py.test import raises
|
||||
from mock import mock
|
||||
|
||||
from elasticsearch_dsl.query import Bool, Match, Term, Wildcard, MatchPhrase, MatchPhrasePrefix, Range, Terms, Exists
|
||||
from graphene import Schema, ObjectType
|
||||
|
||||
from graphene_django.elasticsearch.filter.fields import DjangoESFilterConnectionField
|
||||
from graphene_django.elasticsearch.filter.filterset import FilterSetES
|
||||
from graphene_django.filter.tests.test_fields import ArticleNode
|
||||
from graphene_django.tests.models import Article, Reporter
|
||||
from graphene_django.utils import DJANGO_FILTER_INSTALLED, DJANGO_ELASTICSEARCH_DSL_INSTALLED
|
||||
from graphene_django.elasticsearch.tests.filters import ESFilterQuery, ArticleDocument
|
||||
|
||||
pytestmark = []
|
||||
|
||||
if not DJANGO_FILTER_INSTALLED or not DJANGO_ELASTICSEARCH_DSL_INSTALLED:
|
||||
pytestmark.append(
|
||||
pytest.mark.skipif(
|
||||
True, reason="django_filters not installed or not compatible"
|
||||
)
|
||||
)
|
||||
|
||||
pytestmark.append(pytest.mark.django_db)
|
||||
|
||||
|
||||
def fake_data():
|
||||
r1 = Reporter.objects.create(first_name="r1", last_name="r1", email="r1@test.com")
|
||||
a1 = Article.objects.create(
|
||||
headline="a1",
|
||||
pub_date=datetime.now(),
|
||||
pub_date_time=datetime.now(),
|
||||
reporter=r1,
|
||||
editor=r1,
|
||||
)
|
||||
a2 = Article.objects.create(
|
||||
headline="a2",
|
||||
pub_date=datetime.now(),
|
||||
pub_date_time=datetime.now(),
|
||||
reporter=r1,
|
||||
editor=r1,
|
||||
)
|
||||
return a1, a2
|
||||
|
||||
|
||||
def generate_query(field, query_str):
|
||||
query = """
|
||||
query {
|
||||
%s(%s) {
|
||||
edges {
|
||||
node {
|
||||
headline
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
""" % (field, query_str)
|
||||
return query
|
||||
|
||||
|
||||
def filter_generation(field, query_str, expected_arguments, method_to_mock="query"):
|
||||
a1, a2 = fake_data()
|
||||
|
||||
query = generate_query(field, query_str)
|
||||
|
||||
mock_count = mock.Mock(return_value=3)
|
||||
mock_slice = mock.Mock(return_value=mock.Mock(to_queryset=mock.Mock(
|
||||
return_value=Article.objects.filter(pk__in=[a1.id, a2.id])
|
||||
)))
|
||||
mock_query = mock.Mock(return_value=ArticleDocument.search())
|
||||
|
||||
with mock.patch('django_elasticsearch_dsl.search.Search.count', mock_count), \
|
||||
mock.patch('django_elasticsearch_dsl.search.Search.__getitem__', mock_slice), \
|
||||
mock.patch("elasticsearch_dsl.Search.%s" % method_to_mock, mock_query):
|
||||
schema = Schema(query=ESFilterQuery)
|
||||
result = schema.execute(query)
|
||||
|
||||
assert not result.errors
|
||||
|
||||
mock_query.assert_called_with(expected_arguments)
|
||||
|
||||
assert len(result.data[field]["edges"]) == 2
|
||||
assert result.data[field]["edges"][0]["node"]["headline"] == "a1"
|
||||
assert result.data[field]["edges"][1]["node"]["headline"] == "a2"
|
||||
|
||||
|
||||
def test_filter_string():
|
||||
filter_generation(
|
||||
"articlesAsField",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_string_date():
|
||||
filter_generation(
|
||||
"articlesAsField",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_as_field_order_by():
|
||||
filter_generation(
|
||||
"articlesAsField",
|
||||
"headline: \"A text\", sort:{order:desc, field:id}",
|
||||
{'id': {'order': 'desc'}},
|
||||
"sort"
|
||||
)
|
||||
|
||||
|
||||
def test_filter_as_field_order_by_dict():
|
||||
filter_generation(
|
||||
"articlesInMeta",
|
||||
"headline: \"A text\", sort:{order:desc, field:id}",
|
||||
{'es_id': {'order': 'desc'}},
|
||||
"sort"
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_meta():
|
||||
filter_generation(
|
||||
"articlesInMeta",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_meta_dict():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_meta_dict_foreign():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"reporterEamail: \"A mail\"",
|
||||
Bool(must=[Match(reporter__email={'query': 'A mail', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_multi_field():
|
||||
filter_generation(
|
||||
"articlesInMultiField",
|
||||
"contain: \"A text\"",
|
||||
Bool(must=[Bool(should=[
|
||||
Match(headline={'query': 'A text', 'fuzziness': 'auto'}),
|
||||
Match(lang={'query': 'A text', 'fuzziness': 'auto'})
|
||||
])]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_generating_all():
|
||||
filter_generation(
|
||||
"articlesInGenerateAll",
|
||||
"headline: \"A text\", "
|
||||
"pubDate: \"0000-00-00\", "
|
||||
"pubDateTime: \"00:00:00\", "
|
||||
"lang: \"es\", "
|
||||
"importance: 1, ",
|
||||
Bool(must=[
|
||||
Match(headline={'query': 'A text', 'fuzziness': 'auto'}),
|
||||
Match(pub_date={'query': '0000-00-00', 'fuzziness': 'auto'}),
|
||||
Match(pub_date_time={'query': '00:00:00', 'fuzziness': 'auto'}),
|
||||
Match(lang={'query': 'es', 'fuzziness': 'auto'}),
|
||||
Term(importance=1)
|
||||
]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_generating_exclude():
|
||||
query = generate_query("articlesInExcludes", "headline: \"A text\", ")
|
||||
|
||||
schema = Schema(query=ESFilterQuery)
|
||||
result = schema.execute(query)
|
||||
|
||||
assert len(result.errors) > 0
|
||||
|
||||
|
||||
def test_filter_bad_processor():
|
||||
class ArticleFilterBadProcessor(FilterSetES):
|
||||
"""Article Filter for ES"""
|
||||
|
||||
class Meta(object):
|
||||
"""Metaclass data"""
|
||||
index = ArticleDocument
|
||||
includes = {
|
||||
'headline': {
|
||||
'lookup_expressions': ['bad_processor']
|
||||
}
|
||||
}
|
||||
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, filterset_class=ArticleFilterBadProcessor
|
||||
)
|
||||
|
||||
assert "bad_processor" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_field_without_filterset_class():
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode
|
||||
)
|
||||
|
||||
assert "filterset_class" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_field_with_fields():
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, fields=['headline']
|
||||
)
|
||||
|
||||
assert "fields" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_field_with_order_by():
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, order_by=['headline']
|
||||
)
|
||||
|
||||
assert "order_by" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_filterset_without_index():
|
||||
with raises(ValueError) as error_info:
|
||||
class ArticleFilterBadProcessor(FilterSetES):
|
||||
"""Article Filter for ES"""
|
||||
|
||||
class Meta(object):
|
||||
"""Metaclass data"""
|
||||
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, filterset_class=ArticleFilterBadProcessor
|
||||
)
|
||||
|
||||
assert "Index in Meta" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_filterset_without_xcludes():
|
||||
with raises(ValueError) as error_info:
|
||||
class ArticleFilterBadProcessor(FilterSetES):
|
||||
"""Article Filter for ES"""
|
||||
|
||||
class Meta(object):
|
||||
"""Metaclass data"""
|
||||
index = ArticleDocument
|
||||
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, filterset_class=ArticleFilterBadProcessor
|
||||
)
|
||||
|
||||
assert "includes or excludes field in Meta" in str(error_info.value)
|
||||
|
||||
|
||||
def test_processor_term():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineTerm: \"A text\"",
|
||||
Bool(must=[Term(headline='A text')]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_regex():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineRegex: \"A text\"",
|
||||
Bool(must=[Wildcard(headline='A text')]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_phrase():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlinePhrase: \"A text\"",
|
||||
Bool(must=[MatchPhrase(headline={'query': 'A text'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_prefix():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlinePrefix: \"A text\"",
|
||||
Bool(must=[MatchPhrasePrefix(headline={'query': 'A text'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_in():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineIn: [\"A text 1\", \"A text 2\"]",
|
||||
Bool(must=[Terms(headline=['A text 1', 'A text 2'])]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_exits():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineExits: true",
|
||||
Bool(must=[Bool(must=[Exists(field='headline')])]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_lte():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineLte: \"A text\"",
|
||||
Bool(must=Range(headline={'lte': 'A text'})),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_gte():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineGte: \"A text\"",
|
||||
Bool(must=Range(headline={'gte': 'A text'})),
|
||||
)
|
||||
98
graphene_django/elasticsearch/tests/test_filter_fields.py
Normal file
98
graphene_django/elasticsearch/tests/test_filter_fields.py
Normal file
|
|
@ -0,0 +1,98 @@
|
|||
import pytest
|
||||
from py.test import raises
|
||||
|
||||
from graphene_django.elasticsearch.filter.fields import DjangoESFilterConnectionField
|
||||
from graphene_django.elasticsearch.filter.filterset import FilterSetES
|
||||
from graphene_django.filter.tests.test_fields import ArticleNode
|
||||
from graphene_django.elasticsearch.tests.filters import ArticleDocument
|
||||
from graphene_django.utils import DJANGO_FILTER_INSTALLED, DJANGO_ELASTICSEARCH_DSL_INSTALLED
|
||||
|
||||
pytestmark = []
|
||||
|
||||
if not DJANGO_FILTER_INSTALLED or not DJANGO_ELASTICSEARCH_DSL_INSTALLED:
|
||||
pytestmark.append(
|
||||
pytest.mark.skipif(
|
||||
True, reason="django_filters not installed or not compatible"
|
||||
)
|
||||
)
|
||||
|
||||
pytestmark.append(pytest.mark.django_db)
|
||||
|
||||
|
||||
def test_filter_bad_processor():
|
||||
class ArticleFilterBadProcessor(FilterSetES):
|
||||
"""Article Filter for ES"""
|
||||
|
||||
class Meta(object):
|
||||
"""Metaclass data"""
|
||||
index = ArticleDocument
|
||||
includes = {
|
||||
'headline': {
|
||||
'lookup_expressions': ['bad_processor']
|
||||
}
|
||||
}
|
||||
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, filterset_class=ArticleFilterBadProcessor
|
||||
)
|
||||
|
||||
assert "bad_processor" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_field_without_filterset_class():
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode
|
||||
)
|
||||
|
||||
assert "filterset_class" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_field_with_fields():
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, fields=['headline']
|
||||
)
|
||||
|
||||
assert "fields" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_field_with_order_by():
|
||||
with raises(ValueError) as error_info:
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, order_by=['headline']
|
||||
)
|
||||
|
||||
assert "order_by" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_filterset_without_index():
|
||||
with raises(ValueError) as error_info:
|
||||
class ArticleFilterBadProcessor(FilterSetES):
|
||||
"""Article Filter for ES"""
|
||||
|
||||
class Meta(object):
|
||||
"""Metaclass data"""
|
||||
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, filterset_class=ArticleFilterBadProcessor
|
||||
)
|
||||
|
||||
assert "Index in Meta" in str(error_info.value)
|
||||
|
||||
|
||||
def test_filter_filterset_without_xcludes():
|
||||
with raises(ValueError) as error_info:
|
||||
class ArticleFilterBadProcessor(FilterSetES):
|
||||
"""Article Filter for ES"""
|
||||
|
||||
class Meta(object):
|
||||
"""Metaclass data"""
|
||||
index = ArticleDocument
|
||||
|
||||
DjangoESFilterConnectionField(
|
||||
ArticleNode, filterset_class=ArticleFilterBadProcessor
|
||||
)
|
||||
|
||||
assert "includes or excludes field in Meta" in str(error_info.value)
|
||||
114
graphene_django/elasticsearch/tests/test_filter_filters.py
Normal file
114
graphene_django/elasticsearch/tests/test_filter_filters.py
Normal file
|
|
@ -0,0 +1,114 @@
|
|||
import pytest
|
||||
from elasticsearch_dsl.query import Bool, Match, Term
|
||||
from graphene import Schema
|
||||
|
||||
from graphene_django.elasticsearch.tests.commons import filter_generation, generate_query
|
||||
from graphene_django.elasticsearch.tests.filters import ESFilterQuery
|
||||
from graphene_django.utils import DJANGO_FILTER_INSTALLED, DJANGO_ELASTICSEARCH_DSL_INSTALLED
|
||||
|
||||
pytestmark = []
|
||||
|
||||
if not DJANGO_FILTER_INSTALLED or not DJANGO_ELASTICSEARCH_DSL_INSTALLED:
|
||||
pytestmark.append(
|
||||
pytest.mark.skipif(
|
||||
True, reason="django_filters not installed or not compatible"
|
||||
)
|
||||
)
|
||||
|
||||
pytestmark.append(pytest.mark.django_db)
|
||||
|
||||
|
||||
def test_filter_string():
|
||||
filter_generation(
|
||||
"articlesAsField",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_string_date():
|
||||
filter_generation(
|
||||
"articlesAsField",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_as_field_order_by():
|
||||
filter_generation(
|
||||
"articlesAsField",
|
||||
"headline: \"A text\", sort:{order:desc, field:id}",
|
||||
{'id': {'order': 'desc'}},
|
||||
"sort"
|
||||
)
|
||||
|
||||
|
||||
def test_filter_as_field_order_by_dict():
|
||||
filter_generation(
|
||||
"articlesInMeta",
|
||||
"headline: \"A text\", sort:{order:desc, field:id}",
|
||||
{'es_id': {'order': 'desc'}},
|
||||
"sort"
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_meta():
|
||||
filter_generation(
|
||||
"articlesInMeta",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_meta_dict():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headline: \"A text\"",
|
||||
Bool(must=[Match(headline={'query': 'A text', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_meta_dict_foreign():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"reporterEmail: \"A mail\"",
|
||||
Bool(must=[Match(reporter__email={'query': 'A mail', 'fuzziness': 'auto'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_in_multi_field():
|
||||
filter_generation(
|
||||
"articlesInMultiField",
|
||||
"contain: \"A text\"",
|
||||
Bool(must=[Bool(should=[
|
||||
Match(headline={'query': 'A text', 'fuzziness': 'auto'}),
|
||||
Match(lang={'query': 'A text', 'fuzziness': 'auto'})
|
||||
])]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_generating_all():
|
||||
filter_generation(
|
||||
"articlesInGenerateAll",
|
||||
"headline: \"A text\", "
|
||||
"pubDate: \"0000-00-00\", "
|
||||
"pubDateTime: \"00:00:00\", "
|
||||
"lang: \"es\", "
|
||||
"importance: 1, ",
|
||||
Bool(must=[
|
||||
Match(headline={'query': 'A text', 'fuzziness': 'auto'}),
|
||||
Match(pub_date={'query': '0000-00-00', 'fuzziness': 'auto'}),
|
||||
Match(pub_date_time={'query': '00:00:00', 'fuzziness': 'auto'}),
|
||||
Match(lang={'query': 'es', 'fuzziness': 'auto'}),
|
||||
Term(importance=1)
|
||||
]),
|
||||
)
|
||||
|
||||
|
||||
def test_filter_generating_exclude():
|
||||
query = generate_query("articlesInExcludes", "headline: \"A text\", ")
|
||||
|
||||
schema = Schema(query=ESFilterQuery)
|
||||
result = schema.execute(query)
|
||||
|
||||
assert len(result.errors) > 0
|
||||
80
graphene_django/elasticsearch/tests/test_filter_processor.py
Normal file
80
graphene_django/elasticsearch/tests/test_filter_processor.py
Normal file
|
|
@ -0,0 +1,80 @@
|
|||
import pytest
|
||||
from elasticsearch_dsl.query import Bool, Term, Wildcard, MatchPhrase, MatchPhrasePrefix, Range, Terms, Exists
|
||||
|
||||
from graphene_django.elasticsearch.tests.commons import filter_generation
|
||||
from graphene_django.utils import DJANGO_FILTER_INSTALLED, DJANGO_ELASTICSEARCH_DSL_INSTALLED
|
||||
|
||||
pytestmark = []
|
||||
|
||||
if not DJANGO_FILTER_INSTALLED or not DJANGO_ELASTICSEARCH_DSL_INSTALLED:
|
||||
pytestmark.append(
|
||||
pytest.mark.skipif(
|
||||
True, reason="django_filters not installed or not compatible"
|
||||
)
|
||||
)
|
||||
|
||||
pytestmark.append(pytest.mark.django_db)
|
||||
|
||||
|
||||
def test_processor_term():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineTerm: \"A text\"",
|
||||
Bool(must=[Term(headline='A text')]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_regex():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineRegex: \"A text\"",
|
||||
Bool(must=[Wildcard(headline='A text')]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_phrase():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlinePhrase: \"A text\"",
|
||||
Bool(must=[MatchPhrase(headline={'query': 'A text'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_prefix():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlinePrefix: \"A text\"",
|
||||
Bool(must=[MatchPhrasePrefix(headline={'query': 'A text'})]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_in():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineIn: [\"A text 1\", \"A text 2\"]",
|
||||
Bool(must=[Terms(headline=['A text 1', 'A text 2'])]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_exits():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineExits: true",
|
||||
Bool(must=[Bool(must=[Exists(field='headline')])]),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_lte():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineLte: \"A text\"",
|
||||
Bool(must=Range(headline={'lte': 'A text'})),
|
||||
)
|
||||
|
||||
|
||||
def test_processor_gte():
|
||||
filter_generation(
|
||||
"articlesInMetaDict",
|
||||
"headlineGte: \"A text\"",
|
||||
Bool(must=Range(headline={'gte': 'A text'})),
|
||||
)
|
||||
Loading…
Reference in New Issue
Block a user