"""
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
        )