test for 'correct' execution paths on datastores

integration_tests/blockstack_integration_tests/scenarios/name_preorder_register_update_app_datastore.py

@@ -0,0 +1,276 @@
+#!/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 os
+import testlib
+import pybitcoin
+import urllib2
+import json
+import blockstack_client
+import blockstack_profiles
+import blockstack_gpg
+import sys
+
+wallets = [
+    testlib.Wallet( "5JesPiN68qt44Hc2nT8qmyZ1JDwHebfoh9KQ52Lazb1m1LaKNj9", 100000000000 ),
+    testlib.Wallet( "5KHqsiU9qa77frZb6hQy9ocV7Sus9RWJcQGYYBJJBb2Efj1o77e", 100000000000 ),
+    testlib.Wallet( "5Kg5kJbQHvk1B64rJniEmgbD83FpZpbw2RjdAZEzTefs9ihN3Bz", 100000000000 ),
+    testlib.Wallet( "5JuVsoS9NauksSkqEjbUZxWwgGDQbMwPsEfoRBSpLpgDX1RtLX7", 100000000000 ),
+    testlib.Wallet( "5KEpiSRr1BrT8vRD7LKGCEmudokTh1iMHbiThMQpLdwBwhDJB1T", 100000000000 ),
+    testlib.Wallet( "5K5hDuynZ6EQrZ4efrchCwy6DLhdsEzuJtTDAf3hqdsCKbxfoeD", 100000000000 ),
+    testlib.Wallet( "5J39aXEeHh9LwfQ4Gy5Vieo7sbqiUMBXkPH7SaMHixJhSSBpAqz", 100000000000 ),
+    testlib.Wallet( "5K9LmMQskQ9jP1p7dyieLDAeB6vsAj4GK8dmGNJAXS1qHDqnWhP", 100000000000 ),
+    testlib.Wallet( "5KcNen67ERBuvz2f649t9F2o1ddTjC5pVUEqcMtbxNgHqgxG2gZ", 100000000000 )
+]
+
+consensus = "17ac43c1d8549c3181b200f1bf97eb7d"
+wallet_keys = None
+error = False
+
+def scenario( wallets, **kw ):
+
+    global wallet_keys, error, index_file_data, resource_data
+
+    testlib.blockstack_namespace_preorder( "test", wallets[1].addr, wallets[0].privkey )
+    testlib.next_block( **kw )
+
+    testlib.blockstack_namespace_reveal( "test", wallets[1].addr, 52595, 250, 4, [6,5,4,3,2,1,0,0,0,0,0,0,0,0,0,0], 10, 10, wallets[0].privkey )
+    testlib.next_block( **kw )
+
+    testlib.blockstack_namespace_ready( "test", wallets[1].privkey )
+    testlib.next_block( **kw )
+
+    testlib.blockstack_name_preorder( "foo.test", wallets[2].privkey, wallets[3].addr )
+    testlib.next_block( **kw )
+    
+    testlib.blockstack_name_register( "foo.test", wallets[2].privkey, wallets[3].addr )
+    testlib.next_block( **kw )
+
+    test_proxy = testlib.TestAPIProxy()
+    blockstack_client.set_default_proxy( test_proxy )
+
+    wallet_keys = blockstack_client.make_wallet_keys( payment_privkey=wallets[5].privkey, owner_privkey=wallets[3].privkey, data_privkey=wallets[4].privkey )
+
+    wallet = testlib.blockstack_client_initialize_wallet( "0123456789abcdef", wallets[5].privkey, wallets[3].privkey, wallets[4].privkey )
+    
+    # migrate profiles 
+    res = testlib.migrate_profile( "foo.test", proxy=test_proxy, wallet_keys=wallet_keys )
+    if 'error' in res:
+        res['test'] = 'Failed to initialize foo.test profile'
+        print json.dumps(res, indent=4, sort_keys=True)
+        error = True
+        return 
+
+    # tell serialization-checker that value_hash can be ignored here
+    print "BLOCKSTACK_SERIALIZATION_CHECK_IGNORE value_hash"
+    sys.stdout.flush()
+    
+    testlib.next_block( **kw )
+
+    # make a datastore 
+    res = testlib.blockstack_cli_advanced_put_datastore( "foo.test", "foo_datastore", drivers=['disk'], password="0123456789abcdef" )
+    if 'error' in res:
+        print 'failed to create datastore: {}'.format(res['error'])
+        return False
+
+    # get the datastore (for fun) 
+    dsres = testlib.blockstack_cli_advanced_get_datastore( "foo.test", "foo_datastore", password='0123456789abcdef' )
+    if 'error' in res:
+        print 'failed to get datastore: {}'.format(res['error'])
+        return False
+
+    # make directories
+    for dpath in ['/dir1', '/dir2', '/dir1/dir3', '/dir1/dir3/dir4']:
+        print 'mkdir {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_mkdir( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to mkdir {}: {}'.format(dpath, res['error'])
+            return False
+
+    # list directories 
+    for dpath, expected in [('/', ['dir1', 'dir2']), ('/dir1', ['dir3']), ('/dir1/dir3', ['dir4']), ('/dir1/dir3/dir4', [])]:
+        print 'listdir {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_listdir( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to listdir {}: {}'.format(dpath, res['error'])
+            return False
+
+        res = res['dir']
+        if len(res['idata'].keys()) != len(expected):
+            print 'invalid directory: expected:\n{}\ngot:\n{}\n'.format(expected, res['idata'])
+            return False
+
+        for child in expected: 
+            if not res['idata'].has_key(child):
+                print 'invalid directory: missing {} in {}'.format(child, res['idata'])
+                return False
+
+    # put files 
+    for dpath in ['/file1', '/file2', '/dir1/file3', '/dir1/dir3/file4', '/dir1/dir3/dir4/file5']:
+        print 'putfile {}'.format(dpath)
+        data = 'hello {}'.format(os.path.basename(dpath))
+        res = testlib.blockstack_cli_advanced_datastore_putfile( 'foo.test', 'foo_datastore', dpath, data )
+        if 'error' in res:
+            print 'failed to putfile {}: {}'.format(dpath, res['error'])
+            return False
+
+    # list directories again 
+    for dpath, expected in [('/', ['dir1', 'dir2', 'file1', 'file2']), ('/dir1', ['dir3', 'file3']), ('/dir1/dir3', ['dir4', 'file4']), ('/dir1/dir3/dir4', ['file5'])]:
+        print 'listdir {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_listdir( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to listdir {}: {}'.format(dpath, res['error'])
+            return False
+
+        res = res['dir']
+        if len(res['idata'].keys()) != len(expected):
+            print 'invalid directory: expected:\n{}\ngot:\n{}\n'.format(expected, res['idata'])
+            return False
+
+        for child in expected: 
+            if not res['idata'].has_key(child):
+                print 'invalid directory: missing {} in {}'.format(child, res['idata'])
+                return False
+     
+    # get files
+    for dpath in ['/file1', '/file2', '/dir1/file3', '/dir1/dir3/file4', '/dir1/dir3/dir4/file5']:
+        print 'getfile {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_getfile( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to getfile {}: {}'.format(dpath, res['error'])
+            return False
+
+        res = res['file']
+        if res['idata'] != 'hello {}'.format(os.path.basename(dpath)):
+            print 'failed to read {}: got "{}"'.format(dpath, res['idata'])
+            return False
+
+    # remove files
+    for dpath in ['/file1', '/file2', '/dir1/file3', '/dir1/dir3/file4', '/dir1/dir3/dir4/file5']:
+        print 'deletefile {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_deletefile( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to deletefile {}: {}'.format(dpath, res['error'])
+            return False
+
+    # list directories, 3rd time 
+    for dpath, expected in [('/', ['dir1', 'dir2']), ('/dir1', ['dir3']), ('/dir1/dir3', ['dir4']), ('/dir1/dir3/dir4', [])]:
+        print 'listdir {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_listdir( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to listdir {}: {}'.format(dpath, res['error'])
+            return False
+
+        res = res['dir']
+        if len(res['idata'].keys()) != len(expected):
+            print 'invalid directory: expected:\n{}\ngot:\n{}\n'.format(expected, res['idata'])
+            return False
+
+        for child in expected: 
+            if not res['idata'].has_key(child):
+                print 'invalid directory: missing {} in {}'.format(child, res['idata'])
+                return False
+
+    # remove directories 
+    for dpath in ['/dir1/dir3/dir4', '/dir1/dir3', '/dir2', '/dir1']:
+        print 'rmdir {}'.format(dpath)
+        res = testlib.blockstack_cli_advanced_datastore_rmdir( 'foo.test', 'foo_datastore', dpath )
+        if 'error' in res:
+            print 'failed to rmdir {}: {}'.format(dpath, res['error'])
+            return False
+
+    # root should be empty 
+    print 'listdir {}'.format('/')
+    res = testlib.blockstack_cli_advanced_datastore_listdir( 'foo.test', 'foo_datastore', '/' )
+    if 'error' in res:
+        print 'failed to listdir /: {}'.format(res['error'])
+        return False
+
+    res = res['dir']
+    if len(res['idata'].keys()) > 0:
+        print 'root still has children: {}'.format(res['idata'].keys())
+        return False
+
+    # delete datastore (shouldn't have to force)
+    res = testlib.blockstack_cli_advanced_delete_datastore( 'foo.test', 'foo_datastore', password="0123456789abcdef")
+    if 'error' in res:
+        print 'failed to delete datastore: {}'.format(res['error'])
+        return False
+
+    testlib.next_block( **kw )
+
+
+
+def check( state_engine ):
+
+    global wallet_keys, error
+
+    if error:
+        print "Key operation failed."
+        return False
+
+    # not revealed, but ready 
+    ns = state_engine.get_namespace_reveal( "test" )
+    if ns is not None:
+        print "namespace not ready"
+        return False 
+
+    ns = state_engine.get_namespace( "test" )
+    if ns is None:
+        print "no namespace"
+        return False 
+
+    if ns['namespace_id'] != 'test':
+        print "wrong namespace"
+        return False 
+
+    names = ['foo.test']
+    wallet_keys_list = [wallet_keys]
+    test_proxy = testlib.TestAPIProxy()
+
+    for i in xrange(0, len(names)):
+        name = names[i]
+        wallet_payer = 3 * (i+1) - 1
+        wallet_owner = 3 * (i+1)
+        wallet_data_pubkey = 3 * (i+1) + 1
+        wallet_keys = wallet_keys_list[i]
+
+        # not preordered
+        preorder = state_engine.get_name_preorder( name, pybitcoin.make_pay_to_address_script(wallets[wallet_payer].addr), wallets[wallet_owner].addr )
+        if preorder is not None:
+            print "still have preorder"
+            return False
+    
+        # registered 
+        name_rec = state_engine.get_name( name )
+        if name_rec is None:
+            print "name does not exist"
+            return False 
+
+        # owned 
+        if name_rec['address'] != wallets[wallet_owner].addr or name_rec['sender'] != pybitcoin.make_pay_to_address_script(wallets[wallet_owner].addr):
+            print "name has wrong owner"
+            return False 
+
+        # try to authenticate
+
+    return True