import logging
import unittest

from aucoin import dsa
from aucoin.blockchain import Blockchain
from aucoin.wallet import Wallet


logging.basicConfig(level=logging.DEBUG)


class TestWallet(unittest.TestCase):
    def setUp(self):
        self.wallet = Wallet(Blockchain())
        self.data = b"Hello, world!"

    def test_sign(self):
        signature = self.wallet.sign(self.data)
        self.assertTrue(dsa.verify(self.wallet._public_key, self.data, signature))

    def test_save_load(self):
        public_key_old = self.wallet.public_key
        self.wallet.save()
        self.wallet._private_key = None
        self.wallet._public_key = None
        self.wallet.load()
        self.assertEqual(self.wallet.public_key, public_key_old)

if __name__ == '__main__':
    unittest.main()