Add getcommittee and unregister candidate ut (neo-project#1831)

* add unregister candidate and getcommittee ut

* fix format

Co-authored-by: Shargon <[email protected]>

Author: cloud8little

tests/neo.UnitTests/SmartContract/Native/Tokens/UT_NeoToken.cs

@@ -223,6 +223,107 @@ public void Check_RegisterValidator()
             Assert.AreEqual(2, members.Length);
         }
 
+        [TestMethod]
+        public void Check_UnregisterCandidate()
+        {
+            var snapshot = Blockchain.Singleton.GetSnapshot();
+
+            var keyCount = snapshot.Storages.GetChangeSet().Count();
+            var point = Blockchain.StandbyValidators[0].EncodePoint(true);
+
+            //without register
+            var ret = Check_UnregisterCandidate(snapshot, point);
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeTrue();
+            snapshot.Storages.GetChangeSet().Count().Should().Be(keyCount);
+
+            //register and then unregister
+            ret = Check_RegisterValidator(snapshot, point);
+            StorageItem item = snapshot.Storages.GetAndChange(CreateStorageKey(33, point));
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeTrue();
+
+            var members = NativeContract.NEO.GetCandidates(snapshot);
+            Assert.AreEqual(1, members.Length);
+            snapshot.Storages.GetChangeSet().Count().Should().Be(keyCount + 1);
+            StorageKey key = CreateStorageKey(33, point);
+            snapshot.Storages.TryGet(key).Should().NotBeNull();
+
+            ret = Check_UnregisterCandidate(snapshot, point);
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeTrue();
+            snapshot.Storages.GetChangeSet().Count().Should().Be(keyCount);
+
+            members = NativeContract.NEO.GetCandidates(snapshot);
+            Assert.AreEqual(0, members.Length);
+            snapshot.Storages.TryGet(key).Should().BeNull();
+
+            //register with votes, then unregister
+            ret = Check_RegisterValidator(snapshot, point);
+            var G_Account = Contract.CreateSignatureContract(ECCurve.Secp256r1.G).ScriptHash.ToArray();
+            snapshot.Storages.Add(CreateStorageKey(20, G_Account), new StorageItem(new NeoAccountState()));
+            var accountState = snapshot.Storages.TryGet(CreateStorageKey(20, G_Account)).GetInteroperable<NeoAccountState>();
+            accountState.Balance = 100;
+            Check_Vote(snapshot, G_Account, Blockchain.StandbyValidators[0].ToArray(), true);
+            ret = Check_UnregisterCandidate(snapshot, point);
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeTrue();
+            snapshot.Storages.TryGet(key).Should().NotBeNull();
+            StorageItem pointItem = snapshot.Storages.TryGet(key);
+            CandidateState pointState = pointItem.GetInteroperable<CandidateState>();
+            pointState.Registered.Should().BeFalse();
+            pointState.Votes.Should().Be(100);
+
+            //vote fail
+            ret = Check_Vote(snapshot, G_Account, Blockchain.StandbyValidators[0].ToArray(), true);
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeFalse();
+            accountState.VoteTo.Should().Be(Blockchain.StandbyValidators[0]);
+        }
+
+        [TestMethod]
+        public void Check_GetCommittee()
+        {
+            var snapshot = Blockchain.Singleton.GetSnapshot();
+            var keyCount = snapshot.Storages.GetChangeSet().Count();
+            var point = Blockchain.StandbyValidators[0].EncodePoint(true);
+
+            //register with votes with 20000000
+            var G_Account = Contract.CreateSignatureContract(ECCurve.Secp256r1.G).ScriptHash.ToArray();
+            snapshot.Storages.Add(CreateStorageKey(20, G_Account), new StorageItem(new NeoAccountState()));
+            var accountState = snapshot.Storages.TryGet(CreateStorageKey(20, G_Account)).GetInteroperable<NeoAccountState>();
+            accountState.Balance = 20000000;
+            var ret = Check_RegisterValidator(snapshot, ECCurve.Secp256r1.G.ToArray());
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeTrue();
+            ret = Check_Vote(snapshot, G_Account, ECCurve.Secp256r1.G.ToArray(), true);
+            ret.State.Should().BeTrue();
+            ret.Result.Should().BeTrue();
+
+            var committeemembers = NativeContract.NEO.GetCommittee(snapshot);
+            var defaultCommittee = Blockchain.StandbyCommittee.OrderBy(p => p).ToArray();
+            committeemembers.GetType().Should().Be(typeof(ECPoint[]));
+            for (int i = 0; i < ProtocolSettings.Default.CommitteeMembersCount; i++)
+            {
+                committeemembers[i].Should().Be(defaultCommittee[i]);
+            }
+
+            //register more candidates,committee member change
+            for (int i = 0; i < ProtocolSettings.Default.CommitteeMembersCount - 1; i++)
+            {
+                Check_RegisterValidator(snapshot, Blockchain.StandbyCommittee[i].ToArray());
+                var currentCandidates = NativeContract.NEO.GetCandidates(snapshot);
+            }
+            committeemembers = NativeContract.NEO.GetCommittee(snapshot);
+            committeemembers.Length.Should().Be(ProtocolSettings.Default.CommitteeMembersCount);
+            committeemembers.Contains(ECCurve.Secp256r1.G).Should().BeTrue();
+            for (int i = 0; i < ProtocolSettings.Default.CommitteeMembersCount - 1; i++)
+            {
+                committeemembers.Contains(Blockchain.StandbyCommittee[i]).Should().BeTrue();
+            }
+            committeemembers.Contains(Blockchain.StandbyCommittee[ProtocolSettings.Default.CommitteeMembersCount - 1]).Should().BeFalse();
+        }
+
         [TestMethod]
         public void Check_Transfer()
         {
@@ -653,5 +754,30 @@ internal static StorageKey CreateStorageKey(byte prefix, byte[] key = null)
             key?.CopyTo(storageKey.Key.AsSpan(1));
             return storageKey;
         }
+
+        internal static (bool State, bool Result) Check_UnregisterCandidate(StoreView snapshot, byte[] pubkey)
+        {
+            var engine = ApplicationEngine.Create(TriggerType.Application,
+                new Nep5NativeContractExtensions.ManualWitness(Contract.CreateSignatureRedeemScript(ECPoint.DecodePoint(pubkey, ECCurve.Secp256r1)).ToScriptHash()), snapshot);
+
+            engine.LoadScript(NativeContract.NEO.Script);
+
+            var script = new ScriptBuilder();
+            script.EmitPush(pubkey);
+            script.EmitPush(1);
+            script.Emit(OpCode.PACK);
+            script.EmitPush("unregisterCandidate");
+            engine.LoadScript(script.ToArray());
+
+            if (engine.Execute() == VMState.FAULT)
+            {
+                return (false, false);
+            }
+
+            var result = engine.ResultStack.Pop();
+            result.Should().BeOfType(typeof(VM.Types.Boolean));
+
+            return (true, result.GetBoolean());
+        }
     }
 }