[incubator-linkis] branch dev-1.3.1 updated: add unit test for DataSourceInfoServiceImpl (#3198)

[ca...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

casion pushed a commit to branch dev-1.3.1
in repository https://gitbox.apache.org/repos/asf/incubator-linkis.git


The following commit(s) were added to refs/heads/dev-1.3.1 by this push:
     new 4d604d8a1 add unit test for DataSourceInfoServiceImpl (#3198)
4d604d8a1 is described below

commit 4d604d8a1cfc02ed78eec9bb7c0352b637649fbc
Author: chenmutime <ch...@outlook.com>
AuthorDate: Thu Sep 1 21:53:14 2022 +0800

    add unit test for DataSourceInfoServiceImpl (#3198)
    
    Co-authored-by: v_minminghe <v_...@webank.com>
---
 .../core/service/DataSourceInfoServiceTest.java    | 55 ++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/linkis-public-enhancements/linkis-datasource/linkis-datasource-manager/server/src/test/java/org/apache/linkis/datasourcemanager/core/service/DataSourceInfoServiceTest.java b/linkis-public-enhancements/linkis-datasource/linkis-datasource-manager/server/src/test/java/org/apache/linkis/datasourcemanager/core/service/DataSourceInfoServiceTest.java
index 9e749ea4f..8ac8b5e43 100644
--- a/linkis-public-enhancements/linkis-datasource/linkis-datasource-manager/server/src/test/java/org/apache/linkis/datasourcemanager/core/service/DataSourceInfoServiceTest.java
+++ b/linkis-public-enhancements/linkis-datasource/linkis-datasource-manager/server/src/test/java/org/apache/linkis/datasourcemanager/core/service/DataSourceInfoServiceTest.java
@@ -38,6 +38,7 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 @ExtendWith(MockitoExtension.class)
@@ -71,6 +72,17 @@ public class DataSourceInfoServiceTest {
     return dataSource;
   }
 
+  private DataSourceEnv buildDataSourceEnv() {
+    DataSourceEnv dataSourceEnv = new DataSourceEnv();
+    dataSourceEnv.setId(1L);
+    dataSourceEnv.setEnvName("test_env_name");
+    dataSourceEnv.setCreateUser("test");
+    dataSourceEnv.setParameter("{\"username\":\"test_dev\",\"password\":\"12e12e12e1\"}");
+    dataSourceEnv.setModifyUser("test");
+    dataSourceEnv.setDataSourceTypeId(1L);
+    return dataSourceEnv;
+  }
+
   @Test
   void testSaveDataSourceInfo() throws ErrorException {
     DataSource dataSource = buildDataSource();
@@ -322,4 +334,47 @@ public class DataSourceInfoServiceTest {
             keyDefinitionList, datasourceId, connectParams, username, comment);
     assertTrue(expectedVersion == res);
   }
+
+  @Test
+  void testExistDataSourceEnv() {
+    String dataSourceEnvName = "test_env_name";
+    DataSourceEnv dataSourceEnv = new DataSourceEnv();
+    dataSourceEnv.setEnvName(dataSourceEnvName);
+    dataSourceEnv.setId(1L);
+    Mockito.when(dataSourceEnvDao.selectOneByName(dataSourceEnvName)).thenReturn(dataSourceEnv);
+    Boolean result = dataSourceInfoService.existDataSourceEnv(dataSourceEnvName);
+    assertTrue(result);
+
+    Mockito.when(dataSourceEnvDao.selectOneByName(dataSourceEnvName)).thenReturn(null);
+    result = dataSourceInfoService.existDataSourceEnv(dataSourceEnvName);
+    assertFalse(result);
+
+    assertFalse(dataSourceInfoService.existDataSourceEnv(""));
+  }
+
+  @Test
+  void testSaveBatchDataSourceEnv() throws ErrorException {
+    List<DataSourceEnv> list = new ArrayList<>();
+    for (int i = 0; i < 5; i++) {
+      DataSourceEnv dataSourceEnv = buildDataSourceEnv();
+      list.add(dataSourceEnv);
+      Mockito.doNothing().when(dataSourceEnvDao).insertOne(dataSourceEnv);
+    }
+
+    dataSourceInfoService.saveBatchDataSourceEnv(list);
+  }
+
+  @Test
+  void testUpdateBatchDataSourceEnv() throws ErrorException {
+    List<DataSourceEnv> list = new ArrayList<>();
+    for (int i = 0; i < 5; i++) {
+      DataSourceEnv dataSourceEnv = buildDataSourceEnv();
+      list.add(dataSourceEnv);
+      Mockito.doNothing().when(dataSourceEnvDao).updateOne(dataSourceEnv);
+      Mockito.when(dataSourceEnvDao.selectOneDetail(dataSourceEnv.getId()))
+          .thenReturn(dataSourceEnv);
+    }
+
+    dataSourceInfoService.updateBatchDataSourceEnv(list);
+  }
 }


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@linkis.apache.org
For additional commands, e-mail: commits-help@linkis.apache.org