Import from blockstack-server

integration_tests/blockstack_integration_tests/unit_tests.py

@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+    Blockstack
+    ~~~~~
+    copyright: (c) 2014-2015 by Halfmoon Labs, Inc.
+    copyright: (c) 2016 by Blockstack.org
+
+    This file is part of Blockstack
+
+    Blockstack is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    Blockstack is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+    You should have received a copy of the GNU General Public License
+    along with Blockstack. If not, see <http://www.gnu.org/licenses/>.
+"""
+
+import json
+import traceback
+import unittest
+import string
+from test import test_support
+
+from opennamelib import *
+from pybitcoin import *
+
+try:
+    with open('data/secrets.json', 'r') as f:
+        SECRETS = json.loads(f.read())
+except:
+    traceback.print_exc()
+
+blockchain_client = ChainComClient(
+    api_key_id=SECRETS['chain_api_id'],
+    api_key_secret=SECRETS['chain_api_secret'])
+
+registration_example_1 = {
+    'name': 'ryanshea',
+    'data': """{  }""",
+    'recipient': '1DuckDmHTXVxSHC7UafaBiUZB81qYhKprF'
+}
+
+
+class NamePreorderTest(unittest.TestCase):
+    def setUp(self):
+        self.data = registration_example_1
+        self.namedb = NameDb('data/namespace.txt')
+
+    def test_name_preorder(self):
+        consensus_hash128 = str(self.namedb.consensus_hashes['current'])
+        resp = preorder_name(
+            self.data['name'], consensus_hash128, SECRETS['private_keys'][0],
+            blockchain_client=blockchain_client,
+            testset=True)
+        print resp
+        self.assertTrue('success' in resp)
+
+
+class NameRegistrationTest(unittest.TestCase):
+    def setUp(self):
+        self.data = registration_example_1
+
+    def tearDown(self):
+        pass
+
+    def test_name_registration(self):
+        resp = register_name(
+            self.data['name'], SECRETS['private_keys'][0],
+            blockchain_client=blockchain_client, testset=True)
+        print resp
+        self.assertTrue('success' in resp)
+
+
+class NameUpdateTest(unittest.TestCase):
+    def setUp(self):
+        self.data = registration_example_1
+
+    def tearDown(self):
+        pass
+
+    def test_name_update(self):
+        resp = update_name(
+            self.data['name'], hex_hash160(self.data['data']), SECRETS['private_keys'][0],
+            blockchain_client=blockchain_client, testset=True)
+        print resp
+        self.assertTrue('success' in resp)
+
+
+class NameTransferTest(unittest.TestCase):
+    def setUp(self):
+        self.data = registration_example_1
+
+    def tearDown(self):
+        pass
+
+    def test_name_transfer(self):
+        resp = transfer_name(
+            self.data['name'], self.data['recipient'],
+            SECRETS['private_keys'][0], blockchain_client=blockchain_client,
+            testset=True)
+        print resp
+        self.assertTrue('success' in resp)
+
+
+class MerkleRootTest(unittest.TestCase):
+    def setUp(self):
+        self.hashes = [
+            'f484b014c55a43b409a59de3177d49a88149b4473f9a7b81ea9e3535d4b7a301',
+            '7b5636e9bc6ec910157e88702699bc7892675e8b489632c9166764341a4d4cfe',
+            'f8b02b8bf25cb6008e38eb5453a22c502f37e76375a86a0f0cfaa3c301aa1209'
+        ]
+        self.merkle_root = ("4f4c8c201e85a64a410cc7272c77f443d8b8df3289c67af9"
+                            "dab1e87d9e61985e")
+
+    def tearDown(self):
+        pass
+
+    def test_merkle_tree(self):
+        merkle_tree = MerkleTree(self.hashes)
+        self.assertEqual(merkle_tree.root(), self.merkle_root)
+
+    def test_calculate_merkle_root(self):
+        merkle_root = calculate_merkle_root(self.hashes)
+        self.assertEqual(merkle_root, self.merkle_root)
+
+
+def test_main():
+    test_support.run_unittest(
+        MerkleRootTest,
+        # NamePreorderTest,
+        # NameRegistrationTest,
+        NameUpdateTest,
+        # NameTransferTest,
+        # NameOperationSequenceTest
+    )
+
+if __name__ == '__main__':
+    test_main()