split test for kmd and assetchain

KomodoPlatform · May 30, 2023 · 8d0cc33 · 8d0cc33
1 parent e065ddf
commit 8d0cc33
Showing 1 changed file with 33 additions and 6 deletions.
diff --git a/src/test-komodo/test_notary.cpp b/src/test-komodo/test_notary.cpp
@@ -166,8 +166,8 @@ TEST(TestNotary, ElectedNotary)
     EXPECT_EQ(numnotaries, 64);
 }
 
-// season 7 tests:
-TEST(TestNotary, KomodoNotaries_S7)
+// season 7 tests for KMD:
+TEST(TestNotary, KomodoNotaries_S7_KMD)
 {
     uint8_t pubkeys[64][33];
 
@@ -178,21 +178,48 @@ TEST(TestNotary, KomodoNotaries_S7)
     EXPECT_EQ( getkmdseason(3484958+1), 8);
     EXPECT_EQ( getkmdseason(8113400), 8);
     EXPECT_EQ( getkmdseason(8113400+1), 0);
-    int32_t result = komodo_notaries(pubkeys, 3484958+1, 0);
+    int32_t result1 = komodo_notaries(pubkeys, 3484958+1, 0);
+    EXPECT_EQ(result1, 64);
     EXPECT_EQ( my_key(pubkeys[0]), my_key("03955c7999538cee313bf196a7df59db208c651f6a5a1b0eed94732ba753b4f3f4"));
     EXPECT_EQ( my_key(pubkeys[63]), my_key("02f9a7b49282885cd03969f1f5478287497bc8edfceee9eac676053c107c5fcdaf"));
 
-    // this should work for KMD net
+    // try wrong pubkey
+    int32_t numnotaries;
+    my_key wrong_pk("02ebfc784a4ba768aad88d44d1045d240d47b26e248cafaf1c5169a42d7a61d344");
+    uint8_t pubkey[33];
+    wrong_pk.fill(pubkey);
+    int32_t result2 = komodo_electednotary(&numnotaries, pubkey, 3484958+1, 0);
+    EXPECT_EQ(result2, -1);
+    EXPECT_EQ(numnotaries, 64);
+
+    // cleanup
+    komodo_notaries_uninit();
+}
+
+// season 7 tests for asset chain:
+TEST(TestNotary, KomodoNotaries_S7_AS)
+{
+    uint8_t pubkeys[64][33];
+
+    SelectParams(CBaseChainParams::MAIN);
+    komodo_notaries_uninit();
+    chainName = assetchain("MYASSET");
+    EXPECT_EQ( getacseason(1688132253+1), 8);
     EXPECT_EQ( getacseason(1951328000), 8);
     EXPECT_EQ( getacseason(1951328001), 0);
 
+    int32_t result1 = komodo_notaries(pubkeys, 0, 1688132253+1);
+    EXPECT_EQ(result1, 64);
+    EXPECT_EQ( my_key(pubkeys[0]), my_key("03955c7999538cee313bf196a7df59db208c651f6a5a1b0eed94732ba753b4f3f4"));
+    EXPECT_EQ( my_key(pubkeys[63]), my_key("02f9a7b49282885cd03969f1f5478287497bc8edfceee9eac676053c107c5fcdaf"));
+
     // try wrong pubkey
     int32_t numnotaries;
     my_key wrong_pk("02ebfc784a4ba768aad88d44d1045d240d47b26e248cafaf1c5169a42d7a61d344");
     uint8_t pubkey[33];
     wrong_pk.fill(pubkey);
-    result = komodo_electednotary(&numnotaries, pubkey, 3484958+1, 0);
-    EXPECT_EQ(result, -1);
+    int32_t result2 = komodo_electednotary(&numnotaries, pubkey, 3484958+1, 1688132253+1);
+    EXPECT_EQ(result2, -1);
     EXPECT_EQ(numnotaries, 64);
 
     // cleanup