Merge pull request #545 from sguoo/test/collection-test

Add Collection Test

Lib9c.Models/States/CollectionState.cs

@@ -18,6 +18,8 @@ public class CollectionState : IBencodable
     [BsonIgnore, GraphQLIgnore, JsonIgnore]
     public IValue Bencoded => List.Empty.Add(new List(Ids));
 
+    public CollectionState() { }
+
     public CollectionState(IValue bencoded)
     {
         if (bencoded is not List l)

Mimir.MongoDB/Repositories/CollectionRepository.cs

@@ -6,7 +6,12 @@
 
 namespace Mimir.MongoDB.Repositories;
 
-public class CollectionRepository(IMongoDbService dbService)
+public interface ICollectionRepository
+{
+    Task<CollectionDocument> GetByAddressAsync(Address address);
+}
+
+public class CollectionRepository(IMongoDbService dbService) : ICollectionRepository
 {
     public async Task<CollectionDocument> GetByAddressAsync(Address address)
     {

Mimir.Tests/QueryTests/CollectionTest.GraphQL_Query_Collection_Returns_CorrectValue.verified.txt

@@ -0,0 +1,11 @@
+{
+  "data": {
+    "collection": {
+      "ids": [
+        1,
+        2,
+        3
+      ]
+    }
+  }
+}

Mimir.Tests/QueryTests/CollectionTest.cs

@@ -0,0 +1,38 @@
+using Mimir.MongoDB.Repositories;
+using Moq;
+using Lib9c.Models.States;
+using Libplanet.Crypto;
+using Mimir.MongoDB.Bson;
+
+namespace Mimir.Tests.QueryTests;
+
+public class CollectionTest
+{
+    [Fact]
+    public async Task GraphQL_Query_Collection_Returns_CorrectValue()
+    {
+        var address = new Address("0x0000000001000000000200000000030000000004");
+        var state = new CollectionState
+        {
+            Ids = new SortedSet<int>(new[] { 1, 2, 3 })
+        };
+        var mockRepo = new Mock<ICollectionRepository>();
+        mockRepo
+            .Setup(repo => repo.GetByAddressAsync(It.IsAny<Address>()))
+            .ReturnsAsync(new CollectionDocument(1, address, state));
+        var serviceProvider = TestServices.Builder
+            .With(mockRepo.Object)
+            .Build();
+            var query = $$"""
+                           query {
+                             collection(address: "{{address}}") {
+                               ids
+                             }
+                           }
+                           """;
+
+           var result = await TestServices.ExecuteRequestAsync(serviceProvider, b => b.SetDocument(query));
+
+           await Verify(result);
+    }
+}

Mimir/GraphQL/Queries/Query.cs

@@ -80,7 +80,7 @@ public async Task<string> GetBalanceAsync(
     /// </summary>
     /// <param name="address">The address of the avatar.</param>
     /// <returns>The collection state for the specified avatar address.</returns>
-    public async Task<CollectionState> GetCollectionAsync(Address address, [Service] CollectionRepository repo) =>
+    public async Task<CollectionState> GetCollectionAsync(Address address, [Service] ICollectionRepository repo) =>
         (await repo.GetByAddressAsync(address)).Object;
 
     /// <summary>

Mimir/Program.cs

@@ -43,7 +43,7 @@
 builder.Services.AddSingleton<ArenaParticipantRepository>();
 builder.Services.AddSingleton<IAvatarRepository, AvatarRepository>();
 builder.Services.AddSingleton<BalanceRepository>();
-builder.Services.AddSingleton<CollectionRepository>();
+builder.Services.AddSingleton<ICollectionRepository,CollectionRepository>();
 builder.Services.AddSingleton<IDailyRewardRepository, DailyRewardRepository>();
 builder.Services.AddSingleton<InventoryRepository>();
 builder.Services.AddSingleton<ItemSlotRepository>();