Merge pull request #3396 from AElfProject/hotfix/increase-backup-subs…

…idy-totalshare Increase backup subsidy total share
AElfProject · Apr 18, 2023 · a0804d7 · a0804d7
2 parents b4df4f6 + 6be5cd9
commit a0804d7
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 0 deletions.
diff --git a/contract/AElf.Contracts.Profit/ProfitContract.cs b/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)
     {

diff --git a/contract/AElf.Contracts.Profit/ProfitContractState.cs b/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; }
 }
diff --git a/protobuf/profit_contract.proto b/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) {
     }

diff --git a/test/AElf.Contracts.Profit.Tests/ProfitTests.cs b/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)
     {