test.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801
  1. # python3 -m pip install pycryptodomex uvarint pyteal web3 coincurve
  2. import sys
  3. sys.path.append("..")
  4. from admin import PortalCore, Account
  5. from gentest import GenTest
  6. from base64 import b64decode
  7. from typing import List, Tuple, Dict, Any, Optional, Union
  8. import base64
  9. import random
  10. import time
  11. import hashlib
  12. import uuid
  13. import json
  14. from algosdk.v2client.algod import AlgodClient
  15. from algosdk.kmd import KMDClient
  16. from algosdk import account, mnemonic
  17. from algosdk.encoding import decode_address, encode_address
  18. from algosdk.future import transaction
  19. from pyteal import compileTeal, Mode, Expr
  20. from pyteal import *
  21. from algosdk.logic import get_application_address
  22. from vaa_verify import get_vaa_verify
  23. from algosdk.future.transaction import LogicSig
  24. from test_contract import get_test_app
  25. from algosdk.v2client import indexer
  26. import pprint
  27. class AlgoTest(PortalCore):
  28. def __init__(self) -> None:
  29. super().__init__()
  30. def getBalances(self, client: AlgodClient, account: str) -> Dict[int, int]:
  31. balances: Dict[int, int] = dict()
  32. accountInfo = client.account_info(account)
  33. # set key 0 to Algo balance
  34. balances[0] = accountInfo["amount"]
  35. assets: List[Dict[str, Any]] = accountInfo.get("assets", [])
  36. for assetHolding in assets:
  37. assetID = assetHolding["asset-id"]
  38. amount = assetHolding["amount"]
  39. balances[assetID] = amount
  40. return balances
  41. def createTestApp(
  42. self,
  43. client: AlgodClient,
  44. sender: Account,
  45. ) -> int:
  46. approval, clear = get_test_app(client)
  47. globalSchema = transaction.StateSchema(num_uints=4, num_byte_slices=30)
  48. localSchema = transaction.StateSchema(num_uints=0, num_byte_slices=16)
  49. app_args = []
  50. txn = transaction.ApplicationCreateTxn(
  51. sender=sender.getAddress(),
  52. on_complete=transaction.OnComplete.NoOpOC,
  53. approval_program=b64decode(approval["result"]),
  54. clear_program=b64decode(clear["result"]),
  55. global_schema=globalSchema,
  56. local_schema=localSchema,
  57. app_args=app_args,
  58. sp=client.suggested_params(),
  59. )
  60. signedTxn = txn.sign(sender.getPrivateKey())
  61. client.send_transaction(signedTxn)
  62. response = self.waitForTransaction(client, signedTxn.get_txid())
  63. assert response.applicationIndex is not None and response.applicationIndex > 0
  64. txn = transaction.PaymentTxn(sender = sender.getAddress(), sp = client.suggested_params(),
  65. receiver = get_application_address(response.applicationIndex), amt = 300000)
  66. signedTxn = txn.sign(sender.getPrivateKey())
  67. client.send_transaction(signedTxn)
  68. return response.applicationIndex
  69. def parseSeqFromLog(self, txn):
  70. try:
  71. return int.from_bytes(b64decode(txn.innerTxns[-1]["logs"][0]), "big")
  72. except Exception as err:
  73. pprint.pprint(txn.__dict__)
  74. raise
  75. def getVAA(self, client, sender, sid, app):
  76. if sid == None:
  77. raise Exception("getVAA called with a sid of None")
  78. saddr = get_application_address(app)
  79. # SOOO, we send a nop txn through to push the block forward
  80. # one
  81. # This is ONLY needed on a local net... the indexer will sit
  82. # on the last block for 30 to 60 seconds... we don't want this
  83. # log in prod since it is wasteful of gas
  84. if (self.INDEXER_ROUND > 512 and not self.args.testnet): # until they fix it
  85. print("indexer is broken in local net... stop/clean/restart the sandbox")
  86. sys.exit(0)
  87. txns = []
  88. txns.append(
  89. transaction.ApplicationCallTxn(
  90. sender=sender.getAddress(),
  91. index=self.tokenid,
  92. on_complete=transaction.OnComplete.NoOpOC,
  93. app_args=[b"nop"],
  94. sp=client.suggested_params(),
  95. )
  96. )
  97. self.sendTxn(client, sender, txns, False)
  98. if self.myindexer == None:
  99. print("indexer address: " + self.INDEXER_ADDRESS)
  100. self.myindexer = indexer.IndexerClient(indexer_token=self.INDEXER_TOKEN, indexer_address=self.INDEXER_ADDRESS)
  101. while True:
  102. nexttoken = ""
  103. while True:
  104. response = self.myindexer.search_transactions( min_round=self.INDEXER_ROUND, next_page=nexttoken)
  105. # pprint.pprint(response)
  106. for x in response["transactions"]:
  107. # pprint.pprint(x)
  108. if 'inner-txns' not in x:
  109. continue
  110. for y in x["inner-txns"]:
  111. if "application-transaction" not in y:
  112. continue
  113. if y["application-transaction"]["application-id"] != self.coreid:
  114. continue
  115. if len(y["logs"]) == 0:
  116. continue
  117. args = y["application-transaction"]["application-args"]
  118. if len(args) < 2:
  119. continue
  120. if base64.b64decode(args[0]) != b'publishMessage':
  121. continue
  122. seq = int.from_bytes(base64.b64decode(y["logs"][0]), "big")
  123. if seq != sid:
  124. continue
  125. if y["sender"] != saddr:
  126. continue;
  127. emitter = decode_address(y["sender"])
  128. payload = base64.b64decode(args[1])
  129. # pprint.pprint([seq, y["sender"], payload.hex()])
  130. # sys.exit(0)
  131. return self.gt.genVaa(emitter, seq, payload)
  132. if 'next-token' in response:
  133. nexttoken = response['next-token']
  134. else:
  135. self.INDEXER_ROUND = response['current-round'] + 1
  136. break
  137. time.sleep(1)
  138. def publishMessage(self, client, sender, vaa, appid):
  139. aa = decode_address(get_application_address(appid)).hex()
  140. emitter_addr = self.optin(client, sender, self.coreid, 0, aa)
  141. txns = []
  142. sp = client.suggested_params()
  143. a = transaction.ApplicationCallTxn(
  144. sender=sender.getAddress(),
  145. index=appid,
  146. on_complete=transaction.OnComplete.NoOpOC,
  147. app_args=[b"test1", vaa, self.coreid],
  148. foreign_apps = [self.coreid],
  149. accounts=[emitter_addr],
  150. sp=sp
  151. )
  152. a.fee = a.fee * 2
  153. txns.append(a)
  154. resp = self.sendTxn(client, sender, txns, True)
  155. self.INDEXER_ROUND = resp.confirmedRound
  156. return self.parseSeqFromLog(resp)
  157. def createTestAsset(self, client, sender):
  158. txns = []
  159. a = transaction.PaymentTxn(
  160. sender = sender.getAddress(),
  161. sp = client.suggested_params(),
  162. receiver = get_application_address(self.testid),
  163. amt = 300000
  164. )
  165. txns.append(a)
  166. sp = client.suggested_params()
  167. a = transaction.ApplicationCallTxn(
  168. sender=sender.getAddress(),
  169. index=self.testid,
  170. on_complete=transaction.OnComplete.NoOpOC,
  171. app_args=[b"setup"],
  172. sp=sp
  173. )
  174. a.fee = a.fee * 2
  175. txns.append(a)
  176. transaction.assign_group_id(txns)
  177. grp = []
  178. pk = sender.getPrivateKey()
  179. for t in txns:
  180. grp.append(t.sign(pk))
  181. client.send_transactions(grp)
  182. resp = self.waitForTransaction(client, grp[-1].get_txid())
  183. aid = int.from_bytes(resp.__dict__["logs"][0], "big")
  184. print("Opting " + sender.getAddress() + " into " + str(aid))
  185. self.asset_optin(client, sender, aid, sender.getAddress())
  186. txns = []
  187. a = transaction.ApplicationCallTxn(
  188. sender=sender.getAddress(),
  189. index=self.testid,
  190. on_complete=transaction.OnComplete.NoOpOC,
  191. app_args=[b"mint"],
  192. foreign_assets = [aid],
  193. sp=sp
  194. )
  195. a.fee = a.fee * 2
  196. txns.append(a)
  197. resp = self.sendTxn(client, sender, txns, True)
  198. # self.INDEXER_ROUND = resp.confirmedRound
  199. return aid
  200. def getCreator(self, client, sender, asset_id):
  201. return client.asset_info(asset_id)["params"]["creator"]
  202. def testAttest(self, client, sender, asset_id):
  203. taddr = get_application_address(self.tokenid)
  204. aa = decode_address(taddr).hex()
  205. emitter_addr = self.optin(client, sender, self.coreid, 0, aa)
  206. if asset_id != 0:
  207. creator = self.getCreator(client, sender, asset_id)
  208. c = client.account_info(creator)
  209. wormhole = c.get("auth-addr") == taddr
  210. else:
  211. c = None
  212. wormhole = False
  213. if not wormhole:
  214. creator = self.optin(client, sender, self.tokenid, asset_id, b"native".hex())
  215. txns = []
  216. sp = client.suggested_params()
  217. txns.append(transaction.ApplicationCallTxn(
  218. sender=sender.getAddress(),
  219. index=self.tokenid,
  220. on_complete=transaction.OnComplete.NoOpOC,
  221. app_args=[b"nop"],
  222. sp=sp
  223. ))
  224. mfee = self.getMessageFee()
  225. if (mfee > 0):
  226. txns.append(transaction.PaymentTxn(sender = sender.getAddress(), sp = sp, receiver = get_application_address(self.tokenid), amt = mfee))
  227. accts = [emitter_addr, creator, get_application_address(self.coreid)]
  228. if c != None:
  229. accts.append(c["address"])
  230. a = transaction.ApplicationCallTxn(
  231. sender=sender.getAddress(),
  232. index=self.tokenid,
  233. on_complete=transaction.OnComplete.NoOpOC,
  234. app_args=[b"attestToken", asset_id],
  235. foreign_apps = [self.coreid],
  236. foreign_assets = [asset_id],
  237. accounts=accts,
  238. sp=sp
  239. )
  240. if (mfee > 0):
  241. a.fee = a.fee * 3
  242. else:
  243. a.fee = a.fee * 2
  244. txns.append(a)
  245. resp = self.sendTxn(client, sender, txns, True)
  246. # Point us at the correct round
  247. self.INDEXER_ROUND = resp.confirmedRound
  248. # print(encode_address(resp.__dict__["logs"][0]))
  249. # print(encode_address(resp.__dict__["logs"][1]))
  250. # pprint.pprint(resp.__dict__)
  251. return self.parseSeqFromLog(resp)
  252. def transferFromAlgorand(self, client, sender, asset_id, quantity, receiver, chain, fee, payload = None):
  253. # pprint.pprint(["transferFromAlgorand", asset_id, quantity, receiver, chain, fee])
  254. taddr = get_application_address(self.tokenid)
  255. aa = decode_address(taddr).hex()
  256. emitter_addr = self.optin(client, sender, self.coreid, 0, aa)
  257. # asset_id 0 is ALGO
  258. if asset_id == 0:
  259. wormhole = False
  260. else:
  261. creator = self.getCreator(client, sender, asset_id)
  262. c = client.account_info(creator)
  263. wormhole = c.get("auth-addr") == taddr
  264. txns = []
  265. mfee = self.getMessageFee()
  266. if (mfee > 0):
  267. txns.append(transaction.PaymentTxn(sender = sender.getAddress(), sp = sp, receiver = get_application_address(self.tokenid), amt = mfee))
  268. if not wormhole:
  269. creator = self.optin(client, sender, self.tokenid, asset_id, b"native".hex())
  270. print("non wormhole account " + creator)
  271. sp = client.suggested_params()
  272. if (asset_id != 0) and (not self.asset_optin_check(client, sender, asset_id, creator)):
  273. print("Looks like we need to optin")
  274. txns.append(
  275. transaction.PaymentTxn(
  276. sender=sender.getAddress(),
  277. receiver=creator,
  278. amt=100000,
  279. sp=sp
  280. )
  281. )
  282. # The tokenid app needs to do the optin since it has signature authority
  283. a = transaction.ApplicationCallTxn(
  284. sender=sender.getAddress(),
  285. index=self.tokenid,
  286. on_complete=transaction.OnComplete.NoOpOC,
  287. app_args=[b"optin", asset_id],
  288. foreign_assets = [asset_id],
  289. accounts=[creator],
  290. sp=sp
  291. )
  292. a.fee = a.fee * 2
  293. txns.append(a)
  294. self.sendTxn(client, sender, txns, False)
  295. txns = []
  296. txns.insert(0,
  297. transaction.ApplicationCallTxn(
  298. sender=sender.getAddress(),
  299. index=self.tokenid,
  300. on_complete=transaction.OnComplete.NoOpOC,
  301. app_args=[b"nop"],
  302. sp=client.suggested_params(),
  303. )
  304. )
  305. if asset_id == 0:
  306. print("asset_id == 0")
  307. txns.append(transaction.PaymentTxn(
  308. sender=sender.getAddress(),
  309. receiver=creator,
  310. amt=quantity,
  311. sp=sp,
  312. ))
  313. accounts=[emitter_addr, creator, creator]
  314. else:
  315. print("asset_id != 0")
  316. txns.append(
  317. transaction.AssetTransferTxn(
  318. sender = sender.getAddress(),
  319. sp = sp,
  320. receiver = creator,
  321. amt = quantity,
  322. index = asset_id
  323. ))
  324. accounts=[emitter_addr, creator, c["address"]]
  325. args = [b"sendTransfer", asset_id, quantity, receiver, chain, fee]
  326. if None != payload:
  327. args.append(payload)
  328. #pprint.pprint(args)
  329. # print(self.tokenid)
  330. a = transaction.ApplicationCallTxn(
  331. sender=sender.getAddress(),
  332. index=self.tokenid,
  333. on_complete=transaction.OnComplete.NoOpOC,
  334. app_args=args,
  335. foreign_apps = [self.coreid],
  336. foreign_assets = [asset_id],
  337. accounts=accounts,
  338. sp=sp
  339. )
  340. a.fee = a.fee * 2
  341. txns.append(a)
  342. resp = self.sendTxn(client, sender, txns, True)
  343. self.INDEXER_ROUND = resp.confirmedRound
  344. # pprint.pprint([self.coreid, self.tokenid, resp.__dict__,
  345. # int.from_bytes(resp.__dict__["logs"][1], "big"),
  346. # int.from_bytes(resp.__dict__["logs"][2], "big"),
  347. # int.from_bytes(resp.__dict__["logs"][3], "big"),
  348. # int.from_bytes(resp.__dict__["logs"][4], "big"),
  349. # int.from_bytes(resp.__dict__["logs"][5], "big")
  350. # ])
  351. # print(encode_address(resp.__dict__["logs"][0]))
  352. # print(encode_address(resp.__dict__["logs"][1]))
  353. return self.parseSeqFromLog(resp)
  354. def asset_optin_check(self, client, sender, asset, receiver):
  355. if receiver not in self.asset_cache:
  356. self.asset_cache[receiver] = {}
  357. if asset in self.asset_cache[receiver]:
  358. return True
  359. ai = client.account_info(receiver)
  360. if "assets" in ai:
  361. for x in ai["assets"]:
  362. if x["asset-id"] == asset:
  363. self.asset_cache[receiver][asset] = True
  364. return True
  365. return False
  366. def asset_optin(self, client, sender, asset, receiver):
  367. if self.asset_optin_check(client, sender, asset, receiver):
  368. return
  369. pprint.pprint(["asset_optin", asset, receiver])
  370. sp = client.suggested_params()
  371. optin_txn = transaction.AssetTransferTxn(
  372. sender = sender.getAddress(),
  373. sp = sp,
  374. receiver = receiver,
  375. amt = 0,
  376. index = asset
  377. )
  378. transaction.assign_group_id([optin_txn])
  379. signed_optin = optin_txn.sign(sender.getPrivateKey())
  380. client.send_transactions([signed_optin])
  381. resp = self.waitForTransaction(client, signed_optin.get_txid())
  382. assert self.asset_optin_check(client, sender, asset, receiver), "The optin failed"
  383. print("woah! optin succeeded")
  384. def simple_test(self):
  385. #vaa = self.parseVAA(bytes.fromhex("01000000000100ddc6993585b909c3e861830244122e0daf45101663942484aa56ee2b51fa3ff016411102f993935d428a5aa0c3ace74facd60822435893b74b24fadde0fbad49006277c3fe0000000000088edf5b0e108c3a1a0a4b704cc89591f2ad8d50df24e991567e640ed720a94be200000000000000060003000000000000000000000000000000000000000000000000000000000000006400000000000000000000000000000000000000000000000000000000000000000008000000000000000000000000000000000000000000000000000000000000000800080000000000000000000000000000000000000000000000000000000000000000ff"))
  386. #pprint.pprint(vaa)
  387. #sys.exit(0)
  388. # q = bytes.fromhex(gt.genAssetMeta(gt.guardianPrivKeys, 1, 1, 1, bytes.fromhex("4523c3F29447d1f32AEa95BEBD00383c4640F1b4"), 1, 8, b"USDC", b"CircleCoin"))
  389. # pprint.pprint(self.parseVAA(q))
  390. # sys.exit(0)
  391. # vaa = self.parseVAA(bytes.fromhex("0100000001010001ca2fbf60ac6227d47dda4fe2e7bccc087f27d22170a212b9800da5b4cbf0d64c52deb2f65ce58be2267bf5b366437c267b5c7b795cd6cea1ac2fee8a1db3ad006225f801000000010001000000000000000000000000000000000000000000000000000000000000000400000000000000012000000000000000000000000000000000000000000000000000000000436f72650200000000000001beFA429d57cD18b7F8A4d91A2da9AB4AF05d0FBe"))
  392. # pprint.pprint(vaa)
  393. # vaa = self.parseVAA(bytes.fromhex("01000000010100c22ce0a3c995fca993cb0e91af74d745b6ec1a04b3adf0bb3e432746b3e2ab5e635b65d34d5148726cac10e84bf5932a7f21b9545c362bd512617aa980e0fbf40062607566000000010001000000000000000000000000000000000000000000000000000000000000000400000000000000012000000000000000000000000000000000000000000000000000000000436f72650200000000000101beFA429d57cD18b7F8A4d91A2da9AB4AF05d0FBe"))
  394. # pprint.pprint(vaa)
  395. # sys.exit(0)
  396. gt = GenTest(True)
  397. self.gt = gt
  398. self.setup_args()
  399. if self.args.testnet:
  400. self.testnet()
  401. else:
  402. self.devnet = True
  403. client = self.client = self.getAlgodClient()
  404. self.genTeal()
  405. self.vaa_verify = self.client.compile(get_vaa_verify())
  406. self.vaa_verify["lsig"] = LogicSig(base64.b64decode(self.vaa_verify["result"]))
  407. vaaLogs = []
  408. args = self.args
  409. if self.args.mnemonic:
  410. self.foundation = Account.FromMnemonic(self.args.mnemonic)
  411. if self.foundation == None:
  412. print("Generating the foundation account...")
  413. self.foundation = self.getTemporaryAccount(self.client)
  414. if self.foundation == None:
  415. print("We dont have a account? ")
  416. sys.exit(1)
  417. foundation = self.foundation
  418. seq = int(time.time())
  419. print("Creating the PortalCore app")
  420. self.coreid = self.createPortalCoreApp(client=client, sender=foundation)
  421. print("coreid = " + str(self.coreid) + " " + get_application_address(self.coreid))
  422. print("bootstrapping the guardian set...")
  423. bootVAA = bytes.fromhex(gt.genGuardianSetUpgrade(gt.guardianPrivKeys, 1, 1, seq, seq))
  424. self.bootGuardians(bootVAA, client, foundation, self.coreid)
  425. seq += 1
  426. print("grabbing a untrusted account")
  427. player = self.getTemporaryAccount(client)
  428. print(player.getAddress())
  429. print("")
  430. bal = self.getBalances(client, player.getAddress())
  431. pprint.pprint(bal)
  432. print("upgrading the the guardian set using untrusted account...")
  433. upgradeVAA = bytes.fromhex(gt.genGuardianSetUpgrade(gt.guardianPrivKeys, 1, 2, seq, seq))
  434. vaaLogs.append(["guardianUpgrade", upgradeVAA.hex()])
  435. self.submitVAA(upgradeVAA, client, player, self.coreid)
  436. bal = self.getBalances(client, player.getAddress())
  437. pprint.pprint(bal)
  438. seq += 1
  439. print("Create the token bridge")
  440. self.tokenid = self.createTokenBridgeApp(client, foundation)
  441. print("token bridge " + str(self.tokenid) + " address " + get_application_address(self.tokenid))
  442. ret = self.devnetUpgradeVAA()
  443. # pprint.pprint(ret)
  444. print("Submitting core")
  445. self.submitVAA(bytes.fromhex(ret[0]), self.client, foundation, self.coreid)
  446. print("Submitting token")
  447. self.submitVAA(bytes.fromhex(ret[1]), self.client, foundation, self.tokenid)
  448. print("successfully sent upgrade requests")
  449. for r in range(1, 6):
  450. print("Registering chain " + str(r))
  451. v = gt.genRegisterChain(gt.guardianPrivKeys, 2, seq, seq, r)
  452. vaa = bytes.fromhex(v)
  453. # pprint.pprint((v, self.parseVAA(vaa)))
  454. if r == 2:
  455. vaaLogs.append(["registerChain", v])
  456. self.submitVAA(vaa, client, player, self.tokenid)
  457. seq += 1
  458. bal = self.getBalances(client, player.getAddress())
  459. pprint.pprint(bal)
  460. print("Create a asset")
  461. attestVAA = bytes.fromhex(gt.genAssetMeta(gt.guardianPrivKeys, 2, seq, seq, bytes.fromhex("4523c3F29447d1f32AEa95BEBD00383c4640F1b4"), 1, 8, b"USDC", b"CircleCoin"))
  462. # paul - createWrappedOnAlgorand
  463. vaaLogs.append(["createWrappedOnAlgorand", attestVAA.hex()])
  464. self.submitVAA(attestVAA, client, player, self.tokenid)
  465. seq += 1
  466. p = self.parseVAA(attestVAA)
  467. chain_addr = self.optin(client, player, self.tokenid, p["FromChain"], p["Contract"])
  468. print("Create the same asset " + str(seq))
  469. # paul - updateWrappedOnAlgorand
  470. attestVAA = bytes.fromhex(gt.genAssetMeta(gt.guardianPrivKeys, 2, seq, seq, bytes.fromhex("4523c3F29447d1f32AEa95BEBD00383c4640F1b4"), 1, 8, b"USD2C", b"Circle2Coin"))
  471. self.submitVAA(attestVAA, client, player, self.tokenid)
  472. seq += 1
  473. print("Transfer the asset " + str(seq))
  474. transferVAA = bytes.fromhex(gt.genTransfer(gt.guardianPrivKeys, 1, 1, 1, 1, bytes.fromhex("4523c3F29447d1f32AEa95BEBD00383c4640F1b4"), 1, decode_address(player.getAddress()), 8, 0))
  475. # paul - redeemOnAlgorand
  476. vaaLogs.append(["redeemOnAlgorand", transferVAA.hex()])
  477. self.submitVAA(transferVAA, client, player, self.tokenid)
  478. seq += 1
  479. aid = client.account_info(player.getAddress())["assets"][0]["asset-id"]
  480. print("generate an attest of the asset we just received: " + str(aid))
  481. # paul - attestFromAlgorand
  482. self.testAttest(client, player, aid)
  483. print("Create the test app we will use to torture ourselves using a new player")
  484. player2 = self.getTemporaryAccount(client)
  485. print("player2 address " + player2.getAddress())
  486. player3 = self.getTemporaryAccount(client)
  487. print("player3 address " + player3.getAddress())
  488. self.testid = self.createTestApp(client, player2)
  489. print("testid " + str(self.testid) + " address " + get_application_address(self.testid))
  490. print("Sending a message payload to the core contract")
  491. sid = self.publishMessage(client, player, b"you also suck", self.testid)
  492. self.publishMessage(client, player2, b"second suck", self.testid)
  493. self.publishMessage(client, player3, b"last message", self.testid)
  494. print("Lets create a brand new non-wormhole asset and try to attest and send it out")
  495. self.testasset = self.createTestAsset(client, player2)
  496. print("test asset id: " + str(self.testasset))
  497. print("Now lets create an attest of ALGO")
  498. sid = self.testAttest(client, player2, 0)
  499. vaa = self.getVAA(client, player, sid, self.tokenid)
  500. v = self.parseVAA(bytes.fromhex(vaa))
  501. print("We got a " + v["Meta"])
  502. print("Lets try to create an attest for a non-wormhole thing with a huge number of decimals")
  503. # paul - attestFromAlgorand
  504. sid = self.testAttest(client, player2, self.testasset)
  505. print("... track down the generated VAA")
  506. vaa = self.getVAA(client, player, sid, self.tokenid)
  507. v = self.parseVAA(bytes.fromhex(vaa))
  508. print("We got a " + v["Meta"])
  509. pprint.pprint(self.getBalances(client, player.getAddress()))
  510. pprint.pprint(self.getBalances(client, player2.getAddress()))
  511. pprint.pprint(self.getBalances(client, player3.getAddress()))
  512. print("Lets transfer that asset to one of our other accounts... first lets create the vaa")
  513. # paul - transferFromAlgorand
  514. sid = self.transferFromAlgorand(client, player2, self.testasset, 100, decode_address(player3.getAddress()), 8, 0)
  515. print("... track down the generated VAA")
  516. vaa = self.getVAA(client, player, sid, self.tokenid)
  517. print(".. and lets pass that to player3")
  518. vaaLogs.append(["transferFromAlgorand", vaa])
  519. #pprint.pprint(vaaLogs)
  520. self.submitVAA(bytes.fromhex(vaa), client, player3, self.tokenid)
  521. pprint.pprint(["player", self.getBalances(client, player.getAddress())])
  522. pprint.pprint(["player2", self.getBalances(client, player2.getAddress())])
  523. pprint.pprint(["player3", self.getBalances(client, player3.getAddress())])
  524. # Lets split it into two parts... the payload and the fee
  525. print("Lets split it into two parts... the payload and the fee (400 should go to player, 600 should go to player3)")
  526. sid = self.transferFromAlgorand(client, player2, self.testasset, 1000, decode_address(player3.getAddress()), 8, 400)
  527. print("... track down the generated VAA")
  528. vaa = self.getVAA(client, player, sid, self.tokenid)
  529. # pprint.pprint(self.parseVAA(bytes.fromhex(vaa)))
  530. print(".. and lets pass that to player3 with fees being passed to player acting as a relayer (" + str(self.tokenid) + ")")
  531. self.submitVAA(bytes.fromhex(vaa), client, player, self.tokenid)
  532. pprint.pprint(["player", self.getBalances(client, player.getAddress())])
  533. pprint.pprint(["player2", self.getBalances(client, player2.getAddress())])
  534. pprint.pprint(["player3", self.getBalances(client, player3.getAddress())])
  535. # sys.exit(0)
  536. # Now it gets tricky, lets create a virgin account...
  537. pk, addr = account.generate_account()
  538. emptyAccount = Account(pk)
  539. print("How much is in the empty account? (" + addr + ")")
  540. pprint.pprint(self.getBalances(client, emptyAccount.getAddress()))
  541. # paul - transferFromAlgorand
  542. print("Lets transfer algo this time.... first lets create the vaa")
  543. sid = self.transferFromAlgorand(client, player2, 0, 1000000, decode_address(emptyAccount.getAddress()), 8, 0)
  544. print("... track down the generated VAA")
  545. vaa = self.getVAA(client, player, sid, self.tokenid)
  546. # pprint.pprint(vaa)
  547. print(".. and lets pass that to the empty account.. but use somebody else to relay since we cannot pay for it")
  548. # paul - redeemOnAlgorand
  549. self.submitVAA(bytes.fromhex(vaa), client, player, self.tokenid)
  550. print("=================================================")
  551. print("How much is in the source account now?")
  552. pprint.pprint(self.getBalances(client, player2.getAddress()))
  553. print("How much is in the empty account now?")
  554. pprint.pprint(self.getBalances(client, emptyAccount.getAddress()))
  555. print("How much is in the player3 account now?")
  556. pprint.pprint(self.getBalances(client, player3.getAddress()))
  557. print("Lets transfer more algo.. split 40/60 with the relayer.. going to player3")
  558. sid = self.transferFromAlgorand(client, player2, 0, 1000000, decode_address(player3.getAddress()), 8, 400000)
  559. print("... track down the generated VAA")
  560. vaa = self.getVAA(client, player, sid, self.tokenid)
  561. print(".. and lets pass that to player3.. but use the previously empty account to relay it")
  562. self.submitVAA(bytes.fromhex(vaa), client, emptyAccount, self.tokenid)
  563. print("How much is in the source account now?")
  564. pprint.pprint(self.getBalances(client, player2.getAddress()))
  565. print("How much is in the empty account now?")
  566. pprint.pprint(self.getBalances(client, emptyAccount.getAddress()))
  567. print("How much is in the player3 account now?")
  568. pprint.pprint(self.getBalances(client, player3.getAddress()))
  569. print("How about a payload3: " + self.testid.to_bytes(32, "big").hex())
  570. sid = self.transferFromAlgorand(client, player2, 0, 100, self.testid.to_bytes(32, "big"), 8, 0, b'hi mom')
  571. print("... track down the generated VAA")
  572. vaa = self.getVAA(client, player, sid, self.tokenid)
  573. pprint.pprint(self.parseVAA(bytes.fromhex(vaa)))
  574. print("testid balance before = ", self.getBalances(client, get_application_address(self.testid)))
  575. print(".. Lets let player3 relay it for us")
  576. self.submitVAA(bytes.fromhex(vaa), client, player3, self.tokenid)
  577. print("testid balance after = ", self.getBalances(client, get_application_address(self.testid)))
  578. # sys.exit(0)
  579. print(".. Ok, now it is time to up the message fees")
  580. bal = self.getBalances(client, get_application_address(self.coreid))
  581. print("core contract has " + str(bal) + " algo (" + get_application_address(self.coreid) + ")")
  582. print("core contract has a MessageFee set to " + str(self.getMessageFee()))
  583. seq += 1
  584. v = gt.genGSetFee(gt.guardianPrivKeys, 2, seq, seq, 2000000)
  585. self.submitVAA(bytes.fromhex(v), client, player, self.coreid)
  586. seq += 1
  587. print("core contract now has a MessageFee set to " + str(self.getMessageFee()))
  588. # v = gt.genGSetFee(gt.guardianPrivKeys, 2, seq, seq, 0)
  589. # self.submitVAA(bytes.fromhex(v), client, player, self.coreid)
  590. # seq += 1
  591. # print("core contract is back to " + str(self.getMessageFee()))
  592. print("Generating an attest.. This will cause a message to get published .. which should cause fees to get sent to the core contract")
  593. sid = self.testAttest(client, player2, self.testasset)
  594. print("... track down the generated VAA")
  595. vaa = self.getVAA(client, player, sid, self.tokenid)
  596. v = self.parseVAA(bytes.fromhex(vaa))
  597. print("We got a " + v["Meta"])
  598. bal = self.getBalances(client, get_application_address(self.coreid))
  599. print("core contract has " + str(bal) + " algo (" + get_application_address(self.coreid) + ")")
  600. # print("player account: " + player.getAddress())
  601. # pprint.pprint(client.account_info(player.getAddress()))
  602. # print("player2 account: " + player2.getAddress())
  603. # pprint.pprint(client.account_info(player2.getAddress()))
  604. # print("foundation account: " + foundation.getAddress())
  605. # pprint.pprint(client.account_info(foundation.getAddress()))
  606. #
  607. # print("core app: " + get_application_address(self.coreid))
  608. # pprint.pprint(client.account_info(get_application_address(self.coreid))),
  609. #
  610. # print("token app: " + get_application_address(self.tokenid))
  611. # pprint.pprint(client.account_info(get_application_address(self.tokenid))),
  612. #
  613. # print("asset app: " + chain_addr)
  614. # pprint.pprint(client.account_info(chain_addr))
  615. if __name__ == "__main__":
  616. core = AlgoTest()
  617. core.simple_test()