Refactor with repository pattern

NineChronicles.Headless.Executable.Tests/Commands/TxCommandTest.cs

@@ -73,18 +73,9 @@ public void Sign_Stake(bool gas)
         }
 
         [Theory]
-        [InlineData(null, null, false)]
-        [InlineData(0, null, true)]
-        [InlineData(ClaimStakeReward2.ObsoletedIndex - 1, null, false)]
-        [InlineData(ClaimStakeReward2.ObsoletedIndex, null, true)]
-        [InlineData(ClaimStakeReward2.ObsoletedIndex + 1, null, false)]
-        [InlineData(long.MaxValue, null, true)]
-        [InlineData(null, 1, false)]
-        [InlineData(null, 2, true)]
-        [InlineData(null, 3, false)]
-        [InlineData(null, 4, true)]
-        [InlineData(null, 5, false)]
-        public void Sign_ClaimStakeReward(long? blockIndex, int? actionVersion, bool gas)
+        [InlineData(true)]
+        [InlineData(false)]
+        public void Sign_ClaimStakeReward(bool gas)
         {
             var filePath = Path.Combine(Path.GetTempPath(), Path.GetTempFileName());
             var actionCommand = new ActionCommand(_console);

NineChronicles.Headless.Executable/Program.cs

@@ -442,6 +442,7 @@ IActionLoader MakeSingleActionLoader()
                 hostBuilder.ConfigureServices(services =>
                 {
                     services.AddSingleton(_ => standaloneContext);
+                    services.AddSingleton<IKeyStore>(standaloneContext.KeyStore);
                     services.AddOpenTelemetry()
                         .ConfigureResource(resource => resource.AddService(
                                 serviceName: Assembly.GetEntryAssembly()?.GetName().Name ?? "NineChronicles.Headless",

NineChronicles.Headless.Tests/Action/ActionContext.cs

@@ -0,0 +1,44 @@
+using System.Collections.Generic;
+using Libplanet.Action;
+using Libplanet.Action.State;
+using Libplanet.Crypto;
+using Libplanet.Extensions.ActionEvaluatorCommonComponents;
+using Libplanet.Types.Evidence;
+using Libplanet.Types.Tx;
+
+namespace NineChronicles.Headless.Tests.Action;
+
+public class ActionContext : IActionContext
+{
+    private long UsedGas { get; set; }
+
+    public Address Signer { get; init; }
+    public TxId? TxId { get; init; }
+    public Address Miner { get; init; }
+    public long BlockIndex { get; init; }
+    public int BlockProtocolVersion { get; init; }
+    public IWorld PreviousState { get; init; }
+    public int RandomSeed { get; init; }
+    public bool IsPolicyAction { get; init; }
+    public IReadOnlyList<ITransaction> Txs { get; init; }
+    public IReadOnlyList<EvidenceBase> Evidence { get; init; }
+    public void UseGas(long gas)
+    {
+        UsedGas += gas;
+    }
+
+    public IRandom GetRandom()
+    {
+        return new Random(RandomSeed);
+    }
+
+    public long GasUsed()
+    {
+        return UsedGas;
+    }
+
+    public long GasLimit()
+    {
+        return 0L;
+    }
+}

NineChronicles.Headless.Tests/GraphQLTestUtils.cs

@@ -24,6 +24,10 @@
 using Nekoyume.Action.Loader;
 using Nekoyume.Model.State;
 using Nekoyume.Module;
+using NineChronicles.Headless.Repositories.BlockChain;
+using NineChronicles.Headless.Repositories.StateTrie;
+using NineChronicles.Headless.Repositories.Transaction;
+using NineChronicles.Headless.Repositories.WorldState;
 using NineChronicles.Headless.Utils;
 
 namespace NineChronicles.Headless.Tests
@@ -61,6 +65,10 @@ public static Task<ExecutionResult> ExecuteQueryAsync<TObjectGraphType>(
 
             services.AddLibplanetExplorer();
             services.AddSingleton<StateMemoryCache>();
+            services.AddTransient<IWorldStateRepository, WorldStateRepository>();
+            services.AddTransient<IBlockChainRepository, BlockChainRepository>();
+            services.AddSingleton<IStateTrieRepository, StateTrieRepository>();
+            services.AddSingleton<ITransactionRepository, TransactionRepository>();
 
             var serviceProvider = services.BuildServiceProvider();
             return ExecuteQueryAsync<TObjectGraphType>(
@@ -78,6 +86,15 @@ public static Task<ExecutionResult> ExecuteQueryAsync<TObjectGraphType>(
             where TObjectGraphType : IObjectGraphType
         {
             var graphType = (IObjectGraphType)serviceProvider.GetService(typeof(TObjectGraphType))!;
+            return ExecuteQueryAsync(graphType, query, userContext, source);
+        }
+
+        public static Task<ExecutionResult> ExecuteQueryAsync(
+            IObjectGraphType graphType,
+            string query,
+            IDictionary<string, object>? userContext = null,
+            object? source = null)
+        {
             var documentExecutor = new DocumentExecuter();
             return documentExecutor.ExecuteAsync(new ExecutionOptions
             {

NineChronicles.Headless.Tests/GraphTypes/GraphQLTestBase.cs

@@ -34,7 +34,16 @@
 using System.Threading;
 using System.Threading.Tasks;
 using Bencodex.Types;
+using Libplanet.Action.State;
+using Libplanet.Mocks;
 using Libplanet.Types.Tx;
+using Moq;
+using NineChronicles.Headless.Executable.Tests.KeyStore;
+using NineChronicles.Headless.Repositories;
+using NineChronicles.Headless.Repositories.BlockChain;
+using NineChronicles.Headless.Repositories.StateTrie;
+using NineChronicles.Headless.Repositories.Transaction;
+using NineChronicles.Headless.Repositories.WorldState;
 using Xunit.Abstractions;
 
 namespace NineChronicles.Headless.Tests.GraphTypes
@@ -86,12 +95,10 @@ public GraphQLTestBase(ITestOutputHelper output)
                 privateKey: AdminPrivateKey);
 
             var ncService = ServiceBuilder.CreateNineChroniclesNodeService(genesisBlock, ProposerPrivateKey);
-            var tempKeyStorePath = Path.Join(Path.GetTempPath(), Path.GetRandomFileName());
-            var keyStore = new Web3KeyStore(tempKeyStorePath);
 
             StandaloneContextFx = new StandaloneContext
             {
-                KeyStore = keyStore,
+                KeyStore = KeyStore,
                 DifferentAppProtocolVersionEncounterInterval = TimeSpan.FromSeconds(1),
                 NotificationInterval = TimeSpan.FromSeconds(1),
                 NodeExceptionInterval = TimeSpan.FromSeconds(1),
@@ -117,6 +124,12 @@ public GraphQLTestBase(ITestOutputHelper output)
             );
             services.AddSingleton(publisher);
             services.AddSingleton(StandaloneContextFx);
+            services.AddTransient(provider => provider.GetService<StandaloneContext>().BlockChain);
+            services.AddSingleton<IWorldStateRepository>(WorldStateRepository.Object);
+            services.AddSingleton<IBlockChainRepository>(BlockChainRepository.Object);
+            services.AddSingleton<IStateTrieRepository>(StateTrieRepository.Object);
+            services.AddSingleton<ITransactionRepository>(TransactionRepository.Object);
+            services.AddSingleton<IKeyStore>(KeyStore);
             services.AddSingleton<IConfiguration>(configuration);
             services.AddGraphTypes();
             services.AddLibplanetExplorer();
@@ -129,6 +142,12 @@ public GraphQLTestBase(ITestOutputHelper output)
             DocumentExecutor = new DocumentExecuter();
         }
 
+        protected Mock<IWorldStateRepository> WorldStateRepository { get; } = new();
+        protected Mock<IStateTrieRepository> StateTrieRepository { get; } = new();
+        protected Mock<IBlockChainRepository> BlockChainRepository { get; } = new();
+        protected Mock<ITransactionRepository> TransactionRepository { get; } = new();
+        protected IKeyStore KeyStore { get; } = new InMemoryKeyStore();
+
         protected PrivateKey AdminPrivateKey { get; } = new PrivateKey();
 
         protected Address AdminAddress => AdminPrivateKey.Address;
@@ -150,9 +169,6 @@ protected List<PrivateKey> GenesisValidators
         protected BlockChain BlockChain =>
             StandaloneContextFx.BlockChain!;
 
-        protected IKeyStore KeyStore =>
-            StandaloneContextFx.KeyStore!;
-
         protected IDocumentExecuter DocumentExecutor { get; }
 
         protected SubscriptionDocumentExecuter SubscriptionDocumentExecuter { get; } = new SubscriptionDocumentExecuter();
@@ -187,6 +203,25 @@ protected async Task<Task> StartAsync(
             return task;
         }
 
+        protected void SetupStatesOnTip(Func<IWorld, IWorld> func)
+        {
+            var worldState = func(new World(MockUtil.MockModernWorldState));
+            var stateRootHash = worldState.Trie.Hash;
+            var tip = new Domain.Model.BlockChain.Block(
+                BlockHash.FromString("613dfa26e104465790625ae7bc03fc27a64947c02a9377565ec190405ef7154b"),
+                BlockHash.FromString("36456be15af9a5b9b13a02c7ce1e849ae9cba8781ec309010499cdb93e29237d"),
+                default(Address),
+                0,
+                Timestamp: DateTimeOffset.UtcNow,
+                StateRootHash: stateRootHash,
+                Transactions: ImmutableArray<Transaction>.Empty
+            );
+            BlockChainRepository.Setup(repository => repository.GetTip())
+                .Returns(tip);
+            WorldStateRepository.Setup(repository => repository.GetWorldState(stateRootHash))
+                .Returns(worldState);
+        }
+
         protected LibplanetNodeService CreateLibplanetNodeService(
             Block genesisBlock,
             AppProtocolVersion appProtocolVersion,