test: Fix StageTransaction test

planetarium · Jul 2, 2024 · e5f31ee · e5f31ee
1 parent 83ee4df
commit e5f31ee
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/NineChronicles.Headless.Tests/GraphTypes/StandaloneMutationTest.cs b/NineChronicles.Headless.Tests/GraphTypes/StandaloneMutationTest.cs
@@ -903,14 +903,31 @@ public async Task StageTransaction()
             StandaloneContextFx.BlockChain = service.Swarm?.BlockChain;
 
             var pk = new PrivateKey();
-            ActionBase action = new ApprovePledge
+            ActionBase patronPrepareAction = new PrepareRewardAssets
             {
-                PatronAddress = new PrivateKey().Address
+                RewardPoolAddress = MeadConfig.PatronAddress,
+                Assets = new List<FungibleAssetValue>
+                {
+                    1 * Currencies.Mead,
+                },
             };
-            var tx = Transaction.Create(0, pk, BlockChain.Genesis.Hash, new[] { action.PlainValue });
+            var tx = Transaction.Create(0, pk, BlockChain.Genesis.Hash, new[] { patronPrepareAction.PlainValue });
             var payload = ByteUtil.Hex(tx.Serialize());
             var stageTxMutation = $"mutation {{ stageTransaction(payload: \"{payload}\") }}";
             var stageTxResult = await ExecuteQueryAsync(stageTxMutation);
+            Assert.Null(stageTxResult.Errors);
+
+            var block = service.BlockChain.ProposeBlock(ProposerPrivateKey, null);
+            service.BlockChain.Append(block, GenerateBlockCommit(1, block.Hash, new List<PrivateKey>() { ProposerPrivateKey }));
+
+            ActionBase action = new ApprovePledge
+            {
+                PatronAddress = new PrivateKey().Address
+            };
+            tx = Transaction.Create(0, pk, BlockChain.Genesis.Hash, new[] { action.PlainValue });
+            payload = ByteUtil.Hex(tx.Serialize());
+            stageTxMutation = $"mutation {{ stageTransaction(payload: \"{payload}\") }}";
+            stageTxResult = await ExecuteQueryAsync(stageTxMutation);
             var error = Assert.Single(stageTxResult.Errors!);
             Assert.Contains("gas", error.Message);
         }