mirror of
https://github.com/aljazceru/bitfinex-api-py.git
synced 2025-12-19 23:04:21 +01:00
25 lines
948 B
Python
25 lines
948 B
Python
import unittest
|
|
|
|
from dataclasses import dataclass
|
|
from ..labeler import generate_labeler_serializer
|
|
from ..notification import _Type, _Notification, Notification
|
|
|
|
class TestNotification(unittest.TestCase):
|
|
def test_notification(self):
|
|
@dataclass
|
|
class Test(_Type):
|
|
A: int
|
|
B: float
|
|
C: str
|
|
|
|
test = generate_labeler_serializer("Test", Test,
|
|
[ "A", "_PLACEHOLDER", "B", "_PLACEHOLDER", "C" ])
|
|
|
|
notification = _Notification[Test](test)
|
|
|
|
self.assertEqual(notification.parse(*[1675787861506, "test", None, None, [ 5, None, 65.0, None, "X" ], 0, "SUCCESS", "This is just a test notification."]),
|
|
Notification[Test](1675787861506, "test", None, Test(5, 65.0, "X"), 0, "SUCCESS", "This is just a test notification."),
|
|
msg="_Notification should produce the right notification.")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |