WIP

moreal · Sep 5, 2024 · a6e6abc · a6e6abc
1 parent d332132
commit a6e6abc
Show file tree

Hide file tree

Showing 23 changed files with 651 additions and 505 deletions.
diff --git a/NineChronicles.Headless.Executable/Program.cs b/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",

diff --git a/NineChronicles.Headless.Tests/Action/ActionContext.cs b/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;
+    }
+}
diff --git a/NineChronicles.Headless.Tests/GraphQLTestUtils.cs b/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
             {

diff --git a/NineChronicles.Headless.Tests/GraphTypes/GraphQLTestBase.cs b/NineChronicles.Headless.Tests/GraphTypes/GraphQLTestBase.cs
@@ -35,6 +35,13 @@
 using System.Threading.Tasks;
 using Bencodex.Types;
 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 +93,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 +122,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 +140,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 +167,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();