add UT for NamespacePersist (#11852)
* UT for NamespacePersist * NamespacePersistTest fix
This commit is contained in:
hth
GitHub
parent
9fd6ff6a04
commit
9db51e58ed
@ -0,0 +1,219 @@
|
||||
/*
|
||||
* Copyright 1999-2024 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.alibaba.nacos.core.namespace.repository;
|
||||
|
||||
import com.alibaba.nacos.api.exception.runtime.NacosRuntimeException;
|
||||
import com.alibaba.nacos.core.namespace.model.TenantInfo;
|
||||
import com.alibaba.nacos.persistence.configuration.DatasourceConfiguration;
|
||||
import com.alibaba.nacos.persistence.datasource.DataSourceService;
|
||||
import com.alibaba.nacos.persistence.datasource.DynamicDataSource;
|
||||
import com.alibaba.nacos.persistence.repository.embedded.operate.DatabaseOperate;
|
||||
import com.alibaba.nacos.sys.env.EnvUtil;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
import org.springframework.mock.env.MockEnvironment;
|
||||
import org.springframework.test.util.ReflectionTestUtils;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import static com.alibaba.nacos.core.namespace.repository.NamespaceRowMapperInjector.TENANT_INFO_ROW_MAPPER;
|
||||
import static org.mockito.ArgumentMatchers.anyList;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.ArgumentMatchers.eq;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class EmbeddedNamespacePersistServiceTest {
|
||||
|
||||
@Mock
|
||||
private DatabaseOperate databaseOperate;
|
||||
|
||||
@Mock
|
||||
private DataSourceService dataSourceService;
|
||||
|
||||
private EmbeddedNamespacePersistServiceImpl embeddedNamespacePersistService;
|
||||
|
||||
private MockEnvironment environment;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
EnvUtil.setIsStandalone(true);
|
||||
DatasourceConfiguration.setEmbeddedStorage(true);
|
||||
environment = new MockEnvironment();
|
||||
EnvUtil.setEnvironment(environment);
|
||||
DynamicDataSource instance = DynamicDataSource.getInstance();
|
||||
ReflectionTestUtils.setField(instance, "localDataSourceService", dataSourceService);
|
||||
embeddedNamespacePersistService = new EmbeddedNamespacePersistServiceImpl(databaseOperate);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void insertTenantInfoAtomicTest1() {
|
||||
|
||||
String namespaceId = "testNsId";
|
||||
String kp = "1";
|
||||
String namespaceName = "testNs";
|
||||
String namespaceDesc = "testDes";
|
||||
String createRes = "nacos";
|
||||
when(databaseOperate.update(anyList())).thenReturn(true);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
|
||||
embeddedNamespacePersistService.insertTenantInfoAtomic(kp, namespaceId, namespaceName, namespaceDesc, createRes,
|
||||
System.currentTimeMillis());
|
||||
|
||||
verify(databaseOperate).update(anyList());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void insertTenantInfoAtomicTest2() {
|
||||
|
||||
String namespaceId = "testNsId";
|
||||
String kp = "1";
|
||||
String namespaceName = "testNs";
|
||||
String namespaceDesc = "testDes";
|
||||
String createRes = "nacos";
|
||||
when(databaseOperate.update(anyList())).thenReturn(false);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
|
||||
Assert.assertThrows(NacosRuntimeException.class,
|
||||
() -> embeddedNamespacePersistService.insertTenantInfoAtomic(kp, namespaceId, namespaceName,
|
||||
namespaceDesc, createRes, System.currentTimeMillis()));
|
||||
|
||||
verify(databaseOperate).update(anyList());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void removeTenantInfoAtomicTest() {
|
||||
String namespaceId = "testNsId";
|
||||
String kp = "1";
|
||||
|
||||
when(databaseOperate.update(anyList())).thenReturn(true);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
|
||||
embeddedNamespacePersistService.removeTenantInfoAtomic(kp, namespaceId);
|
||||
|
||||
verify(databaseOperate).update(anyList());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updateTenantNameAtomicTest1() {
|
||||
|
||||
String namespaceId = "testNsId";
|
||||
String kp = "1";
|
||||
String namespaceName = "testNs";
|
||||
String namespaceDesc = "testDes";
|
||||
when(databaseOperate.update(anyList())).thenReturn(true);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
|
||||
embeddedNamespacePersistService.updateTenantNameAtomic(kp, namespaceId, namespaceName, namespaceDesc);
|
||||
|
||||
verify(databaseOperate).update(anyList());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updateTenantNameAtomicTest2() {
|
||||
|
||||
String namespaceId = "testNsId";
|
||||
String kp = "1";
|
||||
String namespaceName = "testNs";
|
||||
String namespaceDesc = "testDes";
|
||||
when(databaseOperate.update(anyList())).thenReturn(false);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
|
||||
Assert.assertThrows(NacosRuntimeException.class,
|
||||
() -> embeddedNamespacePersistService.updateTenantNameAtomic(kp, namespaceId, namespaceName,
|
||||
namespaceDesc));
|
||||
verify(databaseOperate).update(anyList());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void findTenantByKpTest1() {
|
||||
String kp = "1";
|
||||
List<TenantInfo> tenantInfoList = new ArrayList<>(1);
|
||||
tenantInfoList.add(new TenantInfo());
|
||||
when(databaseOperate.queryMany(anyString(), eq(new Object[] {kp}), eq(TENANT_INFO_ROW_MAPPER))).thenReturn(
|
||||
tenantInfoList);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
List<TenantInfo> tenantByKp = embeddedNamespacePersistService.findTenantByKp(kp);
|
||||
|
||||
Assert.assertEquals(tenantByKp.get(0), tenantInfoList.get(0));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void findTenantByKpTest2() {
|
||||
String kp = "1";
|
||||
String tenantId = "tenantId";
|
||||
TenantInfo tenantInfo = new TenantInfo();
|
||||
tenantInfo.setTenantId(tenantId);
|
||||
when(databaseOperate.queryOne(anyString(), eq(new Object[] {kp, tenantId}),
|
||||
eq(TENANT_INFO_ROW_MAPPER))).thenReturn(tenantInfo);
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
TenantInfo tenantByKp = embeddedNamespacePersistService.findTenantByKp(kp, tenantId);
|
||||
|
||||
Assert.assertEquals(tenantInfo.getTenantId(), tenantByKp.getTenantId());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void generateLikeArgumentTest() {
|
||||
|
||||
String test = embeddedNamespacePersistService.generateLikeArgument("test");
|
||||
|
||||
String testB = embeddedNamespacePersistService.generateLikeArgument("test*");
|
||||
|
||||
Assert.assertEquals("test", test);
|
||||
|
||||
Assert.assertEquals("test%", testB);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void isExistTableTest() {
|
||||
String tableName = "tableName";
|
||||
String sql = String.format("SELECT 1 FROM %s FETCH FIRST ROW ONLY", tableName);
|
||||
|
||||
when(databaseOperate.queryOne(sql, Integer.class)).thenReturn(1);
|
||||
boolean existTableA = embeddedNamespacePersistService.isExistTable(tableName);
|
||||
Assert.assertTrue(existTableA);
|
||||
|
||||
when(databaseOperate.queryOne(sql, Integer.class)).thenThrow(new RuntimeException("test"));
|
||||
boolean existTableB = embeddedNamespacePersistService.isExistTable(tableName);
|
||||
Assert.assertFalse(existTableB);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void tenantInfoCountByTenantIdTest() {
|
||||
String tenantId = "tenantId";
|
||||
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("derby");
|
||||
|
||||
Assert.assertThrows(IllegalArgumentException.class,
|
||||
() -> embeddedNamespacePersistService.tenantInfoCountByTenantId(null));
|
||||
|
||||
when(databaseOperate.queryOne(anyString(), eq(new String[] {tenantId}), eq(Integer.class))).thenReturn(null);
|
||||
int i = embeddedNamespacePersistService.tenantInfoCountByTenantId(tenantId);
|
||||
Assert.assertEquals(0, i);
|
||||
|
||||
when(databaseOperate.queryOne(anyString(), eq(new String[] {tenantId}), eq(Integer.class))).thenReturn(1);
|
||||
int j = embeddedNamespacePersistService.tenantInfoCountByTenantId(tenantId);
|
||||
Assert.assertEquals(1, j);
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,228 @@
|
||||
/*
|
||||
* Copyright 1999-2024 Alibaba Group Holding Ltd.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.alibaba.nacos.core.namespace.repository;
|
||||
|
||||
import com.alibaba.nacos.core.namespace.model.TenantInfo;
|
||||
import com.alibaba.nacos.persistence.configuration.DatasourceConfiguration;
|
||||
import com.alibaba.nacos.persistence.datasource.DataSourceService;
|
||||
import com.alibaba.nacos.persistence.datasource.DynamicDataSource;
|
||||
import com.alibaba.nacos.persistence.exception.NJdbcException;
|
||||
import com.alibaba.nacos.persistence.repository.embedded.operate.DatabaseOperate;
|
||||
import com.alibaba.nacos.sys.env.EnvUtil;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.junit.MockitoJUnitRunner;
|
||||
import org.springframework.dao.DataAccessException;
|
||||
import org.springframework.dao.EmptyResultDataAccessException;
|
||||
import org.springframework.jdbc.CannotGetJdbcConnectionException;
|
||||
import org.springframework.jdbc.core.JdbcTemplate;
|
||||
import org.springframework.mock.env.MockEnvironment;
|
||||
import org.springframework.test.util.ReflectionTestUtils;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import static com.alibaba.nacos.core.namespace.repository.NamespaceRowMapperInjector.TENANT_INFO_ROW_MAPPER;
|
||||
import static org.mockito.ArgumentMatchers.anyLong;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.ArgumentMatchers.eq;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class ExternalNamespacePersistServiceTest {
|
||||
|
||||
@Mock
|
||||
private DatabaseOperate databaseOperate;
|
||||
|
||||
@Mock
|
||||
private DataSourceService dataSourceService;
|
||||
|
||||
@Mock
|
||||
private JdbcTemplate jt;
|
||||
|
||||
private ExternalNamespacePersistServiceImpl externalNamespacePersistService;
|
||||
|
||||
private MockEnvironment environment;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
EnvUtil.setIsStandalone(false);
|
||||
DatasourceConfiguration.setEmbeddedStorage(false);
|
||||
environment = new MockEnvironment();
|
||||
EnvUtil.setEnvironment(environment);
|
||||
DynamicDataSource instance = DynamicDataSource.getInstance();
|
||||
ReflectionTestUtils.setField(instance, "basicDataSourceService", dataSourceService);
|
||||
externalNamespacePersistService = new ExternalNamespacePersistServiceImpl();
|
||||
|
||||
ReflectionTestUtils.setField(externalNamespacePersistService, "jt", jt);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void insertTenantInfoAtomicTest() {
|
||||
String kp = "1";
|
||||
String namespaceId = "namespaceId";
|
||||
String namespaceName = "namespaceName";
|
||||
String namespaceDesc = "namespaceDesc";
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("mysql");
|
||||
externalNamespacePersistService.insertTenantInfoAtomic(kp, namespaceId, namespaceName, namespaceDesc, "nacos",
|
||||
System.currentTimeMillis());
|
||||
|
||||
when(jt.update(anyString(), anyString(), anyString(), anyString(), anyString(), anyString(), anyLong(),
|
||||
anyLong())).thenThrow(new NJdbcException("test"));
|
||||
Assert.assertThrows(DataAccessException.class,
|
||||
() -> externalNamespacePersistService.insertTenantInfoAtomic(kp, namespaceId, namespaceName,
|
||||
namespaceDesc, "nacos", System.currentTimeMillis()));
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void removeTenantInfoAtomicTest() {
|
||||
String kp = "1";
|
||||
String namespaceId = "namespaceId";
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("mysql");
|
||||
|
||||
externalNamespacePersistService.removeTenantInfoAtomic(kp, namespaceId);
|
||||
|
||||
when(jt.update(anyString(), anyString(), anyString())).thenThrow(new CannotGetJdbcConnectionException("test"));
|
||||
Assert.assertThrows(CannotGetJdbcConnectionException.class,
|
||||
() -> externalNamespacePersistService.removeTenantInfoAtomic(kp, namespaceId));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updateTenantNameAtomicTest() {
|
||||
String kp = "1";
|
||||
String namespaceId = "namespaceId";
|
||||
String namespaceName = "namespaceName";
|
||||
String namespaceDesc = "namespaceDesc";
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("mysql");
|
||||
externalNamespacePersistService.updateTenantNameAtomic(kp, namespaceId, namespaceName, namespaceDesc);
|
||||
|
||||
when(jt.update(anyString(), anyString(), anyString(), anyLong(), anyString(), anyString())).thenThrow(
|
||||
new NJdbcException("test"));
|
||||
Assert.assertThrows(DataAccessException.class,
|
||||
() -> externalNamespacePersistService.updateTenantNameAtomic(kp, namespaceId, namespaceName,
|
||||
namespaceDesc));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void findTenantByKpTest() {
|
||||
String kp = "1";
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("mysql");
|
||||
List<TenantInfo> tenantInfoList = new ArrayList<>(1);
|
||||
tenantInfoList.add(new TenantInfo());
|
||||
|
||||
when(jt.query(anyString(), eq(new Object[] {kp}), eq(TENANT_INFO_ROW_MAPPER))).thenReturn(tenantInfoList);
|
||||
|
||||
List<TenantInfo> tenantByKp = externalNamespacePersistService.findTenantByKp(kp);
|
||||
Assert.assertEquals(tenantInfoList.get(0), tenantByKp.get(0));
|
||||
|
||||
when(jt.query(anyString(), eq(new Object[] {kp}), eq(TENANT_INFO_ROW_MAPPER))).thenThrow(
|
||||
new CannotGetJdbcConnectionException("test"));
|
||||
Assert.assertThrows(CannotGetJdbcConnectionException.class,
|
||||
() -> externalNamespacePersistService.findTenantByKp(kp));
|
||||
|
||||
when(jt.query(anyString(), eq(new Object[] {kp}), eq(TENANT_INFO_ROW_MAPPER))).thenThrow(
|
||||
new EmptyResultDataAccessException(1));
|
||||
List<TenantInfo> tenantByKp1 = externalNamespacePersistService.findTenantByKp(kp);
|
||||
Assert.assertEquals(Collections.emptyList(), tenantByKp1);
|
||||
|
||||
when(jt.query(anyString(), eq(new Object[] {kp}), eq(TENANT_INFO_ROW_MAPPER))).thenThrow(
|
||||
new RuntimeException("test"));
|
||||
Assert.assertThrows(RuntimeException.class, () -> externalNamespacePersistService.findTenantByKp(kp));
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void findTenantByKpTest2() {
|
||||
String kp = "1";
|
||||
String namespaceId = "namespaceId";
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("mysql");
|
||||
TenantInfo tenantInfo = new TenantInfo();
|
||||
tenantInfo.setTenantId(namespaceId);
|
||||
|
||||
when(jt.queryForObject(anyString(), eq(new Object[] {kp, namespaceId}), eq(TENANT_INFO_ROW_MAPPER))).thenReturn(
|
||||
tenantInfo);
|
||||
|
||||
TenantInfo tenantByKp = externalNamespacePersistService.findTenantByKp(kp, namespaceId);
|
||||
Assert.assertEquals(tenantInfo.getTenantId(), tenantByKp.getTenantId());
|
||||
|
||||
when(jt.queryForObject(anyString(), eq(new Object[] {kp, namespaceId}), eq(TENANT_INFO_ROW_MAPPER))).thenThrow(
|
||||
new CannotGetJdbcConnectionException("test"));
|
||||
Assert.assertThrows(CannotGetJdbcConnectionException.class,
|
||||
() -> externalNamespacePersistService.findTenantByKp(kp, namespaceId));
|
||||
|
||||
when(jt.queryForObject(anyString(), eq(new Object[] {kp, namespaceId}), eq(TENANT_INFO_ROW_MAPPER))).thenThrow(
|
||||
new EmptyResultDataAccessException(1));
|
||||
TenantInfo tenantByKp1 = externalNamespacePersistService.findTenantByKp(kp, namespaceId);
|
||||
Assert.assertNull(tenantByKp1);
|
||||
|
||||
when(jt.queryForObject(anyString(), eq(new Object[] {kp, namespaceId}), eq(TENANT_INFO_ROW_MAPPER))).thenThrow(
|
||||
new RuntimeException("test"));
|
||||
Assert.assertThrows(RuntimeException.class,
|
||||
() -> externalNamespacePersistService.findTenantByKp(kp, namespaceId));
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void generateLikeArgumentTest() {
|
||||
|
||||
String test = externalNamespacePersistService.generateLikeArgument("test");
|
||||
|
||||
String testB = externalNamespacePersistService.generateLikeArgument("test*");
|
||||
|
||||
Assert.assertEquals("test", test);
|
||||
|
||||
Assert.assertEquals("test%", testB);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void isExistTableTest() {
|
||||
String tableName = "tableName";
|
||||
String sql = String.format("SELECT 1 FROM %s LIMIT 1", tableName);
|
||||
|
||||
when(jt.queryForObject(eq(sql), eq(Integer.class))).thenReturn(1);
|
||||
boolean existTableA = externalNamespacePersistService.isExistTable(tableName);
|
||||
Assert.assertTrue(existTableA);
|
||||
|
||||
when(jt.queryForObject(eq(sql), eq(Integer.class))).thenThrow(new RuntimeException("test"));
|
||||
boolean existTableB = externalNamespacePersistService.isExistTable(tableName);
|
||||
Assert.assertFalse(existTableB);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void tenantInfoCountByTenantIdTest() {
|
||||
String tenantId = "tenantId";
|
||||
|
||||
when(dataSourceService.getDataSourceType()).thenReturn("mysql");
|
||||
|
||||
Assert.assertThrows(IllegalArgumentException.class,
|
||||
() -> externalNamespacePersistService.tenantInfoCountByTenantId(null));
|
||||
|
||||
when(jt.queryForObject(anyString(), eq(new String[] {tenantId}), eq(Integer.class))).thenReturn(null);
|
||||
int i = externalNamespacePersistService.tenantInfoCountByTenantId(tenantId);
|
||||
Assert.assertEquals(0, i);
|
||||
|
||||
when(jt.queryForObject(anyString(), eq(new String[] {tenantId}), eq(Integer.class))).thenReturn(1);
|
||||
int j = externalNamespacePersistService.tenantInfoCountByTenantId(tenantId);
|
||||
Assert.assertEquals(1, j);
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user