[RpcClient] Policy API mismatch (neo-project#388)

* fixed-bug-1021 * Update src/RpcServer/RpcServer.SmartContract.cs * 😂 * NEO3: RPC client Policy API mismatch * update * update * update * UT * Format Co-authored-by: Shargon <shargon@gmail.com> Co-authored-by: Owen Zhang <38493437+superboyiii@users.noreply.github.com>
joeqian10 · Apr 7, 2021 · 0ced959 · 0ced959
1 parent 086877e
commit 0ced959
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 10 deletions.
diff --git a/src/RpcClient/PolicyAPI.cs b/src/RpcClient/PolicyAPI.cs
@@ -2,6 +2,7 @@
 using Neo.VM;
 using System.Linq;
 using System.Threading.Tasks;
+using Neo.IO.Json;
 
 namespace Neo.Network.RPC
 {
@@ -52,11 +53,10 @@ public async Task<long> GetFeePerByteAsync()
         /// Get Ploicy Blocked Accounts
         /// </summary>
         /// <returns></returns>
-        public async Task<UInt160[]> GetBlockedAccountsAsync()
+        public async Task<bool> IsBlockedAsync(UInt160 account)
         {
-            var result = await TestInvokeAsync(scriptHash, "getBlockedAccounts").ConfigureAwait(false);
-            var array = (VM.Types.Array)result.Stack.Single();
-            return array.Select(p => new UInt160(p.GetSpan().ToArray())).ToArray();
+            var result = await TestInvokeAsync(scriptHash, "isBlocked", new object[] { account }).ConfigureAwait(false);
+            return result.Stack.Single().GetBoolean();
         }
     }
 }
diff --git a/tests/Neo.Network.RPC.Tests/UT_PolicyAPI.cs b/tests/Neo.Network.RPC.Tests/UT_PolicyAPI.cs
@@ -57,13 +57,12 @@ public async Task TestGetFeePerByte()
         }
 
         [TestMethod]
-        public async Task TestGetBlockedAccounts()
+        public async Task TestIsBlocked()
         {
-            byte[] testScript = NativeContract.Policy.Hash.MakeScript("getBlockedAccounts");
-            UT_TransactionManager.MockInvokeScript(rpcClientMock, testScript, new ContractParameter { Type = ContractParameterType.Array, Value = new[] { new ContractParameter { Type = ContractParameterType.Hash160, Value = UInt160.Zero } } });
-
-            var result = await policyAPI.GetBlockedAccountsAsync();
-            Assert.AreEqual(UInt160.Zero, result[0]);
+            byte[] testScript = NativeContract.Policy.Hash.MakeScript("isBlocked", UInt160.Zero);
+            UT_TransactionManager.MockInvokeScript(rpcClientMock, testScript, new ContractParameter { Type = ContractParameterType.Boolean, Value = true });
+            var result = await policyAPI.IsBlockedAsync(UInt160.Zero);
+            Assert.AreEqual(true, result);
         }
     }
 }