Commit dd7ec74c by Timm Behner

Merge pull request #1 in CH/common_helper_encoder from byte_encoder to master

# By Peter Weidenbach
# Via Peter Weidenbach
* commit 'c62fa0d0':
  byte conversion added
  PEP8!
parents 97a969ac c62fa0d0
import json import json
import datetime import datetime
from base64 import standard_b64encode
class ReportEncoder(json.JSONEncoder): class ReportEncoder(json.JSONEncoder):
def default(self, obj): def default(self, obj):
if isinstance(obj, datetime.datetime): if isinstance(obj, datetime.datetime):
return obj.isoformat() return obj.isoformat()
return json.JSONEncoder.default(self,obj) elif isinstance(obj, bytes):
return standard_b64encode(obj).decode('utf-8')
return json.JSONEncoder.default(self, obj)
import unittest
from common_helper_encoder.json_encoder import ReportEncoder
class Test_json_encoder(unittest.TestCase):
def setUp(self):
self.encoder = ReportEncoder()
def test_byte_encoding(self):
input_data = b'TEST'
result = self.encoder.default(input_data)
self.assertEqual(result, "VEVTVA==", "base64 encoding not correct")
if __name__ == "__main__":
unittest.main()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment