initial commit

test_wallet.py

@@ -0,0 +1,95 @@
+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
+
+
+async def run_test():
+    SERVER_ENDPOINT = "http://localhost:5000"
+    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())