mirror of
https://github.com/alexgo-io/stacks-puppet-node.git
synced 2026-04-06 22:33:44 +08:00
145 lines
4.1 KiB
Python
145 lines
4.1 KiB
Python
#!/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()
|