Updated pyMetadata

This commit is contained in:
Brage 2023-07-18 03:50:00 +02:00
parent 4920d24415
commit eb426f46ca
3 changed files with 42 additions and 0 deletions

0
pyMetadata/__init__.py Normal file
View File

View File

@ -10,6 +10,9 @@ class Metadata:
summary: str summary: str
genres: List[str] genres: List[str]
def to_dict(self):
return asdict(self)
@dataclass @dataclass
class DataResult: class DataResult:
statusType: str statusType: str

View File

@ -0,0 +1,39 @@
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()