feat(action-factory): apply "claim_stake_reward6" to ClaimStakeReward…

…Factory
boscohyun · Aug 10, 2023 · 9041ecc · 9041ecc
1 parent 5be071d
commit 9041ecc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/.Lib9c.Tests/Action/Factory/ClaimStakeRewardFactoryTest.cs b/.Lib9c.Tests/Action/Factory/ClaimStakeRewardFactoryTest.cs
@@ -47,7 +47,9 @@ public static IEnumerable<object[]> GetAllClaimStakeRewardV1()
         [InlineData(ClaimStakeReward3.ObsoleteBlockIndex + 1, typeof(ClaimStakeReward4))]
         [InlineData(ClaimStakeReward4.ObsoleteBlockIndex, typeof(ClaimStakeReward4))]
         [InlineData(ClaimStakeReward4.ObsoleteBlockIndex + 1, typeof(ClaimStakeReward5))]
-        [InlineData(long.MaxValue, typeof(ClaimStakeReward5))]
+        [InlineData(ClaimStakeReward5.ObsoleteBlockIndex, typeof(ClaimStakeReward5))]
+        [InlineData(ClaimStakeReward5.ObsoleteBlockIndex + 1, typeof(ClaimStakeReward))]
+        [InlineData(long.MaxValue, typeof(ClaimStakeReward))]
         public void Create_ByBlockIndex_Success(
             long blockIndex,
             Type type)

diff --git a/Lib9c/Action/Factory/ClaimStakeRewardFactory.cs b/Lib9c/Action/Factory/ClaimStakeRewardFactory.cs
@@ -11,6 +11,11 @@ public static IClaimStakeReward CreateByBlockIndex(
             long blockIndex,
             Address avatarAddress)
         {
+            if (blockIndex > ClaimStakeReward5.ObsoleteBlockIndex)
+            {
+                return new ClaimStakeReward(avatarAddress);
+            }
+
             if (blockIndex > ClaimStakeReward4.ObsoleteBlockIndex)
             {
                 return new ClaimStakeReward5(avatarAddress);
@@ -42,6 +47,7 @@ public static IClaimStakeReward CreateByVersion(
             3 => new ClaimStakeReward3(avatarAddress),
             4 => new ClaimStakeReward4(avatarAddress),
             5 => new ClaimStakeReward5(avatarAddress),
+            6 => new ClaimStakeReward(avatarAddress),
             _ => throw new ArgumentOutOfRangeException(
                 $"Invalid version: {version}"),
         };