1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 22:18:16 +00:00
trezor-firmware/core/tests/test_apps.nem.mosaic.py

146 lines
4.3 KiB
Python
Raw Normal View History

from common import * # isort:skip
2018-03-29 12:16:58 +00:00
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
2018-03-29 12:16:58 +00:00
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)
2018-03-29 12:16:58 +00:00
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")
2018-03-29 12:16:58 +00:00
res = canonicalize_mosaics([a, b, c, d, e, f, g])
2018-03-29 12:16:58 +00:00
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")
2018-03-29 12:16:58 +00:00
merged = _merge_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(merged[0].quantity, 2)
self.assertEqual(len(merged), 1)
a.quantity = 1
b.quantity = 10
merged = _merge_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(merged[0].quantity, 11)
a.namespace = "abcdef"
merged = _merge_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(len(merged), 2)
merged = _merge_mosaics([a, b, c])
2018-03-29 12:16:58 +00:00
self.assertEqual(len(merged), 3)
a.namespace = "abcdef"
2018-03-29 12:16:58 +00:00
a.quantity = 1
a.mosaic = "mosaic"
b.namespace = "abc"
2018-03-29 12:16:58 +00:00
b.quantity = 2
b.mosaic = "mosaic"
c.namespace = "abc"
c.mosaic = "mosaic"
2018-03-29 12:16:58 +00:00
c.quantity = 3
merged = _merge_mosaics([a, b, c])
2018-03-29 12:16:58 +00:00
self.assertEqual(merged[0].quantity, 1)
self.assertEqual(merged[1].quantity, 5)
self.assertEqual(len(merged), 2)
a.namespace = "abc"
2018-03-29 12:16:58 +00:00
a.quantity = 1
a.mosaic = "mosaic"
b.namespace = "abc"
2018-03-29 12:16:58 +00:00
b.quantity = 2
b.mosaic = "mosaic"
c.namespace = "abc"
c.mosaic = "mosaic"
2018-03-29 12:16:58 +00:00
c.quantity = 3
merged = _merge_mosaics([a, b, c])
2018-03-29 12:16:58 +00:00
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])
2018-03-29 12:16:58 +00:00
a.namespace = ""
b.namespace = "a.b.c"
res = sort_mosaics([a, b])
self.assertListEqual(res, [a, b])
2018-03-29 12:16:58 +00:00
a.namespace = "z.z.z"
b.namespace = "a.b.c"
res = sort_mosaics([a, b])
self.assertListEqual(res, [b, a])
2018-03-29 12:16:58 +00:00
a.namespace = "a"
b.namespace = "a"
a.mosaic = "mosaic"
b.mosaic = "mosaic"
res = sort_mosaics([a, b])
self.assertListEqual(res, [a, b])
2018-03-29 12:16:58 +00:00
a.mosaic = "www"
b.mosaic = "aaa"
res = sort_mosaics([a, b])
self.assertListEqual(res, [b, a])
2018-03-29 12:16:58 +00:00
res = sort_mosaics([a, b, c])
self.assertListEqual(res, [b, a, c])
2018-03-29 12:16:58 +00:00
c.mosaic = "bbb"
res = sort_mosaics([a, b, c])
self.assertListEqual(res, [b, c, a])
2018-03-29 12:16:58 +00:00
if __name__ == "__main__":
2018-03-29 12:16:58 +00:00
unittest.main()