test RpcServer.Utilities, .SmartContract and .Wallet

src/Plugins/RpcServer/RpcServer.Utilities.cs

@@ -18,7 +18,7 @@ namespace Neo.Plugins.RpcServer
     partial class RpcServer
     {
         [RpcMethod]
-        protected virtual JToken ListPlugins(JArray _params)
+        protected internal virtual JToken ListPlugins(JArray _params)
         {
             return new JArray(Plugin.Plugins
                 .OrderBy(u => u.Name)
@@ -34,7 +34,7 @@ protected virtual JToken ListPlugins(JArray _params)
         }
 
         [RpcMethod]
-        protected virtual JToken ValidateAddress(JArray _params)
+        protected internal virtual JToken ValidateAddress(JArray _params)
         {
             string address = Result.Ok_Or(() => _params[0].AsString(), RpcError.InvalidParams.WithData($"Invlid address format: {_params[0]}"));
             JObject json = new();

tests/Neo.Plugins.RpcServer.Tests/UT_RpcServer.SmartContract.cs

@@ -0,0 +1,139 @@
+// Copyright (C) 2015-2024 The Neo Project.
+//
+// UT_RpcServer.Wallet.cs file belongs to the neo project and is free
+// software distributed under the MIT software license, see the
+// accompanying file LICENSE in the main directory of the
+// repository or http://www.opensource.org/licenses/mit-license.php
+// for more details.
+//
+// Redistribution and use in source and binary forms with or without
+// modifications are permitted.
+
+using FluentAssertions;
+using Microsoft.AspNetCore.Http;
+using Microsoft.VisualStudio.TestPlatform.ObjectModel.DataCollection;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Neo.IO;
+using Neo.Json;
+using Neo.Network.P2P.Payloads;
+using Neo.SmartContract;
+using Neo.SmartContract.Native;
+using Neo.UnitTests;
+using Neo.UnitTests.Extensions;
+using Neo.Wallets;
+using System;
+using System.IO;
+using System.Linq;
+
+namespace Neo.Plugins.RpcServer.Tests;
+
+public partial class UT_RpcServer
+{
+    static readonly string NeoScriptHash = "0xef4073a0f2b305a38ec4050e4d3d28bc40ea63f5";
+    static readonly string GasScriptHash = "0xd2a4cff31913016155e38e474a2c06d08be276cf";
+    static readonly string NeoTotalSupplyScript = "wh8MC3RvdGFsU3VwcGx5DBT1Y\u002BpAvCg9TQ4FxI6jBbPyoHNA70FifVtS";
+    static readonly UInt160 ValidatorScriptHash = Contract
+        .CreateSignatureRedeemScript(TestProtocolSettings.SoleNode.StandbyCommittee[0])
+        .ToScriptHash();
+    static readonly string ValidatorAddress = ValidatorScriptHash.ToAddress(ProtocolSettings.Default.AddressVersion);
+    static readonly UInt160 MultisigScriptHash = Contract
+        .CreateMultiSigRedeemScript(1, TestProtocolSettings.SoleNode.StandbyCommittee)
+        .ToScriptHash();
+    static readonly string MultisigAddress = MultisigScriptHash.ToAddress(ProtocolSettings.Default.AddressVersion);
+
+    static readonly JArray validatorSigner = [new JObject()
+    {
+        ["account"] = ValidatorScriptHash.ToString(),
+        ["scopes"] = "CalledByEntry",
+    }];
+    static readonly JArray multisigSigner = [new JObject()
+    {
+        ["account"] = MultisigScriptHash.ToString(),
+        ["scopes"] = "CalledByEntry",
+    }];
+
+    [TestMethod]
+    public void TestInvokeFunction()
+    {
+        JObject resp = (JObject)_rpcServer.InvokeFunction(new JArray(NeoScriptHash, "totalSupply", new JArray([]), validatorSigner, true));
+        Assert.AreEqual(resp.Count, 7);
+        Assert.AreEqual(resp["script"], NeoTotalSupplyScript);
+        Assert.IsTrue(resp.ContainsProperty("gasconsumed"));
+        Assert.IsTrue(resp.ContainsProperty("diagnostics"));
+        Assert.AreEqual(resp["diagnostics"]["invokedcontracts"]["call"][0]["hash"], NeoScriptHash);
+        Assert.IsTrue(((JArray)resp["diagnostics"]["storagechanges"]).Count == 0);
+        Assert.AreEqual(resp["state"], "HALT");
+        Assert.AreEqual(resp["exception"], null);
+        Assert.AreEqual(((JArray)resp["notifications"]).Count, 0);
+        Assert.AreEqual(resp["stack"][0]["type"], "Integer");
+        Assert.AreEqual(resp["stack"][0]["value"], "100000000");
+
+        // This call triggers not only NEO but also unclaimed GAS
+        resp = (JObject)_rpcServer.InvokeFunction(new JArray(NeoScriptHash, "transfer", new JArray([
+            new JObject() { ["type"] = "Hash160", ["value"] = MultisigScriptHash.ToString() },
+            new JObject() { ["type"] = "Hash160", ["value"] = ValidatorScriptHash.ToString() },
+            new JObject() { ["type"] = "Integer", ["value"] = "1" },
+            new JObject() { ["type"] = "Any" },
+        ]), multisigSigner, true));
+        Assert.AreEqual(resp.Count, 7);
+        Assert.IsTrue(resp.ContainsProperty("gasconsumed"));
+        Assert.IsTrue(resp.ContainsProperty("diagnostics"));
+        Assert.AreEqual(resp["diagnostics"]["invokedcontracts"]["call"][0]["hash"], NeoScriptHash);
+        Assert.IsTrue(((JArray)resp["diagnostics"]["storagechanges"]).Count == 4);
+        Assert.AreEqual(resp["state"], "HALT");
+        Assert.AreEqual(resp["exception"], null);
+        JArray notifications = (JArray)resp["notifications"];
+        Assert.AreEqual(notifications.Count, 2);
+        Assert.AreEqual(notifications[0]["eventname"].AsString(), "Transfer");
+        Assert.AreEqual(notifications[0]["contract"].AsString(), NeoScriptHash);
+        Assert.AreEqual(((JArray)resp["notifications"])[0]["state"]["value"][2]["value"], "1");
+        Assert.AreEqual(notifications[1]["eventname"].AsString(), "Transfer");
+        Assert.AreEqual(notifications[1]["contract"].AsString(), GasScriptHash);
+        Assert.AreEqual(((JArray)resp["notifications"])[1]["state"]["value"][2]["value"], "50000000");
+    }
+
+    [TestMethod]
+    public void TestInvokeScript()
+    {
+        JObject resp = (JObject)_rpcServer.InvokeScript(new JArray(NeoTotalSupplyScript, validatorSigner, true));
+        Assert.AreEqual(resp.Count, 7);
+        Assert.IsTrue(resp.ContainsProperty("gasconsumed"));
+        Assert.IsTrue(resp.ContainsProperty("diagnostics"));
+        Assert.AreEqual(resp["diagnostics"]["invokedcontracts"]["call"][0]["hash"], NeoScriptHash);
+        Assert.AreEqual(resp["state"], "HALT");
+        Assert.AreEqual(resp["exception"], null);
+        Assert.AreEqual(((JArray)resp["notifications"]).Count, 0);
+        Assert.AreEqual(resp["stack"][0]["type"], "Integer");
+        Assert.AreEqual(resp["stack"][0]["value"], "100000000");
+    }
+
+    [TestMethod]
+    public void TestTraverseIterator()
+    {
+        JObject resp = (JObject)_rpcServer.InvokeFunction(new JArray(NeoScriptHash, "getAllCandidates", new JArray([]), validatorSigner, true));
+        string sessionId = resp["session"].AsString();
+        string iteratorId = resp["stack"][0]["id"].AsString();
+        JArray respArray = (JArray)_rpcServer.TraverseIterator([sessionId, iteratorId, 100]);
+        Assert.AreEqual(respArray.Count, 0);
+        _rpcServer.TerminateSession([sessionId]);
+        try
+        {
+            respArray = (JArray)_rpcServer.TraverseIterator([sessionId, iteratorId, 100]);
+        }
+        catch (RpcException e)
+        {
+            Assert.AreEqual(e.Message, "Unknown session");
+        }
+    }
+
+    [TestMethod]
+    public void TestGetUnclaimedGas()
+    {
+        JObject resp = (JObject)_rpcServer.GetUnclaimedGas([MultisigAddress]);
+        Assert.AreEqual(resp["unclaimed"], "50000000");
+        Assert.AreEqual(resp["address"], MultisigAddress);
+        resp = (JObject)_rpcServer.GetUnclaimedGas([ValidatorAddress]);
+        Assert.AreEqual(resp["unclaimed"], "0");
+        Assert.AreEqual(resp["address"], ValidatorAddress);
+    }
+}

tests/Neo.Plugins.RpcServer.Tests/UT_RpcServer.Utilities.cs

@@ -0,0 +1,54 @@
+// Copyright (C) 2015-2024 The Neo Project.
+//
+// UT_RpcServer.Wallet.cs file belongs to the neo project and is free
+// software distributed under the MIT software license, see the
+// accompanying file LICENSE in the main directory of the
+// repository or http://www.opensource.org/licenses/mit-license.php
+// for more details.
+//
+// Redistribution and use in source and binary forms with or without
+// modifications are permitted.
+
+using FluentAssertions;
+using Microsoft.AspNetCore.Http;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Neo.IO;
+using Neo.Json;
+using Neo.Network.P2P.Payloads;
+using Neo.SmartContract;
+using Neo.SmartContract.Native;
+using Neo.UnitTests;
+using Neo.UnitTests.Extensions;
+using System;
+using System.IO;
+using System.Linq;
+
+namespace Neo.Plugins.RpcServer.Tests;
+
+public partial class UT_RpcServer
+{
+    [TestMethod]
+    public void TestListPlugins()
+    {
+        JArray resp = (JArray)_rpcServer.ListPlugins([]);
+        Assert.AreEqual(resp.Count, 0);
+        Plugins.Plugin.Plugins.Add(new RpcServerPlugin());
+        resp = (JArray)_rpcServer.ListPlugins([]);
+        Assert.AreEqual(resp.Count, 2);
+        foreach (JObject p in resp)
+            Assert.AreEqual(p["name"], "RpcServer");
+    }
+
+    [TestMethod]
+    public void TestValidateAddress()
+    {
+        string validAddr = "NM7Aky765FG8NhhwtxjXRx7jEL1cnw7PBP";
+        JObject resp = (JObject)_rpcServer.ValidateAddress([validAddr]);
+        Assert.AreEqual(resp["address"], validAddr);
+        Assert.AreEqual(resp["isvalid"], true);
+        string invalidAddr = "ANeo2toNeo3MigrationAddressxwPB2Hz";
+        resp = (JObject)_rpcServer.ValidateAddress([invalidAddr]);
+        Assert.AreEqual(resp["address"], invalidAddr);
+        Assert.AreEqual(resp["isvalid"], false);
+    }
+}

tests/Neo.Plugins.RpcServer.Tests/UT_RpcServer.cs

@@ -19,6 +19,7 @@
 using Neo.Wallets;
 using Neo.Wallets.NEP6;
 using System;
+using System.Net;
 using System.Text;
 
 namespace Neo.Plugins.RpcServer.Tests
@@ -39,7 +40,7 @@ public void TestSetup()
             _memoryStore = new MemoryStore();
             _memoryStoreProvider = new TestMemoryStoreProvider(_memoryStore);
             _neoSystem = new NeoSystem(TestProtocolSettings.SoleNode, _memoryStoreProvider);
-            _rpcServer = new RpcServer(_neoSystem, RpcServerSettings.Default);
+            _rpcServer = new RpcServer(_neoSystem, RpcServerSettings.Default with { SessionEnabled = true });
             _walletAccount = _wallet.Import("KxuRSsHgJMb3AMSN6B9P3JHNGMFtxmuimqgR9MmXPcv3CLLfusTd");
             var key = new KeyBuilder(NativeContract.GAS.Id, 20).Add(_walletAccount.ScriptHash);
             var snapshot = _neoSystem.GetSnapshotCache();