Add unit-test for cross-aa

Author: chopin-fan

protobuf/test_virtual_address_contract.proto

@@ -14,6 +14,7 @@ service VirtualAddressContract {
   option (aelf.base) = "acs1.proto";
 
   rpc VirtualAddressVote(VirtualAddressVoteInput) returns (google.protobuf.Empty);
+  rpc VirtualAddressVoteWithInline(VirtualAddressVoteWithCountInput) returns (google.protobuf.Empty);
   rpc VirtualAddressWithdraw(aelf.Hash) returns (google.protobuf.Empty);
   rpc VirtualAddressChangeVotingOption(VirtualAddressChangeVotingOptionInput) returns (google.protobuf.Empty);
   rpc VirtualAddressClaimProfit(VirtualAddressClaimProfitInput) returns (google.protobuf.Empty);
@@ -32,6 +33,11 @@ message VirtualAddressVoteInput {
   aelf.Hash token = 4;
 }
 
+message VirtualAddressVoteWithCountInput {
+  VirtualAddressVoteInput vote_input = 1; // Existing input type
+  int32 count = 2; // Additional count input
+}
+
 message VirtualAddressChangeVotingOptionInput {
   bool is_reset = 1;
   aelf.Hash vote_id = 2;

test/AElf.Contracts.Election.Tests/BVT/ElectionTests.cs

@@ -1984,6 +1984,133 @@ await VirtualAddressContractStub.VirtualAddressVote.SendAsync(new VirtualAddress
         return result.ActiveVotingRecords.First().VoteId;
     }
 
+    // [Fact]
+    // public async Task<Hash> VirtualAddress_Vote_WithInlineTxIdAndCountTest()
+    // {
+    //     var amount = 100;
+    //     const int lockTime = 100 * 60 * 60 * 24;
+    //     var candidatesKeyPairs = await ElectionContract_AnnounceElection_Test();
+    //     var candidateKeyPair = candidatesKeyPairs[0];
+    //     
+    //     var address = await VirtualAddressContractStub.GetVirtualAddress.CallAsync(new Empty());
+    //     var initBalance = 100000;
+    //
+    //     await TokenContractStub.Transfer.SendAsync(new TransferInput
+    //     {
+    //         Amount = initBalance,
+    //         Symbol = "ELF",
+    //         To = address,
+    //         Memo = "test"
+    //     });
+    //     
+    //     CheckBalance(address, "ELF", initBalance);
+    //     CheckBalance(address, "SHARE", 0);
+    //     CheckBalance(address, "VOTE", 0);
+    //
+    //     var virtualAddressVoteInput = new VirtualAddressVoteInput
+    //     {
+    //         PubKey = candidateKeyPair.PublicKey.ToHex(),
+    //         Amount = amount,
+    //         EndTimestamp = TimestampHelper.GetUtcNow().AddSeconds(lockTime),
+    //         Token = HashHelper.ComputeFrom("token A")
+    //     };
+    //
+    //     var resultawait = await VirtualAddressContractStub.VirtualAddressVoteWithInline.SendAsync(new VirtualAddressVoteWithCountInput
+    //     {
+    //         VoteInput = virtualAddressVoteInput,
+    //         Count = 6
+    //     });
+    //     return null;
+    // }
+    //
+    [Fact]
+    public async Task<Hash> VirtualAddress_Vote_WithInlineTxIdTest()
+    {
+        var amount = 100;
+        const int lockTime = 100 * 60 * 60 * 24;
+        var candidatesKeyPairs = await ElectionContract_AnnounceElection_Test();
+        var candidateKeyPair = candidatesKeyPairs[0];
+        
+        var address = await VirtualAddressContractStub.GetVirtualAddress.CallAsync(new Empty());
+        var initBalance = 100000;
+
+        await TokenContractStub.Transfer.SendAsync(new TransferInput
+        {
+            Amount = initBalance,
+            Symbol = "ELF",
+            To = address,
+            Memo = "test"
+        });
+        
+        CheckBalance(address, "ELF", initBalance);
+        CheckBalance(address, "SHARE", 0);
+        CheckBalance(address, "VOTE", 0);
+
+        var virtualAddressVoteInput = new VirtualAddressVoteInput
+        {
+            PubKey = candidateKeyPair.PublicKey.ToHex(),
+            Amount = amount,
+            EndTimestamp = TimestampHelper.GetUtcNow().AddSeconds(lockTime),
+            Token = HashHelper.ComputeFrom("token A")
+        };
+
+        await VirtualAddressContractStub.VirtualAddressVoteWithInline.SendAsync(new VirtualAddressVoteWithCountInput
+        {
+            VoteInput = virtualAddressVoteInput,
+            Count = 1
+        });
+        
+        var result = await ElectionContractStub.GetElectorVote.CallAsync(new StringValue
+        {
+            Value = address.ToBase58()
+        });
+        result.ActiveVotedVotesAmount.ShouldBe(amount);
+        result = await ElectionContractStub.GetElectorVoteWithRecords.CallAsync(new StringValue
+        {
+            Value = address.ToBase58()
+        });
+        result.ActiveVotedVotesAmount.ShouldBe(amount);
+        result = await ElectionContractStub.GetElectorVoteWithAllRecords.CallAsync(new StringValue
+        {
+            Value = address.ToBase58()
+        });
+        result.AllVotedVotesAmount.ShouldBe(amount);
+        
+        CheckBalance(address, "ELF", initBalance - amount);
+        CheckBalance(address, "SHARE", amount);
+        CheckBalance(address, "VOTE", amount);
+        
+        await VirtualAddressContractStub.VirtualAddressVote.SendAsync(new VirtualAddressVoteInput
+        {
+            PubKey = candidateKeyPair.PublicKey.ToHex(),
+            Amount = amount,
+            EndTimestamp = TimestampHelper.GetUtcNow().AddSeconds(lockTime),
+            Token = HashHelper.ComputeFrom("token A")
+        });
+        
+        result = await ElectionContractStub.GetElectorVote.CallAsync(new StringValue
+        {
+            Value = address.ToBase58()
+        });
+        result.ActiveVotedVotesAmount.ShouldBe(amount + amount);
+        result = await ElectionContractStub.GetElectorVoteWithRecords.CallAsync(new StringValue
+        {
+            Value = address.ToBase58()
+        });
+        result.ActiveVotedVotesAmount.ShouldBe(amount + amount);
+        result = await ElectionContractStub.GetElectorVoteWithAllRecords.CallAsync(new StringValue
+        {
+            Value = address.ToBase58()
+        });
+        result.AllVotedVotesAmount.ShouldBe(amount + amount);
+        
+        CheckBalance(address, "ELF", initBalance - amount - amount);
+        CheckBalance(address, "SHARE", amount + amount);
+        CheckBalance(address, "VOTE", amount + amount);
+        
+        return result.ActiveVotingRecords.First().VoteId;
+    }
+
     [Fact]
     public async Task VirtualAddress_Withdraw_Test()
     {

test/AElf.Contracts.TestContract.VirtualAddress/Action.cs

@@ -9,6 +9,27 @@ namespace AElf.Contracts.TestContract.VirtualAddress;
 
 public partial class Action : VirtualAddressContractContainer.VirtualAddressContractBase
 {
+    
+    public override Empty VirtualAddressVoteWithInline(VirtualAddressVoteWithCountInput input)
+    {
+        if (input.Count == null) return new Empty();
+        Initialize();
+
+        for (int i = 0; i < input.Count; i++)
+        {
+            Context.SendVirtualInline(HashHelper.ComputeFrom("test"), State.ElectionContract.Value, ".Vote", new VoteMinerInput
+            {
+                CandidatePubkey = input.VoteInput.PubKey,
+                Amount = input.VoteInput.Amount,
+                EndTimestamp = input.VoteInput.EndTimestamp,
+                Token = input.VoteInput.Token
+            }.ToByteString());
+        }
+        
+
+        return new Empty();
+    }
+    
     public override Empty VirtualAddressVote(VirtualAddressVoteInput input)
     {
         Initialize();