mirror of
https://github.com/logos-blockchain/logos-blockchain-simulations.git
synced 2026-01-02 13:13:07 +00:00
24 lines
725 B
Python
24 lines
725 B
Python
import time
|
|
from unittest import TestCase
|
|
|
|
from sim.message import Message, UniqueMessageBuilder
|
|
|
|
|
|
class TestMessage(TestCase):
|
|
def test_message_serde(self):
|
|
msg = Message(time.time(), 10, b"hello")
|
|
serialized = bytes(msg)
|
|
deserialized = Message.from_bytes(serialized)
|
|
self.assertEqual(msg, deserialized)
|
|
|
|
|
|
class TestUniqueMessageBuilder(TestCase):
|
|
def test_uniqueness(self):
|
|
builder = UniqueMessageBuilder()
|
|
msg1 = builder.next(time.time(), b"hello")
|
|
msg2 = builder.next(time.time(), b"hello")
|
|
self.assertEqual(0, msg1.id)
|
|
self.assertEqual(1, msg2.id)
|
|
self.assertNotEqual(msg1, msg2)
|
|
self.assertNotEqual(hash(msg1), hash(msg2))
|