From b13dba115a05d0c671edd84e3f155d78021da40f Mon Sep 17 00:00:00 2001
From: Jan Pochyla <jpochyla@gmail.com>
Date: Thu, 7 Jun 2018 13:18:45 +0200
Subject: [PATCH] nem: fix unit tests

---
 tests/test_apps.nem.mosaic.py                          | 4 +++-
 tests/test_apps.nem.mosaic_creation.py                 | 3 +++
 tests/test_apps.nem.mosaic_supply_change.py            | 3 +++
 tests/test_apps.nem.multisig.aggregate_modification.py | 3 +++
 tests/test_apps.nem.multisig.py                        | 4 ++++
 tests/test_apps.nem.namespace.py                       | 3 +++
 tests/test_apps.nem.transfer.py                        | 5 ++++-
 7 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/tests/test_apps.nem.mosaic.py b/tests/test_apps.nem.mosaic.py
index f17d9a046d..7cee25d3e1 100644
--- a/tests/test_apps.nem.mosaic.py
+++ b/tests/test_apps.nem.mosaic.py
@@ -1,6 +1,8 @@
 from common import *
-from apps.nem.transfer import *
+from trezor.messages.NEMMosaic import NEMMosaic
 from apps.nem.mosaic.helpers import get_mosaic_definition
+from apps.nem.transfer import *
+from apps.nem.transfer.serialize import *
 
 
 class TestNemMosaic(unittest.TestCase):
diff --git a/tests/test_apps.nem.mosaic_creation.py b/tests/test_apps.nem.mosaic_creation.py
index 2c445673bb..6c387bfab4 100644
--- a/tests/test_apps.nem.mosaic_creation.py
+++ b/tests/test_apps.nem.mosaic_creation.py
@@ -1,6 +1,9 @@
 from common import *
 
+from apps.nem.helpers import *
 from apps.nem.mosaic import *
+from apps.nem.mosaic.serialize import *
+
 from trezor.crypto import hashlib
 from trezor.messages.NEMSignTx import NEMSignTx
 from trezor.messages.NEMMosaicCreation import NEMMosaicCreation
diff --git a/tests/test_apps.nem.mosaic_supply_change.py b/tests/test_apps.nem.mosaic_supply_change.py
index 834ed941ee..424b0a67e9 100644
--- a/tests/test_apps.nem.mosaic_supply_change.py
+++ b/tests/test_apps.nem.mosaic_supply_change.py
@@ -1,6 +1,9 @@
 from common import *
 
+from apps.nem.helpers import *
 from apps.nem.mosaic import *
+from apps.nem.mosaic.serialize import *
+
 from trezor.crypto import hashlib
 from trezor.messages.NEMSignTx import NEMSignTx
 from trezor.messages.NEMMosaicSupplyChange import NEMMosaicSupplyChange
diff --git a/tests/test_apps.nem.multisig.aggregate_modification.py b/tests/test_apps.nem.multisig.aggregate_modification.py
index 81258f9b22..3b145d3a9d 100644
--- a/tests/test_apps.nem.multisig.aggregate_modification.py
+++ b/tests/test_apps.nem.multisig.aggregate_modification.py
@@ -1,6 +1,9 @@
 from common import *
 
+from apps.nem.helpers import *
 from apps.nem.multisig import *
+from apps.nem.multisig.serialize import *
+
 from trezor.crypto import hashlib
 from trezor.messages.NEMSignTx import NEMSignTx
 from trezor.messages.NEMAggregateModification import NEMAggregateModification
diff --git a/tests/test_apps.nem.multisig.py b/tests/test_apps.nem.multisig.py
index 5980344388..df85f3b7ab 100644
--- a/tests/test_apps.nem.multisig.py
+++ b/tests/test_apps.nem.multisig.py
@@ -1,7 +1,11 @@
 from common import *
 
+from apps.nem.helpers import *
 from apps.nem.multisig import *
+from apps.nem.multisig.serialize import *
 from apps.nem.namespace import *
+from apps.nem.namespace.serialize import *
+
 from trezor.messages.NEMSignTx import NEMSignTx
 from trezor.messages.NEMAggregateModification import NEMAggregateModification
 from trezor.messages.NEMProvisionNamespace import NEMProvisionNamespace
diff --git a/tests/test_apps.nem.namespace.py b/tests/test_apps.nem.namespace.py
index 7165158b42..c9375800d9 100644
--- a/tests/test_apps.nem.namespace.py
+++ b/tests/test_apps.nem.namespace.py
@@ -1,6 +1,9 @@
 from common import *
 
+from apps.nem.helpers import *
 from apps.nem.namespace import *
+from apps.nem.namespace.serialize import *
+
 from trezor.crypto import hashlib
 from trezor.messages.NEMProvisionNamespace import NEMProvisionNamespace
 from trezor.messages.NEMSignTx import NEMSignTx
diff --git a/tests/test_apps.nem.transfer.py b/tests/test_apps.nem.transfer.py
index 0ffa04467f..356dc27ba8 100644
--- a/tests/test_apps.nem.transfer.py
+++ b/tests/test_apps.nem.transfer.py
@@ -1,7 +1,10 @@
 from common import *
 
-from apps.nem.transfer import *
+from apps.nem.helpers import *
 from apps.nem.mosaic import *
+from apps.nem.transfer import *
+from apps.nem.transfer.serialize import *
+
 from trezor.crypto import hashlib
 from trezor.messages.NEMTransfer import NEMTransfer
 from trezor.messages.NEMSignTx import NEMSignTx