Add ut

neo-project · Feb 19, 2024 · 41914d8 · 41914d8
1 parent 48d2349
commit 41914d8
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/tests/Neo.UnitTests/SmartContract/Native/UT_NativeContract.cs b/tests/Neo.UnitTests/SmartContract/Native/UT_NativeContract.cs
@@ -11,6 +11,7 @@
 
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using Neo.SmartContract.Native;
+using System.IO;
 
 namespace Neo.UnitTests.SmartContract.Native
 {
@@ -22,5 +23,25 @@ public void TestGetContract()
         {
             Assert.IsTrue(NativeContract.NEO == NativeContract.GetContract(NativeContract.NEO.Hash));
         }
+
+        [TestMethod]
+        public void TestIsInitializeBlock()
+        {
+            string json = UT_ProtocolSettings.CreateHKSettings("\"HF_Cockatrice\": 20");
+
+            var file = Path.GetTempFileName();
+            File.WriteAllText(file, json);
+            ProtocolSettings settings = ProtocolSettings.Load(file, false);
+            File.Delete(file);
+
+            Assert.IsTrue(NativeContract.CryptoLib.IsInitializeBlock(settings, 0, out var hf));
+            Assert.IsNull(hf);
+
+            Assert.IsFalse(NativeContract.CryptoLib.IsInitializeBlock(settings, 1, out hf));
+            Assert.IsNull(hf);
+
+            Assert.IsTrue(NativeContract.CryptoLib.IsInitializeBlock(settings, 20, out hf));
+            Assert.AreEqual(Hardfork.HF_Cockatrice, hf);
+        }
     }
 }
diff --git a/tests/Neo.UnitTests/UT_ProtocolSettings.cs b/tests/Neo.UnitTests/UT_ProtocolSettings.cs
@@ -124,7 +124,7 @@ public void HardForkTestAMoreThanB()
             File.Delete(file);
         }
 
-        private static string CreateHKSettings(string hf)
+        internal static string CreateHKSettings(string hf)
         {
             return @"
 {