From 1a34eaa43c54a22ae02e0399f41d8ac66a57bd3c Mon Sep 17 00:00:00 2001
From: Alexey Karyabkin <coffein.h@gmail.com>
Date: Mon, 4 Jun 2018 19:29:59 +0400
Subject: [PATCH] added tests for create master & pub keys for HD
 Wallet(BIP0032)

---
 tests/bip0032_fixtures.py | 25 +++++++++++++++++++++++++
 tests/conftest.py         |  2 +-
 tests/test_bip0032.py     | 31 +++++++++++++++++++++++++++++++
 3 files changed, 57 insertions(+), 1 deletion(-)
 create mode 100644 tests/bip0032_fixtures.py
 create mode 100644 tests/test_bip0032.py

diff --git a/tests/bip0032_fixtures.py b/tests/bip0032_fixtures.py
new file mode 100644
index 0000000..0fa9dba
--- /dev/null
+++ b/tests/bip0032_fixtures.py
@@ -0,0 +1,25 @@
+import pytest
+
+
+@pytest.fixture
+def fail_key1():
+    return b'\x00\x00\x00\x00'
+
+
+@pytest.fixture
+def fail_key2():
+    return b'\x97\x8bq\xc6\xd8\xfew\xe5\xfa\xad\xdc\xc6\xc5\x91\xbd\xfb'
+
+
+@pytest.fixture
+def good_key():
+    return b'B\xa8\xe9v>y\xe2\x82\x10\x80\xc2\xa91\x10E\xe0XJ\xe6\xc7\x18\x9eE~\xa0^\xd1\x820\xe7\x18\x0c'
+
+@pytest.fixture
+def master_key_hdwallet():
+    return dict(version=b'\x04\x88\xad\xe4',
+                key=b"Y\x9e'\xe00or'\xacD\x9c(l\x99\x0fxB\x03\xbd/]|+\xfd\xe89K!\x93\x0bN\x9b",
+                chain_code=b'B\xa8\xe9v>y\xe2\x82\x10\x80\xc2\xa91\x10E\xe0XJ\xe6\xc7\x18\x9eE~\xa0^\xd1\x820\xe7\x18\x0c',
+                is_private=True,
+                depth=0,
+                child=0)
diff --git a/tests/conftest.py b/tests/conftest.py
index 3d541ea..d9c3db4 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -2,5 +2,5 @@
 import pytest
 
 
-pytest_plugins = ['bip0039_fixtures']
+pytest_plugins = ['bip0032_fixtures', 'bip0039_fixtures']
 
diff --git a/tests/test_bip0032.py b/tests/test_bip0032.py
new file mode 100644
index 0000000..9c1d34c
--- /dev/null
+++ b/tests/test_bip0032.py
@@ -0,0 +1,31 @@
+import os
+import random
+import hashlib
+import hmac
+
+from binascii import hexlify, unhexlify
+from pybtc.hdwallet import *
+
+
+
+def test_create_master_key(mnemonic_256):
+    passphrase = ' '.join(mnemonic_256)
+    seed = create_seed(passphrase, 'P@ssw0rd')
+    assert seed is not None
+    assert len(seed) == 64
+    
+    master_key = create_master_key_hdwallet(seed)
+    assert master_key is not None
+    assert type(master_key) is dict
+    assert master_key['is_private']
+
+
+def test_create_public_key(master_key_hdwallet):
+    public_key = create_public_key_hdwallet(master_key_hdwallet['key'])
+    assert public_key is not None
+
+
+def test_validate_keys(fail_key1, fail_key2, good_key):
+    assert not validate_keys(fail_key1)
+    assert not validate_keys(fail_key2)
+    assert validate_keys(good_key)