diff --git a/src/pyelliptic/tests/__init__.py b/src/pyelliptic/tests/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/src/pyelliptic/tests/test_arithmetic.py b/src/pyelliptic/tests/test_arithmetic.py new file mode 100644 index 00000000..7b5c59b1 --- /dev/null +++ b/src/pyelliptic/tests/test_arithmetic.py @@ -0,0 +1,84 @@ +""" +Test the arithmetic functions +""" + +from binascii import unhexlify +import unittest + +try: + from pyelliptic import arithmetic +except ImportError: + from pybitmessage.pyelliptic import arithmetic + + +# These keys are from addresses test script +sample_pubsigningkey = ( + b'044a367f049ec16cb6b6118eb734a9962d10b8db59c890cd08f210c43ff08bdf09d' + b'16f502ca26cd0713f38988a1237f1fc8fa07b15653c996dc4013af6d15505ce') +sample_pubencryptionkey = ( + b'044597d59177fc1d89555d38915f581b5ff2286b39d022ca0283d2bdd5c36be5d3c' + b'e7b9b97792327851a562752e4b79475d1f51f5a71352482b241227f45ed36a9') +sample_privsigningkey = \ + b'93d0b61371a54b53df143b954035d612f8efa8a3ed1cf842c2186bfd8f876665' +sample_privencryptionkey = \ + b'4b0b73a54e19b059dc274ab69df095fe699f43b17397bca26fdf40f4d7400a3a' + +sample_factor = \ + 66858749573256452658262553961707680376751171096153613379801854825275240965733 +# G * sample_factor +sample_point = ( + 33567437183004486938355437500683826356288335339807546987348409590129959362313, + 94730058721143827257669456336351159718085716196507891067256111928318063085006 +) + + +class TestArithmetic(unittest.TestCase): + """Test arithmetic functions""" + def test_base10_multiply(self): + """Test arithmetic.base10_multiply""" + self.assertEqual( + sample_point, + arithmetic.base10_multiply(arithmetic.G, sample_factor)) + + def test_decode(self): + """Decode sample privsigningkey from hex to int and compare to factor""" + self.assertEqual( + arithmetic.decode(sample_privsigningkey, 16), sample_factor) + + def test_encode(self): + """Encode sample factor into hex and compare to privsigningkey""" + self.assertEqual( + arithmetic.encode(sample_factor, 16), sample_privsigningkey) + + def test_changebase(self): + """Check the results of changebase()""" + self.assertEqual( + arithmetic.changebase(sample_privsigningkey, 16, 256, minlen=32), + unhexlify(sample_privsigningkey)) + self.assertEqual( + arithmetic.changebase(sample_pubsigningkey, 16, 256, minlen=64), + unhexlify(sample_pubsigningkey)) + self.assertEqual( + 32, # padding + len(arithmetic.changebase(sample_privsigningkey[:5], 16, 256, 32))) + + def test_hex_to_point(self): + """Check that sample_pubsigningkey is sample_point encoded in hex""" + self.assertEqual( + arithmetic.hex_to_point(sample_pubsigningkey), sample_point) + + def test_point_to_hex(self): + """Check that sample_point is sample_pubsigningkey decoded from hex""" + self.assertEqual( + arithmetic.point_to_hex(sample_point), sample_pubsigningkey) + + def test_privtopub(self): + """Generate public keys and check the result""" + self.assertEqual( + arithmetic.privtopub(sample_privsigningkey), + sample_pubsigningkey + ) + self.assertEqual( + arithmetic.privtopub(sample_privencryptionkey), + sample_pubencryptionkey + ) diff --git a/src/tests/test_addresses.py b/src/tests/test_addresses.py index e0b745cf..e37e2854 100644 --- a/src/tests/test_addresses.py +++ b/src/tests/test_addresses.py @@ -15,6 +15,7 @@ class TestAddresses(unittest.TestCase): self.assertEqual( addresses.decodeAddress(sample_address), ('success', 2, 1, unhexlify(sample_ripe))) + status, version, stream, ripe1 = addresses.decodeAddress( '2cWzSnwjJ7yRP3nLEWUV5LisTZyREWSzUK') self.assertEqual(status, 'success') diff --git a/src/tests/test_crypto.py b/src/tests/test_crypto.py index b4d7d3e9..38410359 100644 --- a/src/tests/test_crypto.py +++ b/src/tests/test_crypto.py @@ -7,7 +7,7 @@ import unittest from abc import ABCMeta, abstractmethod from binascii import hexlify -from pybitmessage.pyelliptic import arithmetic +from pybitmessage import highlevelcrypto try: @@ -16,7 +16,7 @@ except ImportError: RIPEMD = None from .samples import ( - sample_factor, sample_point, sample_pubsigningkey, sample_pubencryptionkey, + sample_pubsigningkey, sample_pubencryptionkey, sample_privsigningkey, sample_privencryptionkey, sample_ripe ) @@ -62,19 +62,13 @@ class TestCrypto(RIPEMD160TestCase, unittest.TestCase): class TestHighlevelcrypto(unittest.TestCase): """Test highlevelcrypto public functions""" - def test_base10_multiply(self): - """Test arithmetic.base10_multiply""" - self.assertEqual( - sample_point, - arithmetic.base10_multiply(arithmetic.G, sample_factor)) - def test_privtopub(self): """Generate public keys and check the result""" self.assertEqual( - arithmetic.privtopub(sample_privsigningkey).encode(), + highlevelcrypto.privToPub(sample_privsigningkey), hexlify(sample_pubsigningkey) ) self.assertEqual( - arithmetic.privtopub(sample_privencryptionkey).encode(), + highlevelcrypto.privToPub(sample_privencryptionkey), hexlify(sample_pubencryptionkey) ) diff --git a/tests.py b/tests.py index 0e88ba63..b933b212 100644 --- a/tests.py +++ b/tests.py @@ -14,7 +14,10 @@ def unittest_discover(): # randomize the order of tests in test cases loader.sortTestMethodsUsing = lambda a, b: random.randint(-1, 1) # pybitmessage symlink may disappear on Windows - return loader.discover('src.tests') + testsuite = loader.discover('src.tests') + testsuite.addTests([loader.discover('src.pyelliptic')]) + + return testsuite if __name__ == "__main__":