Add custom JSON encoder to handle attachment objects

This commit is contained in:
Patrick Smith
2019-01-23 10:30:30 -05:00
parent f553c9a9ea
commit e51a9012fd
2 changed files with 24 additions and 0 deletions

15
tests/utils/test_json.py Normal file
View File

@@ -0,0 +1,15 @@
import json
from atst.utils.json import CustomJSONEncoder
from tests.factories import AttachmentFactory
encoder = CustomJSONEncoder()
def test_custom_encoder_serializes_attachments():
filename = "jar_jar_is_secretly_a_sith_lord.pdf"
attachment = AttachmentFactory.create(filename=filename)
encoded = encoder.encode({"file": attachment})
expected = json.dumps({"file": filename})
assert encoded == expected