add rust_port tests

Alrighttt · Mar 22, 2024 · 6c19657 · 6c19657
1 parent 89e9456
commit 6c19657
Showing 1 changed file with 259 additions and 0 deletions.
diff --git a/types/rust_port_test.go b/types/rust_port_test.go
@@ -0,0 +1,259 @@
+package types
+
+import (
+	"testing"
+	"time"
+)
+
+/*
+These tests serve as a sanity check that the Rust port's encoding and hashing functions are working as expected.
+
+If any tests within this file fail at any point in the future, it's an indictation that the Rust port must be updated.
+
+Verbose as possible to enable quickly identifying the source of any discrepancies.
+*/
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L189
+func TestStandardUnlockHash(t *testing.T) {
+	pk := PublicKey{1, 2, 3}
+	p := SpendPolicy{PolicyTypeUnlockConditions(StandardUnlockConditions(pk))}
+	if p.Address().String() != "addr:72b0762b382d4c251af5ae25b6777d908726d75962e5224f98d7f619bb39515dd64b9a56043a" {
+		t.Fatal("wrong address:", p, p.Address())
+	} else if StandardUnlockHash(pk) != p.Address() {
+		t.Fatal("StandardUnlockHash differs from Policy.Address")
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/spend_policy.rs#L202
+func TestUnlockConditions2of2Multisig(t *testing.T) {
+	uc := UnlockConditions{
+		Timelock: 0,
+		PublicKeys: []UnlockKey{
+			PublicKey{1, 2, 3}.UnlockKey(),
+			PublicKey{1, 1, 1}.UnlockKey()},
+		SignaturesRequired: 2,
+	}
+	if unlockConditionsRoot(uc).String() != "addr:1e94357817d236167e54970a8c08bbd41b37bfceeeb52f6c1ce6dd01d50ea1e73a7c081d3178" {
+		t.Fatal("wrong address:", uc, unlockConditionsRoot(uc).String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/spend_policy.rs#L219
+func TestUnlockConditions1of2Multisig(t *testing.T) {
+	uc := UnlockConditions{
+		Timelock: 0,
+		PublicKeys: []UnlockKey{
+			PublicKey{1, 2, 3}.UnlockKey(),
+			PublicKey{1, 1, 1}.UnlockKey()},
+		SignaturesRequired: 1,
+	}
+	if unlockConditionsRoot(uc).String() != "addr:d7f84e3423da09d111a17f64290c8d05e1cbe4cab2b6bed49e3a4d2f659f0585264e9181a51a" {
+		t.Fatal("wrong address:", uc, unlockConditionsRoot(uc).String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L45
+func TestEncoderDefault(t *testing.T) {
+	h := NewHasher()
+	myhash := h.Sum()
+	if myhash.String() != "h:0e5751c026e543b2e8ab2eb06099daa1d1e5df47778f7787faab45cdf12fe3a8" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L53
+func TestEncoderWriteByes(t *testing.T) {
+	h := NewHasher()
+	h.E.WriteBytes([]byte{1, 2, 3, 4})
+	myhash := h.Sum()
+	if myhash.String() != "h:d4a72b52e2e1f40e20ee40ea6d5080a1b1f76164786defbb7691a4427f3388f5" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L63
+func TestEncoderWriteUint8(t *testing.T) {
+	h := NewHasher()
+	h.E.WriteUint8(1)
+	myhash := h.Sum()
+	if myhash.String() != "h:ee155ace9c40292074cb6aff8c9ccdd273c81648ff1149ef36bcea6ebb8a3e25" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L73
+func TestEncoderWriteUint64(t *testing.T) {
+	h := NewHasher()
+	h.E.WriteUint64(1)
+	myhash := h.Sum()
+	if myhash.String() != "h:1dbd7d0b561a41d23c2a469ad42fbd70d5438bae826f6fd607413190c37c363b" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L83
+func TestEncoderWriteDistinguisher(t *testing.T) {
+	h := NewHasher()
+	h.WriteDistinguisher("test")
+	myhash := h.Sum()
+	if myhash.String() != "h:25fb524721bf98a9a1233a53c40e7e198971b003bf23c24f59d547a1bb837f9c" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L93
+func TestEmcoderWriteBool(t *testing.T) {
+	h := NewHasher()
+	h.E.WriteBool(true)
+	myhash := h.Sum()
+	if myhash.String() != "h:ee155ace9c40292074cb6aff8c9ccdd273c81648ff1149ef36bcea6ebb8a3e25" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L103
+func TestReset(t *testing.T) {
+	h := NewHasher()
+	h.E.WriteBool(true)
+	myhash := h.Sum()
+	if myhash.String() != "h:ee155ace9c40292074cb6aff8c9ccdd273c81648ff1149ef36bcea6ebb8a3e25" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+	h.Reset()
+	h.E.WriteBool(false)
+	myhash = h.Sum()
+	if myhash.String() != "h:03170a2e7597b7b7e3d84c05391d139a62b157e78786d8c082f29dcf4c111314" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/b6af96ef5a2f75b4ce3d1308e0f8b9757ec15a95/mm2src/coins/sia/encoding.rs#L120
+func TestEncoderWriteComplex(t *testing.T) {
+	h := NewHasher()
+	h.WriteDistinguisher("test")
+	h.E.WriteBool(true)
+	h.E.WriteUint8(1)
+	h.E.WriteBytes([]byte{1, 2, 3, 4})
+	myhash := h.Sum()
+	if myhash.String() != "h:b66d7a9bef9fb303fe0e41f6b5c5af410303e428c4ff9231f6eb381248693221" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L239
+func TestPolicyAboveEncodeHash(t *testing.T) {
+	h := NewHasher()
+
+	policy := PolicyAbove(1)
+	policy.EncodeTo(h.E)
+
+	myaddress := policy.Address()
+	myhash := h.Sum()
+	if myhash.String() != "h:bebf6cbdfb440a92e3e5d832ac30fe5d226ff6b352ed3a9398b7d35f086a8ab6" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+	if myaddress.String() != "addr:188b997bb99dee13e95f92c3ea150bd76b3ec72e5ba57b0d57439a1a6e2865e9b25ea5d1825e" {
+		t.Fatal("wrong address:", myaddress.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L253
+func TestPolicyAfterEncodeHash(t *testing.T) {
+	h := NewHasher()
+
+	time := time.Unix(int64(1), 0)
+	policy := PolicyAfter(time)
+	policy.EncodeTo(h.E)
+
+	myhash := h.Sum()
+	myaddress := policy.Address()
+
+	if myhash.String() != "h:07b0f28eafd87a082ad11dc4724e1c491821260821a30bec68254444f97d9311" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+	if myaddress.String() != "addr:60c74e0ce5cede0f13f83b0132cb195c995bc7688c9fac34bbf2b14e14394b8bbe2991bc017f" {
+		t.Fatal("wrong address:", myaddress.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L267
+func TestPolicyPublicKeyEncodeHash(t *testing.T) {
+	h := NewHasher()
+
+	policy := PolicyPublicKey(PublicKey{1, 2, 3})
+	policy.EncodeTo(h.E)
+
+	myhash := h.Sum()
+	myaddress := policy.Address()
+
+	if myhash.String() != "h:4355c8f80f6e5a98b70c9c2f9a22f17747989b4744783c90439b2b034f698bfe" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+	if myaddress.String() != "addr:55a7793237722c6df8222fd512063cb74228085ef1805c5184713648c159b919ac792fbad0e1" {
+		t.Fatal("wrong address:", myaddress.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L285
+func TestPolicyHash(t *testing.T) {
+	h := NewHasher()
+
+	policy := PolicyHash(Hash256{1, 2, 3})
+	policy.EncodeTo(h.E)
+
+	myhash := h.Sum()
+	myaddress := policy.Address()
+
+	if myhash.String() != "h:9938967aefa6cbecc1f1620d2df5170d6811d4b2f47a879b621c1099a3b0628a" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+	if myaddress.String() != "addr:a4d5a06d8d3c2e45aa26627858ce8e881505ae3c9d122a1d282c7824163751936cffb347e435" {
+		t.Fatal("wrong address:", myaddress.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L301
+func TestPolicyThreshold(t *testing.T) {
+	h := NewHasher()
+
+	policy := PolicyThreshold(1, []SpendPolicy{
+		PolicyAbove(1),
+		PolicyAfter(time.Unix(int64(1), 0)),
+	})
+	policy.EncodeTo(h.E)
+
+	myhash := h.Sum()
+	myaddress := policy.Address()
+
+	if myhash.String() != "h:7d792df6cd0b5e0f795287b3bf4087bbcc4c1bd0c52880a552cdda3e5e33d802" {
+		t.Fatal("wrong hash:", myhash.String())
+	}
+	if myaddress.String() != "addr:4179b53aba165e46e4c85b3c8766bb758fb6f0bfa5721550b81981a3ec38efc460557dc1ded4" {
+		t.Fatal("wrong address:", myaddress.String())
+	}
+}
+
+// https://github.com/KomodoPlatform/komodo-defi-framework/blob/d180505b43f8167bd733263e73804ea60d4c1632/mm2src/coins/sia/spend_policy.rs#L319
+func TestPolicyUnlockConditionEncodeSpecialCase(t *testing.T) {
+	pubkey := PublicKey{1, 2, 3}
+	unlock_condition := PolicyTypeUnlockConditions{
+		PublicKeys:         []UnlockKey{pubkey.UnlockKey()},
+		SignaturesRequired: 1,
+		Timelock:           0,
+	}
+	policy := PolicyThreshold(1, []SpendPolicy{
+		{unlock_condition},
+	})
+
+	// Unlock condition SpendPolicy has a special condition for v1 comaptibility if it is not within a Threshold
+	originalUnlockConditions := UnlockConditions(unlock_condition)
+	uc_address := originalUnlockConditions.UnlockHash()
+	if uc_address.String() != "addr:72b0762b382d4c251af5ae25b6777d908726d75962e5224f98d7f619bb39515dd64b9a56043a" {
+		t.Fatal("wrong address:", uc_address.String())
+	}
+
+	uc_inside_threshold_address := policy.Address()
+	if uc_inside_threshold_address.String() != "addr:1498a58c843ce66740e52421632d67a0f6991ea96db1fc97c29e46f89ae56e3534078876331d" {
+		t.Fatal("wrong address:", uc_inside_threshold_address.String())
+	}
+}