mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-21 11:04:19 +01:00
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
import asyncio
|
|
|
|
from core.helpers import async_unwrap
|
|
from wallet.wallet import Wallet as Wallet1
|
|
from wallet.wallet import Wallet as Wallet2
|
|
from wallet.migrations import m001_initial
|
|
|
|
|
|
SERVER_ENDPOINT = "http://localhost:5000"
|
|
|
|
|
|
async def run_test():
|
|
wallet1 = Wallet1(SERVER_ENDPOINT, "data/wallet1")
|
|
await m001_initial(wallet1.db)
|
|
wallet1.status()
|
|
|
|
wallet2 = Wallet1(SERVER_ENDPOINT, "data/wallet2")
|
|
await m001_initial(wallet2.db)
|
|
wallet2.status()
|
|
|
|
proofs = []
|
|
|
|
# Mint a proof of promise. We obtain a proof for 64 coins
|
|
proofs += await wallet1.mint(64)
|
|
print(proofs)
|
|
assert wallet1.balance == 64
|
|
wallet1.status()
|
|
|
|
# Mint an odd amount (not in 2^n)
|
|
proofs += await wallet1.mint(63)
|
|
assert wallet1.balance == 64 + 63
|
|
|
|
# Error: We try to split by amount higher than available
|
|
w1_fst_proofs, w1_snd_proofs = await wallet1.split(wallet1.proofs, 65)
|
|
# assert w1_fst_proofs == []
|
|
# assert w1_snd_proofs == []
|
|
assert wallet1.balance == 63 + 64
|
|
wallet1.status()
|
|
|
|
# Error: We try to double-spend by providing a valid proof twice
|
|
w1_fst_proofs, w1_snd_proofs = await wallet1.split(wallet1.proofs + proofs, 20)
|
|
assert w1_fst_proofs == []
|
|
assert w1_snd_proofs == []
|
|
assert wallet1.balance == 63 + 64
|
|
wallet1.status()
|
|
|
|
print("PROOFs")
|
|
print(proofs)
|
|
w1_fst_proofs, w1_snd_proofs = await wallet1.split(wallet1.proofs, 20)
|
|
# we expect 44 and 20 -> [4, 8, 32], [4, 16]
|
|
print(w1_fst_proofs)
|
|
print(w1_snd_proofs)
|
|
# assert [p["amount"] for p in w1_fst_proofs] == [4, 8, 32]
|
|
assert [p["amount"] for p in w1_snd_proofs] == [4, 16]
|
|
assert wallet1.balance == 63 + 64
|
|
wallet1.status()
|
|
|
|
# Error: We try to double-spend and it fails
|
|
w1_fst_proofs2_fails, w1_snd_proofs2_fails = await wallet1.split([proofs[0]], 10)
|
|
assert w1_fst_proofs2_fails == []
|
|
assert w1_snd_proofs2_fails == []
|
|
assert wallet1.balance == 63 + 64
|
|
wallet1.status()
|
|
|
|
# Redeem the tokens in wallet2
|
|
w2_fst_proofs, w2_snd_proofs = await wallet2.redeem(w1_snd_proofs)
|
|
print(w2_fst_proofs)
|
|
print(w2_snd_proofs)
|
|
assert wallet1.balance == 63 + 64
|
|
assert wallet2.balance == 20
|
|
wallet2.status()
|
|
|
|
# wallet1 invalidates his proofs
|
|
await wallet1.invalidate(w1_snd_proofs)
|
|
assert wallet1.balance == 63 + 64 - 20
|
|
wallet1.status()
|
|
|
|
w1_fst_proofs2, w1_snd_proofs2 = await wallet1.split(w1_fst_proofs, 5)
|
|
# we expect 15 and 5 -> [1, 2, 4, 8], [1, 4]
|
|
print(w1_fst_proofs2)
|
|
print(w1_snd_proofs2)
|
|
assert wallet1.balance == 63 + 64 - 20
|
|
wallet1.status()
|
|
|
|
# Error: We try to double-spend and it fails
|
|
w1_fst_proofs2, w1_snd_proofs2 = await wallet1.split(w1_snd_proofs, 5)
|
|
assert w1_fst_proofs2 == []
|
|
assert w1_snd_proofs2 == []
|
|
assert wallet1.balance == 63 + 64 - 20
|
|
wallet1.status()
|
|
|
|
assert wallet1.proof_amounts() == [1, 2, 4, 4, 32, 64]
|
|
assert wallet2.proof_amounts() == [4, 16]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
async_unwrap(run_test())
|