External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-04-08 03:29:21 -02:30
Code Issues Packages Projects Releases Wiki Activity

flake8 fixes

Browse Source
This commit is contained in:
Chris Meyers
2016-11-18 14:49:31 -05:00
parent 25915eb55a
commit b6b1f4205c
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
awx/main/tests/functional/models/fact/conftest.py
View File

@@ -5,21 +5,24 @@ import six
from jsonbfield.fields import JSONField from jsonbfield.fields import JSONField
from django.core.serializers.json import DjangoJSONEncoder from django.core.serializers.json import DjangoJSONEncoder
def dumps(value): def dumps(value):
return DjangoJSONEncoder().encode(value) return DjangoJSONEncoder().encode(value)
# Taken from https://github.com/django-extensions/django-extensions/blob/54fe88df801d289882a79824be92d823ab7be33e/django_extensions/db/fields/json.py # Taken from https://github.com/django-extensions/django-extensions/blob/54fe88df801d289882a79824be92d823ab7be33e/django_extensions/db/fields/json.py
def get_db_prep_save(self, value, connection, **kwargs): def get_db_prep_save(self, value, connection, **kwargs):
"""Convert our JSON object to a string before we save""" """Convert our JSON object to a string before we save"""
if value is None and self.null: if value is None and self.null:
return None return None
# default values come in as strings; only non-strings should be # default values come in as strings; only non-strings should be
# run through `dumps` # run through `dumps`
if not isinstance(value, six.string_types): if not isinstance(value, six.string_types):
value = dumps(value) value = dumps(value)
return value return value
@pytest.fixture @pytest.fixture
def monkeypatch_jsonbfield_get_db_prep_save(mocker): def monkeypatch_jsonbfield_get_db_prep_save(mocker):
JSONField.get_db_prep_save = get_db_prep_save JSONField.get_db_prep_save = get_db_prep_save