encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-28 17:09:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix test errors if psycopg2 is not installed

Browse Source
This commit is contained in:
Alexei Ardyakov 2019-12-14 15:11:29 +03:00
parent 483cbd4568
commit 44bf6d2461
1 changed files with 26 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
tests/test_filters.py
View File

@ -369,23 +369,21 @@ class SearchFilterAnnotatedFieldTests(TestCase):
assert response.data[0]['title_text'] == 'ABCDEF' assert response.data[0]['title_text'] == 'ABCDEF'
class SearchFilterModelPostgres(models.Model): if postgres_fields:
json_field = postgres_fields.JSONField() class SearchFilterModelPostgres(models.Model):
json_field = postgres_fields.JSONField()
class SearchFilterPostgresSerializer(serializers.ModelSerializer):
class Meta:
model = SearchFilterModelPostgres
fields = '__all__'
class SearchFilterPostgresSerializer(serializers.ModelSerializer): @pytest.mark.skipif(
class Meta: 'connection.vendor != "postgresql"',
model = SearchFilterModelPostgres reason='not a PostgreSQL database'
fields = '__all__' )
@pytest.mark.skipif('connection.vendor != "postgresql"')
class SearchFilterPostgresTests(TestCase): class SearchFilterPostgresTests(TestCase):
class SearchListView(generics.ListAPIView):
queryset = SearchFilterModelPostgres.objects.all()
serializer_class = SearchFilterPostgresSerializer
filter_backends = (filters.SearchFilter,)
search_fields = ('=json_field__title', 'json_field__text')
def setUp(self): def setUp(self):
for idx in range(10): for idx in range(10):
@ -400,7 +398,13 @@ class SearchFilterPostgresTests(TestCase):
).save() ).save()
def test_default_json_search(self): def test_default_json_search(self):
view = self.SearchListView.as_view() class SearchListView(generics.ListAPIView):
queryset = SearchFilterModelPostgres.objects.all()
serializer_class = SearchFilterPostgresSerializer
filter_backends = (filters.SearchFilter,)
search_fields = ('=json_field__title', 'json_field__text')
view = SearchListView.as_view()
request = factory.get('/', {'search': 'b'}) request = factory.get('/', {'search': 'b'})
response = view(request) response = view(request)
@ -419,7 +423,13 @@ class SearchFilterPostgresTests(TestCase):
assert items == {('zzzz', 'def'), ('zzzzz', 'efg')} assert items == {('zzzz', 'def'), ('zzzzz', 'efg')}
def test_exact_json_search(self): def test_exact_json_search(self):
view = self.SearchListView.as_view() class SearchListView(generics.ListAPIView):
queryset = SearchFilterModelPostgres.objects.all()
serializer_class = SearchFilterPostgresSerializer
filter_backends = (filters.SearchFilter,)
search_fields = ('=json_field__title', 'json_field__text')
view = SearchListView.as_view()
request = factory.get('/', {'search': 'zzzzz'}) request = factory.get('/', {'search': 'zzzzz'})
response = view(request) response = view(request)