54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
# (c) Copyright 2025 by Coinkite Inc. This file is covered by license found in COPYING-CC.
|
|
#
|
|
from uio import BytesIO
|
|
from serializations import ser_push_data, ser_string_vector, deser_string_vector
|
|
from serializations import ser_compact_size, deser_compact_size, disassemble
|
|
|
|
test_data = [
|
|
# data, result
|
|
(55*b"a", b'7aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'),
|
|
(75*b"a", b'Kaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'),
|
|
(76*b"a", b'LLaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'),
|
|
(77*b"a", b'LMaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'),
|
|
(254*b"a", b'L\xfe' + (254 * b"a")),
|
|
(255*b"a", b'L\xff' + (255 * b"a")),
|
|
(256*b"a", b'M\x00\x01' + (256 * b"a")),
|
|
(500*b"a", b'M\xf4\x01' + (500 * b"a")),
|
|
(65535*b"a", b'M\xff\xff' + (65535 * b"a")),
|
|
]
|
|
|
|
for i, (data, result) in enumerate(test_data):
|
|
assert ser_push_data(data) == result, i
|
|
d, _ = list(disassemble(result))[0]
|
|
assert d == data
|
|
|
|
try:
|
|
# PUSHDATA 4 not implemented
|
|
ser_push_data(65536 * b"a")
|
|
raise RuntimeError
|
|
except AssertionError: pass
|
|
|
|
# test serialization/deserialization
|
|
# all M/N combinations
|
|
V = range(1, 16)
|
|
for i, v1 in enumerate(V):
|
|
for j in range(i+1, len(V)):
|
|
M, N = v1, V[j]
|
|
# number of pubkeys times 1 pushdata + 33 pubkey = 34 * N
|
|
# +1 M
|
|
# +1 N
|
|
# +1 OP_CHECKMULTISIG
|
|
ms_script_len = (34 * N) + 1 + 1 + 1
|
|
vec = [b"\x00"] + (M * [71*b"s"]) + [ms_script_len*b"w"]
|
|
assert vec == deser_string_vector(BytesIO(ser_string_vector(vec)))
|
|
|
|
|
|
for i in [253, 0x10000, 0x100000000]:
|
|
for j in [-1, 0, 1]:
|
|
num = i + j
|
|
x = ser_compact_size(num)
|
|
|
|
assert num == deser_compact_size(BytesIO(x))
|
|
|
|
# EOF
|