mirror of
https://github.com/logos-blockchain/logos-blockchain-specs.git
synced 2026-01-09 16:43:10 +00:00
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from unittest import TestCase
|
|
from dataclasses import dataclass
|
|
|
|
from crypto import Field, prf
|
|
from note import InnerNote, PublicNote, SecretNote, nf_pk
|
|
from partial_transaction import PartialTransaction, InputNote, OutputNote
|
|
from transaction_bundle import TransactionBundle
|
|
|
|
import constraints
|
|
|
|
|
|
class TestTransfer(TestCase):
|
|
def test_1_to_1_transfer(self):
|
|
# Alice wants to transfer ownership of a note to Bob.
|
|
|
|
@dataclass
|
|
class User:
|
|
sk: Field
|
|
|
|
@property
|
|
def pk(self) -> Field:
|
|
return nf_pk(self.sk)
|
|
|
|
alice = User(sk=Field.random())
|
|
bob = User(sk=Field.random())
|
|
|
|
alices_note = SecretNote(
|
|
note=InnerNote(
|
|
value=100,
|
|
unit="NMO",
|
|
birth_constraint=constraints.Vacuous(),
|
|
death_constraints=[constraints.Vacuous()],
|
|
state=Field.zero(),
|
|
nonce=Field.random(),
|
|
rand=Field.random(),
|
|
),
|
|
nf_sk=alice.sk,
|
|
)
|
|
|
|
tx_rand = Field.random()
|
|
bobs_note = PublicNote(
|
|
note=InnerNote(
|
|
value=100,
|
|
unit="NMO",
|
|
birth_constraint=constraints.Vacuous(),
|
|
death_constraints=[constraints.Vacuous()],
|
|
state=Field.zero(),
|
|
nonce=Field.random(),
|
|
rand=Field.random(),
|
|
),
|
|
nf_pk=bob.pk,
|
|
)
|
|
# tx_output = Output(
|
|
# note=bobs_note,
|
|
# # TODO: why do we need an Output struct if we can
|
|
# # compute the balance and zero commitment form the
|
|
# # PublicNote itself?
|
|
# balance=bobs_note.balance(tx_rand),
|
|
# zero=bobs_note.zero(tx_rand),
|
|
# )
|
|
|
|
ptx = PartialTransaction(
|
|
inputs=[
|
|
InputNote(
|
|
note=alices_note,
|
|
death_proof=constraints.Vacuous().prove(),
|
|
)
|
|
],
|
|
outputs=[
|
|
OutputNote(
|
|
note=bobs_note,
|
|
birth_proof=constraints.Vacuous().prove(),
|
|
)
|
|
],
|
|
rand=tx_rand,
|
|
)
|
|
|
|
bundle = TransactionBundle(bundle=[ptx])
|
|
|
|
assert bundle.verify()
|