zbxcat/test.py

import zXcat
import bXcat
from xcat import *

htlcTrade = Trade()
print("Starting test of xcat...")

def get_initiator_addresses():
    baddr = bXcat.new_bitcoin_addr()
    zaddr = zXcat.new_zcash_addr()
    # print("type baddr", type(baddr))
    # print("type baddr", type(baddr.to_scriptPubKey()))
    return {'bitcoin': baddr.__str__(), 'zcash': zaddr.__str__()}
    # return {'bitcoin': 'myfFr5twPYNwgeXyjCmGcrzXtCmfmWXKYp', 'zcash': 'tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ'}

def get_fulfiller_addresses():
    baddr = bXcat.new_bitcoin_addr()
    zaddr = zXcat.new_zcash_addr()
    return {'bitcoin': baddr.__str__(), 'zcash': zaddr.__str__()}
    # return {'bitcoin': 'myfFr5twPYNwgeXyjCmGcrzXtCmfmWXKYp', 'zcash': 'tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ'}
    # return {'bitcoin': 'mrQzUGU1dwsWRx5gsKKSDPNtrsP65vCA3Z', 'zcash': 'tmTjZSg4pX2Us6V5HttiwFZwj464fD2ZgpY'}

def initiate(trade):
    # Get amounts
    amounts = {"sell": {"currency": "bitcoin", "amount": "0.5"}, "buy": {"currency": "zcash", "amount": "1.12"}}
    sell = amounts['sell']
    buy = amounts['buy']
    sell_currency = sell['currency']
    buy_currency = buy['currency']
    # Get addresses
    init_addrs = get_initiator_addresses()
    sell['initiator'] = init_addrs[sell_currency]
    buy['initiator'] = init_addrs[buy_currency]
    fulfill_addrs = get_fulfiller_addresses()
    sell['fulfiller'] = fulfill_addrs[sell_currency]
    buy['fulfiller'] = fulfill_addrs[buy_currency]
    # initializing contract classes with addresses, currencies, and amounts
    trade.sellContract = Contract(sell)
    trade.buyContract = Contract(buy)
    print(trade.sellContract.__dict__)
    print(trade.buyContract.__dict__)

    secret = generate_password()
    print("Generating secret to lock funds:", secret)
    save_secret(secret)
    # TODO: Implement locktimes and mock block passage of time
    sell_locktime = 2
    buy_locktime = 4 # Must be more than first tx

    create_sell_p2sh(trade, secret, sell_locktime)
    txid = fund_sell_contract(trade)
    print("Sent")
    create_buy_p2sh(trade, secret, buy_locktime)

# buyer checks that seller funded the sell contract, and if so funds the buy contract
def buyer_fulfill(trade):
    print("BUYER FULFILL")
    print("=============")
    buy = trade.buyContract
    sell = trade.sellContract
    # buy_p2sh_balance = check_p2sh(buy.currency, buy.p2sh)
    sell_p2sh_balance = check_p2sh(sell.currency, sell.p2sh)
    if (sell_p2sh_balance < float(sell.amount)):
                raise ValueError("Sell p2sh not funded, buyer cannot redeem")
    print("Seller has deposited funds, so funding the buy contract:")
    txid = fund_buy_contract(trade)
    print("Buyer Fund tx txid:", txid)

def redeem_seller(trade):
    print("SELLER REDEEMING BUY CONTRACT")
    print("=============================")
    buy = trade.buyContract
    if trade.sellContract.get_status() == 'redeemed':
        raise RuntimeError("Sell contract status was already redeemed before seller could redeem buyer's tx")
    else:
        secret = get_secret() # Just the seller getting his local copy of the secret
        print("SELLER SECRET IN TEST:", secret)
        txid =  redeem_p2sh(trade.buyContract, secret, trade.sellContract)
        setattr(trade.buyContract, 'redeem_tx', txid)
        save(trade)
        print("You have redeemed {0} {1}!".format(buy.amount, buy.currency))

def redeem_buyer(trade):
    print("BUYER REDEEMING SELL CONTRACT")
    print("=============================")

    if trade.sellContract.get_status() == 'redeemed':
        raise RuntimeError("Sell contract was redeemed before buyer could retrieve funds")
    elif trade.buyContract.get_status() == 'refunded':
        print("buyContract was refunded to buyer")
    else:
        # Buy contract is where seller disclosed secret in redeeming
        if trade.buyContract.currency == 'bitcoin':
            secret = bXcat.find_secret(trade.buyContract.p2sh,trade.buyContract.fund_tx)
        else:
            secret = zXcat.find_secret(trade.buyContract.p2sh,trade.buyContract.fund_tx)
        print("Found secret in seller's redeem tx", secret)
        redeem_tx = redeem_p2sh(trade.sellContract, secret, trade.buyContract)
        setattr(trade.sellContract, 'redeem_tx', redeem_tx)
        save(trade)

def generate_blocks(num):
    bXcat.generate(num)
    zXcat.generate(num)

initiate(htlcTrade)
buyer_fulfill(htlcTrade)

# generate_blocks(12)
redeem_seller(htlcTrade)
redeem_buyer(htlcTrade)

# addr = CBitcoinAddress('tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ')
# print(addr)
# # print(b2x('tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ'))
# print(b2x(addr))