1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-30 03:18:20 +00:00
trezor-firmware/core/tests/test_apps.nem.mosaic.py

166 lines
4.5 KiB
Python
Raw Normal View History

2018-03-29 12:16:58 +00:00
from common import *
2018-06-07 11:18:45 +00:00
from trezor.messages.NEMMosaic import NEMMosaic
from apps.nem.mosaic.helpers import get_mosaic_definition
2018-06-07 11:18:45 +00:00
from apps.nem.transfer import *
from apps.nem.transfer.serialize import *
2018-03-29 12:16:58 +00:00
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["fee"], 100)
2018-03-29 12:16:58 +00:00
def test_mosaic_canonicalization(self):
2018-03-29 12:16:58 +00:00
a = NEMMosaic()
a.namespace = 'abc'
a.quantity = 3
a.mosaic = 'mosaic'
b = NEMMosaic()
b.namespace = 'abc'
b.quantity = 4
b.mosaic = 'a'
c = NEMMosaic()
c.namespace = 'zzz'
c.quantity = 3
c.mosaic = 'mosaic'
d = NEMMosaic()
d.namespace = 'abc'
d.quantity = 8
d.mosaic = 'mosaic'
e = NEMMosaic()
e.namespace = 'aaa'
e.quantity = 1
e.mosaic = 'mosaic'
f = NEMMosaic()
f.namespace = 'aaa'
f.quantity = 1
f.mosaic = 'mosaicz'
g = NEMMosaic()
g.namespace = 'zzz'
g.quantity = 30
g.mosaic = 'mosaic'
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):
2018-03-29 12:16:58 +00:00
a = NEMMosaic()
a.namespace = 'abc'
a.quantity = 1
a.mosaic = 'mosaic'
b = NEMMosaic()
b.namespace = 'abc'
b.quantity = 1
b.mosaic = 'mosaic'
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)
c = NEMMosaic()
c.namespace = 'abc'
c.mosaic = 'xxx'
c.quantity = 2
merged = merge_mosaics([a, b, c])
2018-03-29 12:16:58 +00:00
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])
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'
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])
2018-03-29 12:16:58 +00:00
self.assertEqual(merged[0].quantity, 6)
self.assertEqual(len(merged), 1)
def test_mosaic_sort(self):
2018-03-29 12:16:58 +00:00
a = NEMMosaic()
a.namespace = 'abcz'
a.quantity = 1
a.mosaic = 'mosaic'
b = NEMMosaic()
b.namespace = 'abca'
b.quantity = 1
b.mosaic = 'mosaic'
res = sort_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [b, a])
a.namespace = ''
b.namespace = 'a.b.c'
res = sort_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [a, b])
a.namespace = 'z.z.z'
b.namespace = 'a.b.c'
res = sort_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [b, a])
a.namespace = 'a'
b.namespace = 'a'
a.mosaic = 'mosaic'
b.mosaic = 'mosaic'
res = sort_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [a, b])
a.mosaic = 'www'
b.mosaic = 'aaa'
res = sort_mosaics([a, b])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [b, a])
c = NEMMosaic()
c.namespace = 'a'
c.mosaic = 'zzz'
res = sort_mosaics([a, b, c])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [b, a, c])
c.mosaic = 'bbb'
res = sort_mosaics([a, b, c])
2018-03-29 12:16:58 +00:00
self.assertEqual(res, [b, c, a])
if __name__ == '__main__':
unittest.main()