def on_ws_receive(self, data, *arg):
logger.info("do_socket_receive0")
if data.has_key("success") and data["success"]:
if wallethelper and data.has_key("type") and data["type"] == "Payment":
ret = wallethelper.getBalance()
print "2333333", ret
if self.get_wallet_status() == 0:
self.set_wallet_status(1)
elif self.get_wallet_status() == 2:
self.set_wallet_status(12)
elif data.has_key("type") and data["type"] == "OfferCreate":
logger.info("offer created:" + str(data) + str(arg))
# set last order hash for next test
if data.has_key("transaction"):
self.set_last_order_hash(data["transaction"]["hash"])
self.set_wallet_status(6)
elif data.has_key("type") and data["type"] == "OfferCancel":
logger.info("offer canceled:" + str(data) + str(arg))
self.set_wallet_status(8)
elif data.has_key("type") and data["type"] == "TrustSet":
logger.info("trust seted:" + str(data) + str(arg))
self.set_wallet_status(14)
else:
logger.info("do_socket_receive:" + str(data) + str(arg))
def on_ws_receive(self, data, *arg):
logger.info("do_socket_receive0")
if data.has_key("success") and data["success"]:
if wallethelper and data.has_key(
"type") and data["type"] == "Payment":
ret = wallethelper.getBalance()
print "2333333", ret
if self.get_wallet_status() == 0:
self.set_wallet_status(1)
elif self.get_wallet_status() == 2:
self.set_wallet_status(12)
elif data.has_key("type") and data["type"] == "OfferCreate":
logger.info("offer created:" + str(data) + str(arg))
# set last order hash for next test
if data.has_key("transaction"):
self.set_last_order_hash(data["transaction"]["hash"])
self.set_wallet_status(6)
elif data.has_key("type") and data["type"] == "OfferCancel":
logger.info("offer canceled:" + str(data) + str(arg))
self.set_wallet_status(8)
elif data.has_key("type") and data["type"] == "TrustSet":
logger.info("trust seted:" + str(data) + str(arg))
self.set_wallet_status(14)
else:
logger.info("do_socket_receive:" + str(data) + str(arg))
def do_socket_receive(data, *arg):
global test_account_status
logger.info("do_socket_receive0")
if data.has_key("success") and data["success"]:
if data.has_key("type") and data["type"] == "Payment":
arg[0][0].get_balances()
# set currency
if arg[0][0].currency.has_key(("USD", test_issuer)):
test_account_status = 3
elif arg[0][0].currency.has_key(("SWT", "")):
test_account_status = 2
elif data.has_key("type") and data["type"] == "OfferCreate":
logger.info("offer created:" + str(data) + str(arg))
# set last order hash for next test
if data.has_key("transaction"):
arg[0][0].set_last_order_hash(data["transaction"]["hash"])
test_account_status = 4
elif data.has_key("type") and data["type"] == "OfferCancel":
logger.info("offer canceled:" + str(data) + str(arg))
# # set last order hash for next test
# if data.has_key("transaction"):
# arg[0][0].set_last_order_hash(data["transaction"]["hash"])
test_account_status = 7
else:
logger.info("do_socket_receive:" + str(data) + str(arg))
def update_currency(self, data):
self.currency = {}
for _d in data:
self.currency[(_d["currency"], _d["counterparty"])] = Balance(**_d)
logger.info("update currency:" + str(self.currency))
if jt_account is not None:
websockethelper.subscribe_message_by_account(jt_account.address, jt_account.secret)
t = threading.Thread(target=websockethelper.receive, args=(do_socket_receive, jt_account))
t.setDaemon(True)
t.start()
# status check and action
while 1:
#try:
if test_account_status in (0, 1):
pass
elif test_account_status == 2:
if jt_account.last_resource_id is not None:
r = jt_account.get_payment(jt_account.last_resource_id)
logger.info("get_payment test:" + str(r))
jt_test_ulimit_account.payment("USD", 2, jt_account.address, test_issuer)
test_account_status = 0
elif test_account_status == 3:
r = jt_account.get_paths(test_ulimit_address, test_ulimit_secret, jt_account.address, "1.00", "USD", issuer=test_issuer)
logger.info("get_paths test:" + str(r))
jt_account.place_order("buy", "SWT", 5, "USD", 1, None, test_issuer)
test_account_status = 0
elif test_account_status == 4:
r = jt_account.get_payments(results_per_page=1, page=1)
logger.info("get_payments test:" + str(r))
test_account_status = 5
elif test_account_status == 5:
r = jt_account.get_account_orders()
wallethelper = WalletTest(my_address, my_secret, fingatehelper)
ulimit_wallethelper = Wallet(test_ulimit_address, test_ulimit_secret)
# register ws callback
fingatehelper.setTxHandler(wallethelper.on_ws_receive)
while 1:
if wallethelper and wallethelper.isActivated:
if wallethelper.get_wallet_status() == 1: # USD payment, from ulimit wallet to my wallet
ret = ulimit_wallethelper.submitPayment("USD", 2, wallethelper.address, issuer=test_issuer)
if ret.has_key("success") and ret["success"]:
wallethelper.set_last_resource_id(ret["client_resource_id"])
wallethelper.set_wallet_status(2)
elif wallethelper.get_wallet_status() == 3:
r = wallethelper.getPayment(wallethelper.last_resource_id)
logger.info("get_payment test:" + str(r))
r = wallethelper.getPaymentList(results_per_page=3, page=1)
logger.info("get_payments test:" + str(r))
wallethelper.set_wallet_status(4)
elif wallethelper.get_wallet_status() == 4:
r = wallethelper.getPathList(test_ulimit_address, test_ulimit_secret, wallethelper.address,
"1.00", "USD", issuer=test_issuer)
logger.info("get_paths test:" + str(r))
wallethelper.createOrder("buy", "SWT", 5, "USD", 1, None, test_issuer)
wallethelper.set_wallet_status(5)
elif wallethelper.get_wallet_status() == 6:
r = wallethelper.getOrderList()
logger.info("get_account_orders test:" + str(r))
r = wallethelper.cancelOrder(1)
logger.info("cancel_order 1 test:" + str(r))
def removerelation_callback(self, res):
logger.info("removerelation_callback:::::" + str(res))
fingatehelper.setTxHandler(wallethelper.on_ws_receive)
while 1:
if wallethelper and wallethelper.isActivated:
if wallethelper.get_wallet_status(
) == 1: # USD payment, from ulimit wallet to my wallet
ret = ulimit_wallethelper.submitPayment("USD",
2,
wallethelper.address,
issuer=test_issuer)
if ret.has_key("success") and ret["success"]:
wallethelper.set_last_resource_id(ret["client_resource_id"])
wallethelper.set_wallet_status(2)
elif wallethelper.get_wallet_status() == 3:
r = wallethelper.getPayment(wallethelper.last_resource_id)
logger.info("get_payment test:" + str(r))
r = wallethelper.getPaymentList(results_per_page=3, page=1)
logger.info("get_payments test:" + str(r))
wallethelper.set_wallet_status(4)
elif wallethelper.get_wallet_status() == 4:
r = wallethelper.getPathList(test_ulimit_address,
test_ulimit_secret,
wallethelper.address,
"1.00",
"USD",
issuer=test_issuer)
logger.info("get_paths test:" + str(r))
wallethelper.createOrder("buy", "SWT", 5, "USD", 1, None,
test_issuer)
wallethelper.set_wallet_status(5)
def getorderbook_callback(self, res):
logger.info("getorderbook_callback:::::" + str(res))
def addrelation_callback(self, res):
logger.info("addrelation_callback:::::" + str(res))
def cancelorder_callback(self, res):
logger.info("cancelorder_callback:::::" + str(res))
def createorder_callback(self, res):
logger.info("createorder_callback:::::" + str(res))
def payment_callback(self, res):
logger.info("payment_callback:::::" + str(res))
self.set_last_resource_id(res["client_resource_id"])
def active_callback(self, res):
logger.info("active_callback:::::" + str(res))
custom = str(cfg_data["DEV"]["fingate1"]["custom_id"])
test_currency = str(cfg_data["DEV"]["fingate1"]["tum1"])
# init FinGate
fingate = FinGate()
fingate.setMode(FinGate.DEVLOPMENT)
fingate.setAccount(test_secret, test_address)
fingate.setActivateAmount(30)
#tongtong testing
fingate.setToken(custom)
fingate.setKey(ekey)
order = fingate.getNextUUID()
ret = fingate.issueCustomTum(test_currency, "123.45", order,
test_ulimit_address)
logger.info("issueCustomTum:" + str(ret))
logger.info("queryIssue:" + str(fingate.queryIssue(order)))
ret = fingate.queryCustomTum(test_currency)
logger.info("queryCustomTum:" + str(fingate.queryIssue(order)))
#sys.exit(0)
# init test wallet
master_wallet = Wallet(test_secret, test_address)
master_unlimit_wallet = Wallet(test_ulimit_secret, test_ulimit_address)
class WalletClient(Wallet):
def __init__(self, wallet):
