40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import unittest
|
|
import json
|
|
from sources.result import Metadata, DataResult
|
|
|
|
class SerializationTest(unittest.TestCase):
|
|
def test_metadata_to_json(self):
|
|
metadata = Metadata(
|
|
title='Sample Title',
|
|
altTitle='Alternate Title',
|
|
cover='path/to/cover.jpg',
|
|
type='Movie',
|
|
summary='Lorem ipsum dolor sit amet',
|
|
genres=['Action', 'Drama', 'Thriller']
|
|
)
|
|
|
|
metadata_json = json.dumps(metadata.to_dict())
|
|
self.assertIsInstance(metadata_json, str)
|
|
|
|
def test_data_result_to_json(self):
|
|
metadata = Metadata(
|
|
title='Sample Title',
|
|
altTitle='Alternate Title',
|
|
cover='path/to/cover.jpg',
|
|
type='Movie',
|
|
summary='Lorem ipsum dolor sit amet',
|
|
genres=['Action', 'Drama', 'Thriller']
|
|
)
|
|
|
|
data_result = DataResult(
|
|
statusType='SUCCESS',
|
|
errorMessage=None,
|
|
data=metadata
|
|
)
|
|
|
|
data_result_json = json.dumps(data_result.to_dict())
|
|
self.assertIsInstance(data_result_json, str)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|