mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-16 03:18:09 +00:00
147 lines
4.4 KiB
Python
147 lines
4.4 KiB
Python
# flake8: noqa: F403,F405
|
|
from common import * # isort:skip
|
|
|
|
if not utils.BITCOIN_ONLY:
|
|
from trezor.messages import NEMMosaic
|
|
|
|
from apps.nem.mosaic.helpers import get_mosaic_definition
|
|
from apps.nem.transfer import *
|
|
from apps.nem.transfer.serialize import *
|
|
from apps.nem.transfer.serialize import _merge_mosaics
|
|
|
|
|
|
def get_mosaic(namespace: str, quantity: int, mosaic: str) -> NEMMosaic:
|
|
return NEMMosaic(
|
|
namespace=namespace,
|
|
quantity=quantity,
|
|
mosaic=mosaic,
|
|
)
|
|
|
|
|
|
# NOTE: copy-pasted from apps.nem.transfer.serialize.py
|
|
def sort_mosaics(mosaics: list[NEMMosaic]) -> list[NEMMosaic]:
|
|
return sorted(mosaics, key=lambda m: (m.namespace, m.mosaic))
|
|
|
|
|
|
@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
|
|
class TestNemMosaic(unittest.TestCase):
|
|
def test_get_mosaic_definition(self):
|
|
m = get_mosaic_definition("nem", "xem", 104)
|
|
self.assertEqual(m.name, "NEM")
|
|
self.assertEqual(m.ticker, " XEM")
|
|
|
|
m = get_mosaic_definition("nem", "xxx", 104)
|
|
self.assertEqual(m, None)
|
|
|
|
m = get_mosaic_definition("aaaa", "xxx", 104)
|
|
self.assertEqual(m, None)
|
|
|
|
m = get_mosaic_definition("pacnem", "cheese", 104)
|
|
self.assertEqual(m.name, "PacNEM Score Tokens")
|
|
self.assertEqual(m.ticker, " PAC:CHS")
|
|
self.assertEqual(m.levy.fee, 100)
|
|
|
|
def test_mosaic_canonicalization(self):
|
|
a = get_mosaic("abc", 3, "mosaic")
|
|
b = get_mosaic("abc", 4, "a")
|
|
c = get_mosaic("zzz", 3, "mosaic")
|
|
d = get_mosaic("abc", 8, "mosaic")
|
|
e = get_mosaic("aaa", 1, "mosaic")
|
|
f = get_mosaic("aaa", 1, "mosaicz")
|
|
g = get_mosaic("zzz", 30, "mosaic")
|
|
|
|
res = canonicalize_mosaics([a, b, c, d, e, f, g])
|
|
self.assertEqual(res, [e, f, b, a, c])
|
|
self.assertEqual(res[2].quantity, b.quantity)
|
|
self.assertEqual(res[3].quantity, 3 + 8) # a + d
|
|
self.assertEqual(res[4].quantity, 3 + 30) # c + g
|
|
|
|
def test_mosaic_merge(self):
|
|
a = get_mosaic("abc", 1, "mosaic")
|
|
b = get_mosaic("abc", 1, "mosaic")
|
|
c = get_mosaic("abc", 2, "xxx")
|
|
|
|
merged = _merge_mosaics([a, b])
|
|
self.assertEqual(merged[0].quantity, 2)
|
|
self.assertEqual(len(merged), 1)
|
|
|
|
a.quantity = 1
|
|
b.quantity = 10
|
|
merged = _merge_mosaics([a, b])
|
|
self.assertEqual(merged[0].quantity, 11)
|
|
|
|
a.namespace = "abcdef"
|
|
merged = _merge_mosaics([a, b])
|
|
self.assertEqual(len(merged), 2)
|
|
|
|
merged = _merge_mosaics([a, b, c])
|
|
self.assertEqual(len(merged), 3)
|
|
|
|
a.namespace = "abcdef"
|
|
a.quantity = 1
|
|
a.mosaic = "mosaic"
|
|
b.namespace = "abc"
|
|
b.quantity = 2
|
|
b.mosaic = "mosaic"
|
|
c.namespace = "abc"
|
|
c.mosaic = "mosaic"
|
|
c.quantity = 3
|
|
merged = _merge_mosaics([a, b, c])
|
|
self.assertEqual(merged[0].quantity, 1)
|
|
self.assertEqual(merged[1].quantity, 5)
|
|
self.assertEqual(len(merged), 2)
|
|
|
|
a.namespace = "abc"
|
|
a.quantity = 1
|
|
a.mosaic = "mosaic"
|
|
b.namespace = "abc"
|
|
b.quantity = 2
|
|
b.mosaic = "mosaic"
|
|
c.namespace = "abc"
|
|
c.mosaic = "mosaic"
|
|
c.quantity = 3
|
|
merged = _merge_mosaics([a, b, c])
|
|
self.assertEqual(merged[0].quantity, 6)
|
|
self.assertEqual(len(merged), 1)
|
|
|
|
def test_mosaic_sort(self):
|
|
a = get_mosaic("abcz", 1, "mosaic")
|
|
b = get_mosaic("abca", 1, "mosaic")
|
|
c = get_mosaic("a", 0, "zzz")
|
|
|
|
res = sort_mosaics([a, b])
|
|
self.assertListEqual(res, [b, a])
|
|
|
|
a.namespace = ""
|
|
b.namespace = "a.b.c"
|
|
res = sort_mosaics([a, b])
|
|
self.assertListEqual(res, [a, b])
|
|
|
|
a.namespace = "z.z.z"
|
|
b.namespace = "a.b.c"
|
|
res = sort_mosaics([a, b])
|
|
self.assertListEqual(res, [b, a])
|
|
|
|
a.namespace = "a"
|
|
b.namespace = "a"
|
|
a.mosaic = "mosaic"
|
|
b.mosaic = "mosaic"
|
|
res = sort_mosaics([a, b])
|
|
self.assertListEqual(res, [a, b])
|
|
|
|
a.mosaic = "www"
|
|
b.mosaic = "aaa"
|
|
res = sort_mosaics([a, b])
|
|
self.assertListEqual(res, [b, a])
|
|
|
|
res = sort_mosaics([a, b, c])
|
|
self.assertListEqual(res, [b, a, c])
|
|
|
|
c.mosaic = "bbb"
|
|
res = sort_mosaics([a, b, c])
|
|
self.assertListEqual(res, [b, c, a])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|