def test_equal_false(self):
instance = UpdateMessage(self.subject,
[Session('test', [Amount(1000000000)])])
other = UpdateMessage(self.subject,
[Session('test', [Amount(1100000000)])])
self.assertFalse(instance.equal(other))
def test_from_sessions_duplicated(self):
self.assertEqual(UpdateMessage.objects.count(), 0)
UpdateMessage.from_data(self.subject, DATA)
self.assertEqual(UpdateMessage.objects.count(), 1)
self.assertRaises(DuplicatedUpdateMessageError,
UpdateMessage.from_data, self.subject, DATA)
self.assertEqual(UpdateMessage.objects.count(), 1)
def test_get_last_by_subject(self):
UpdateMessage(self.subject, [Session('test', [Amount(1000000000)])],
pendulum.yesterday()).save()
other = UpdateMessage(self.subject,
[Session('test', [Amount(1100000000)])],
pendulum.today()).save()
last = UpdateMessage.get_last_by_subject(self.subject)
self.assertEqual(other, last)
def test_str(self, mocked_get_last):
ticket_price = TicketPrice(10)
mocked_get_last.return_value = ticket_price
instance = UpdateMessage(
self.subject, [Session('test', [Amount(1000000000)])]).save()
self.assertEqual(
instance.__str__(), f"<b>🇧🇷 Decred Brasil</b>\n\n"
f"Ticket price: 10.00 DCR\n\n"
f"<i>Default session: dcrbr1</i>\n\n"
f"<code>test:\t[10.0 DCR]\nTotal: 10.0 DCR</code>")
def test_from_sessions(self):
UpdateMessage.from_data(self.subject, DATA)
instance = UpdateMessage.objects.first()
self.assertEqual(instance.subject, self.subject)
self.assertEqual(len(instance.sessions), 1)
session = instance.sessions[0]
self.assertIsInstance(session, Session)
self.assertEqual(
session.hash, 'c17b1828e97bf66abd5329e737551'
'73b43b98e18ebd4b84b19a016781d8cfa86')
self.assertEqual(len(session.amounts), 2)
self.assertIsInstance(session.amounts[0], Amount)
self.assertIsInstance(session.amounts[1], Amount)
self.assertEqual(session.amounts[0].value, 10.0)
self.assertEqual(session.amounts[1].value, 20.0)
def test_init(self):
self.assertEqual(UpdateMessage.objects.count(), 0)
UpdateMessage(self.subject,
[Session('test', [Amount(1000000000)])]).save()
self.assertEqual(UpdateMessage.objects.count(), 1)
instance = UpdateMessage.objects.first()
self.assertEqual(instance.subject, self.subject)
self.assertIsInstance(instance, UpdateMessage)
def on_message(ws: WebSocketApp, data):
sws: SessionWebSocket = SessionWebSocket.get_sws(ws.url)
try:
msg = UpdateMessage.from_data(sws.subject, data)
logger.info(f'SessionUpdateMessage received from {sws.name}')
logger.info(f'{sws.name} trying to acquire lock...')
sws.lock.acquire()
logger.info(f'{sws.name} got the lock!')
sws.subject.notify(msg)
sws.lock.release()
logger.info(f'{sws.name} released lock!')
except DuplicatedUpdateMessageError as e:
logger.info(f"Supress {e} for creating {UpdateMessage} "
f"from {data} on {sws}")
except Exception as e:
logger.error(e)
finally:
if sws.lock.locked():
sws.lock.release()