github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

base64 encode bytestrings from DB in JSON

Browse source 
Fixes #29
This commit is contained in:
Simon Willison 2017-10-24 07:58:41 -07:00
commit f643f7aee1
2 changed files with 46 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

19
test_helpers.py
View file

@ -1,6 +1,7 @@
import app
import pytest
import sqlite3
import json
@pytest.mark.parametrize('path,expected', [
@ -45,3 +46,21 @@ def test_pks_for_table(sql, table, expected_keys):
def test_path_from_row_pks(row, pks, expected_path):
actual_path = app.path_from_row_pks(row, pks)
assert expected_path == actual_path
@pytest.mark.parametrize('obj,expected', [
({
'Description': 'Soft drinks',
'Picture': b"\x15\x1c\x02\xc7\xad\x05\xfe",
'CategoryID': 1,
}, """
{"CategoryID": 1, "Description": "Soft drinks", "Picture": {"$base64": true, "encoded": "FRwCx60F/g=="}}
""".strip()),
])
def test_custom_json_encoder(obj, expected):
actual = json.dumps(
obj,
cls=app.CustomJSONEncoder,
sort_keys=True
)
assert expected == actual