[Tests] Add Get/Set payload unit-test for ProUpReg txes

random-zebra · random-zebra · commit ceeaf310c303 · 2021-07-06T16:01:02.000+02:00
diff --git a/src/test/evo_specialtx_tests.cpp b/src/test/evo_specialtx_tests.cpp
@@ -58,6 +58,18 @@ static ProUpServPL GetRandomProUpServPayload()
     return pl;
 }
 
+static ProUpRegPL GetRandomProUpRegPayload()
+{
+    ProUpRegPL pl;
+    pl.proTxHash = GetRandHash();
+    pl.keyIDOperator = GetRandomKeyID();
+    pl.keyIDVoting = GetRandomKeyID();
+    pl.scriptPayout = GetRandomScript();
+    pl.inputsHash = GetRandHash();
+    pl.vchSig = InsecureRandBytes(63);
+    return pl;
+}
+
 BOOST_AUTO_TEST_CASE(protx_validation_test)
 {
     CMutableTransaction mtx;
@@ -146,6 +158,22 @@ BOOST_AUTO_TEST_CASE(proupserv_setpayload_test)
     BOOST_CHECK(pl.vchSig == pl2.vchSig);
 }
 
+BOOST_AUTO_TEST_CASE(proupreg_setpayload_test)
+{
+    const ProUpRegPL& pl = GetRandomProUpRegPayload();
+
+    CMutableTransaction mtx;
+    SetTxPayload(mtx, pl);
+    ProUpRegPL pl2;
+    BOOST_CHECK(GetTxPayload(mtx, pl2));
+    BOOST_CHECK(pl.proTxHash == pl2.proTxHash);
+    BOOST_CHECK(pl.keyIDOperator == pl2.keyIDOperator);
+    BOOST_CHECK(pl.keyIDVoting == pl2.keyIDVoting);
+    BOOST_CHECK(pl.scriptPayout == pl2.scriptPayout);
+    BOOST_CHECK(pl.inputsHash == pl2.inputsHash);
+    BOOST_CHECK(pl.vchSig == pl2.vchSig);
+}
+
 BOOST_AUTO_TEST_CASE(proreg_checkstringsig_test)
 {
     ProRegPL pl = GetRandomProRegPayload();
