Merge pull request #3396 from AElfProject/hotfix/increase-backup-subsidy-totalshare

Increase backup subsidy total share

Author: jason-aelf

contract/AElf.Contracts.Profit/ProfitContract.cs

@@ -719,6 +719,16 @@ public override Empty ContributeProfits(ContributeProfitsInput input)
 
         return new Empty();
     }
+    
+    public override Empty IncreaseBackupSubsidyTotalShare(Hash schemeId)
+    {
+        Assert(!State.BackupSubsidyTotalShareIncreased.Value, "Already increased");
+        State.BackupSubsidyTotalShareIncreased.Value = true;
+        var scheme = State.SchemeInfos[schemeId];
+        scheme.TotalShares = scheme.TotalShares.Add(1);
+        State.SchemeInfos[schemeId] = scheme;
+        return new Empty();
+    }
 
     public override Empty ResetManager(ResetManagerInput input)
     {

contract/AElf.Contracts.Profit/ProfitContractState.cs

@@ -17,4 +17,6 @@ public partial class ProfitContractState : ContractState
     public MappedState<string, MethodFees> TransactionFees { get; set; }
 
     public SingletonState<AuthorityInfo> MethodFeeController { get; set; }
+    
+    public BoolState BackupSubsidyTotalShareIncreased { get; set; }
 }

protobuf/profit_contract.proto

@@ -60,6 +60,10 @@ service ProfitContract {
     rpc RemoveSubScheme (RemoveSubSchemeInput) returns (google.protobuf.Empty) {
     }
 
+    // Increase backup subsidy TotalShare
+    rpc IncreaseBackupSubsidyTotalShare (aelf.Hash) returns (google.protobuf.Empty) {
+    }
+    
     // Reset the manager of a scheme.
     rpc ResetManager (ResetManagerInput) returns (google.protobuf.Empty) {
     }

test/AElf.Contracts.Profit.Tests/ProfitTests.cs

@@ -1679,6 +1679,19 @@ await ProfitContractStub.ClaimProfits.SendAsync(new ClaimProfitsInput
             details.Details.First().LastProfitPeriod.ShouldBe(scheme.CurrentPeriod);
         }
     }
+    
+    [Fact]
+    public async Task IncreaseBackupSubsidyTotalShare_Test()
+    {
+        var schemeId = await CreateSchemeAsync();
+        var scheme = await ProfitContractStub.GetScheme.CallAsync(schemeId);
+        scheme.TotalShares.ShouldBe(0);
+        await ProfitContractStub.IncreaseBackupSubsidyTotalShare.SendAsync(schemeId);
+        scheme = await ProfitContractStub.GetScheme.CallAsync(schemeId);
+        scheme.TotalShares.ShouldBe(1);
+        var result = await ProfitContractStub.IncreaseBackupSubsidyTotalShare.SendWithExceptionAsync(schemeId);
+        result.TransactionResult.Error.ShouldContain("Already increased");
+    }
 
     private async Task ContributeProfits(Hash schemeId, long amount = 100)
     {