test: 增加单元测试

Author: lokic

dataloader-plus-core/src/test/java/me/lokic/dataloaderplus/core/DataLoaderTemplateTest.java

@@ -0,0 +1,53 @@
+package me.lokic.dataloaderplus.core;
+
+import me.lokic.dataloaderplus.core.kits.CompletableFutures;
+import me.lokic.dataloaderplus.core.service.UserNameBatchLoader;
+import me.lokic.dataloaderplus.core.service.UserService;
+import org.dataloader.BatchLoaderEnvironment;
+import org.dataloader.DataLoader;
+import org.junit.Test;
+import org.mockito.Mockito;
+
+import java.util.ArrayList;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.concurrent.CompletableFuture;
+
+public class DataLoaderTemplateTest {
+
+
+    /**
+     * 多次执行dataloader，方法{@link UserNameBatchLoader#load(Set, BatchLoaderEnvironment)} 调用一次
+     */
+    @Test
+    public void testBatchLoaderCallOneTimes() {
+
+        UserNameBatchLoader userNameBatchLoader = Mockito.spy(new UserNameBatchLoader());
+
+        DataLoader<?, ?> dataLoader = DataLoader.newMappedDataLoader(userNameBatchLoader, TemplateConfig.DEFAULT.getOptions());
+        DataLoaderFactory factory = Mockito.mock(DataLoaderFactory.class);
+        Mockito.doReturn(dataLoader).when(factory).create(Mockito.any(), Mockito.any());
+
+        DataLoaderTemplate template = new DataLoaderTemplate(TemplateConfig.builder().factory(factory).build());
+
+        UserService userService = ExDataLoaderRegistry.getService(UserService.class);
+
+
+        template.using(registry -> {
+            List<CompletableFuture<String>> li = new ArrayList<>();
+            li.add(userService.getNameById("1"));
+            li.add(userService.getNameById("2"));
+            return CompletableFutures.sequence(li);
+        });
+
+        Set<String> set = new HashSet<>();
+        set.add("1");
+        set.add("2");
+
+        Mockito.verify(userNameBatchLoader, Mockito.times(1))
+                .load(Mockito.eq(set), Mockito.any());
+
+    }
+
+}

dataloader-plus-core/src/test/java/me/lokic/dataloaderplus/core/service/UserNameBatchLoader.java

@@ -0,0 +1,23 @@
+package me.lokic.dataloaderplus.core.service;
+
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import me.lokic.dataloaderplus.core.MultiKeyMappedBatchLoader;
+import org.dataloader.BatchLoaderEnvironment;
+
+import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.CompletionStage;
+import java.util.function.Function;
+import java.util.stream.Collectors;
+
+@Slf4j
+public class UserNameBatchLoader implements MultiKeyMappedBatchLoader<String, String> {
+
+    @SneakyThrows
+    @Override
+    public CompletionStage<Map<String, String>> load(Set<String> set, BatchLoaderEnvironment batchLoaderEnvironment) {
+        return CompletableFuture.supplyAsync(() -> set.stream().collect(Collectors.toMap(Function.identity(), x -> "name:" + x)));
+    }
+}

dataloader-plus-core/src/test/java/me/lokic/dataloaderplus/core/service/UserService.java

@@ -0,0 +1,13 @@
+package me.lokic.dataloaderplus.core.service;
+
+import me.lokic.dataloaderplus.core.annotation.DataLoaderMethod;
+import me.lokic.dataloaderplus.core.annotation.DataLoaderService;
+
+import java.util.concurrent.CompletableFuture;
+
+@DataLoaderService
+public interface UserService {
+
+    @DataLoaderMethod(batchLoader = UserNameBatchLoader.class)
+    CompletableFuture<String> getNameById(String id);
+}