tests: move to /tests to avoid freezing

Makefile

@@ -59,7 +59,7 @@ clean_cross: ## clean mpy-cross build
 	$(MAKE) -C vendor/micropython/mpy-cross clean
 
 test: ## run unit tests
-	cd src/tests ; ./run_tests.sh
+	cd tests ; ./run_tests.sh
 
 flash: ## flash firmware using st-flash
 	st-flash write $(STMHAL_BUILD_DIR)/firmware0.bin 0x8000000

src/trezor/config.py

@@ -12,6 +12,7 @@ else:
 
 
 def _load():
+    global _mock
     try:
         with open(_file, 'rb') as f:
             while True:
@@ -26,6 +27,7 @@ def _load():
 
 
 def _save():
+    global _mock
     with open(_file, 'wb') as f:
         for k, v in _mock.items():
             f.write(ustruct.pack('<HH', k, len(v)))
@@ -35,13 +37,16 @@ _load()
 
 
 def get(app_id, key, default=None):
+    global _mock
     return _mock.get((app_id << 8) | key, default)
 
 
 def set(app_id, key, value):
+    global _mock
     _mock[(app_id << 8) | key] = value
     _save()
 
 def wipe():
+    global _mock
     _mock = {}
     _save()

tests/check_coverage.sh

@@ -1,6 +1,6 @@
 #!/bin/bash
-find ../../mocks -name '*.py' | sort | while read module; do
-    module=$(echo $module | sed 's:^\.\./\.\./mocks/::')
+find ../mocks -name '*.py' | sort | while read module; do
+    module=$(echo $module | sed 's:^\.\./mocks/::')
     base=$(basename $module)
     # skip __init__.py
     if [[ $base == "__init__.py" ]]; then

tests/common.py

@@ -0,0 +1,8 @@
+import sys
+
+sys.path.append('../src')
+sys.path.append('../src/lib')
+
+import unittest
+
+from ubinascii import hexlify, unhexlify

tests/run_tests.sh

@@ -8,7 +8,7 @@ else
 fi
 for i in $list; do
    echo
-    if ../../vendor/micropython/unix/micropython $i; then
+    if ../vendor/micropython/unix/micropython $i; then
        results+=("OK   $i")
     else
        results+=("FAIL $i")

tests/test_trezor.config.py

@@ -1,11 +1,9 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
+from common import *
 
-from trezor import config
 from trezor.crypto import random
 
+from trezor import config
+
 class TestConfig(unittest.TestCase):
 
     def test_set_get(self):

tests/test_trezor.crypto.aes.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto.aes import *
 

tests/test_trezor.crypto.base58.py

@@ -1,12 +1,9 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
-from trezor.crypto import base58
 from trezor.crypto.hashlib import ripemd160
 
+from trezor.crypto import base58
+
 digestfunc_graphene = lambda x: ripemd160(x).digest()[:4]
 
 class TestCryptoBase58(unittest.TestCase):

tests/test_trezor.crypto.bip32.py

@@ -1,14 +1,9 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import bip32
 
 SECP256K1_NAME = 'secp256k1'
 
-
 class TestCryptoBip32(unittest.TestCase):
 
     def test_from_seed_invalid(self):

tests/test_trezor.crypto.bip39.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import bip39
 

tests/test_trezor.crypto.curve.curve25519.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto.curve import curve25519
 

tests/test_trezor.crypto.curve.ed25519.py

@@ -1,12 +1,9 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
-from trezor.crypto.curve import ed25519
 from trezor.crypto import random
 
+from trezor.crypto.curve import ed25519
+
 class TestCryptoEd25519(unittest.TestCase):
 
     # vectors from https://github.com/torproject/tor/blob/master/src/test/ed25519_vectors.inc

tests/test_trezor.crypto.curve.nist256p1.py

@@ -1,12 +1,9 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import hexlify, unhexlify
+from common import *
 
-from trezor.crypto.curve import nist256p1
 from trezor.crypto import random
 
+from trezor.crypto.curve import nist256p1
+
 class TestCryptoNist256p1(unittest.TestCase):
 
     # vectors from http://point-at-infinity.org/ecc/nisttv

tests/test_trezor.crypto.curve.secp256k1.py

@@ -1,12 +1,9 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import hexlify, unhexlify
+from common import *
 
-from trezor.crypto.curve import secp256k1
 from trezor.crypto import random
 
+from trezor.crypto.curve import secp256k1
+
 class TestCryptoSecp256k1(unittest.TestCase):
 
     # vectors from https://crypto.stackexchange.com/questions/784/are-there-any-secp256k1-ecdsa-test-examples-available

tests/test_trezor.crypto.hashlib.blake2s.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hashlib.ripemd160.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hashlib.sha1.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hashlib.sha256.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hashlib.sha3_256.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hashlib.sha3_512.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hashlib.sha512.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
 

tests/test_trezor.crypto.hmac.py

@@ -1,10 +1,7 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import hashlib
+
 from trezor.crypto import hmac
 
 class TestCryptoHmac(unittest.TestCase):

tests/test_trezor.crypto.pbkdf2.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import unhexlify
+from common import *
 
 from trezor.crypto import pbkdf2
 

tests/test_trezor.crypto.random.py

@@ -1,8 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
-from ubinascii import hexlify
+from common import *
 
 from trezor.crypto import random
 

tests/test_trezor.debug.py

@@ -1,7 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
+from common import *
 
 from trezor import debug
 

tests/test_trezor.utils.py

@@ -1,7 +1,4 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
+from common import *
 
 from trezor import utils
 

tests/test_trezor.wire.wire_codec.py

@@ -1,14 +1,12 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
+from common import *
+
 import ustruct
 import ubinascii
 
-from trezor.wire import wire_codec
-from trezor.utils import chunks
 from trezor.crypto import random
+from trezor.utils import chunks
 
+from trezor.wire import wire_codec
 
 class TestWireCodec(unittest.TestCase):
     # pylint: disable=C0301

tests/test_trezor.wire.wire_codec_v1.py

@@ -1,13 +1,12 @@
-import sys
-sys.path.append('..')
-sys.path.append('../lib')
-import unittest
+from common import *
+
 import ustruct
+import ubinascii
 
-from trezor.wire import wire_codec_v1
-from trezor.utils import chunks
 from trezor.crypto import random
+from trezor.utils import chunks
 
+from trezor.wire import wire_codec_v1
 
 class TestWireCodecV1(unittest.TestCase):
     # pylint: disable=C0301