Examples with DiskUtil com.alibaba.nacos.config.server.utils.DiskUtil used on opensource projects

Example 1 with DiskUtil

use of com.alibaba.nacos.config.server.utils.DiskUtil in project nacos by alibaba.

the class DiskServiceUnitTest method setUp.

@Before
public void setUp() throws IOException {
    this.tempFile = File.createTempFile("diskServiceTest", "tmp");
    this.path = tempFile.getParent();
    this.diskService = new DiskUtil();
}

Example 2 with DiskUtil

use of com.alibaba.nacos.config.server.utils.DiskUtil in project nacos by alibaba.

the class ConfigServletInnerTest method testDoGetConfigV2.

@Test
public void testDoGetConfigV2() throws Exception {
    final MockedStatic<DiskUtil> diskUtil = Mockito.mockStatic(DiskUtil.class);
    final MockedStatic<ConfigCacheService> configCacheService = Mockito.mockStatic(ConfigCacheService.class);
    final MockedStatic<PropertyUtil> propertyUtil = Mockito.mockStatic(PropertyUtil.class);
    when(ConfigCacheService.tryReadLock(anyString())).thenReturn(1);
    // isBeta: false
    CacheItem cacheItem = new CacheItem("test");
    cacheItem.setBeta(false);
    List<String> ips4Beta = new ArrayList<>();
    ips4Beta.add("localhost");
    cacheItem.setIps4Beta(ips4Beta);
    when(ConfigCacheService.getContentCache(anyString())).thenReturn(cacheItem);
    // if tag is blank and direct read is true
    when(PropertyUtil.isDirectRead()).thenReturn(true);
    ConfigInfoWrapper configInfoWrapper = new ConfigInfoWrapper();
    configInfoWrapper.setDataId("test");
    configInfoWrapper.setGroup("test");
    configInfoWrapper.setContent("tag is blank and direct read is true");
    when(persistService.findConfigInfo(anyString(), anyString(), anyString())).thenReturn(configInfoWrapper);
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setRemoteAddr("localhost:8080");
    request.addHeader(CLIENT_APPNAME_HEADER, "test");
    MockHttpServletResponse response = new MockHttpServletResponse();
    String actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("tag is blank and direct read is true", response.getContentAsString());
    // if tag is blank and direct read is false
    when(PropertyUtil.isDirectRead()).thenReturn(false);
    response = new MockHttpServletResponse();
    File file = tempFolder.newFile("test.txt");
    when(DiskUtil.targetFile(anyString(), anyString(), anyString())).thenReturn(file);
    actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("", response.getContentAsString());
    // if tag is not blank and direct read is true
    when(PropertyUtil.isDirectRead()).thenReturn(true);
    ConfigInfoTagWrapper configInfoTagWrapper = new ConfigInfoTagWrapper();
    configInfoTagWrapper.setDataId("test");
    configInfoTagWrapper.setGroup("test");
    configInfoTagWrapper.setContent("tag is not blank and direct read is true");
    when(persistService.findConfigInfo4Tag(anyString(), anyString(), anyString(), anyString())).thenReturn(configInfoTagWrapper);
    response = new MockHttpServletResponse();
    actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "test", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("tag is not blank and direct read is true", response.getContentAsString());
    // if tag is not blank and direct read is false
    when(PropertyUtil.isDirectRead()).thenReturn(false);
    response = new MockHttpServletResponse();
    when(DiskUtil.targetTagFile(anyString(), anyString(), anyString(), anyString())).thenReturn(file);
    actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "test", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("", response.getContentAsString());
    // if use auto tag and direct read is true
    when(PropertyUtil.isDirectRead()).thenReturn(true);
    Map<String, String> tagMd5 = new HashMap<>();
    tagMd5.put("auto-tag-test", "auto-tag-test");
    cacheItem.setTagMd5(tagMd5);
    request.addHeader("Vipserver-Tag", "auto-tag-test");
    configInfoTagWrapper.setContent("auto tag mode and direct read is true");
    when(persistService.findConfigInfo4Tag(anyString(), anyString(), anyString(), eq("auto-tag-test"))).thenReturn(configInfoTagWrapper);
    response = new MockHttpServletResponse();
    actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("auto tag mode and direct read is true", response.getContentAsString());
    // if use auto tag and direct read is false
    when(PropertyUtil.isDirectRead()).thenReturn(false);
    response = new MockHttpServletResponse();
    actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("", response.getContentAsString());
    diskUtil.close();
    configCacheService.close();
    propertyUtil.close();
}

Example 3 with DiskUtil

use of com.alibaba.nacos.config.server.utils.DiskUtil in project nacos by alibaba.

the class ConfigServletInnerTest method testDoGetConfigV1.

@Test
public void testDoGetConfigV1() throws Exception {
    final MockedStatic<DiskUtil> diskUtil = Mockito.mockStatic(DiskUtil.class);
    final MockedStatic<ConfigCacheService> configCacheService = Mockito.mockStatic(ConfigCacheService.class);
    final MockedStatic<PropertyUtil> propertyUtil = Mockito.mockStatic(PropertyUtil.class);
    when(ConfigCacheService.tryReadLock(anyString())).thenReturn(1);
    // isBeta: true
    CacheItem cacheItem = new CacheItem("test");
    cacheItem.setBeta(true);
    List<String> ips4Beta = new ArrayList<>();
    ips4Beta.add("localhost");
    cacheItem.setIps4Beta(ips4Beta);
    when(ConfigCacheService.getContentCache(anyString())).thenReturn(cacheItem);
    // if direct read is true
    when(PropertyUtil.isDirectRead()).thenReturn(true);
    ConfigInfoBetaWrapper configInfoBetaWrapper = new ConfigInfoBetaWrapper();
    configInfoBetaWrapper.setDataId("test");
    configInfoBetaWrapper.setGroup("test");
    configInfoBetaWrapper.setContent("isBeta:true, direct read: true");
    when(persistService.findConfigInfo4Beta(anyString(), anyString(), anyString())).thenReturn(configInfoBetaWrapper);
    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setRemoteAddr("localhost:8080");
    request.addHeader(CLIENT_APPNAME_HEADER, "test");
    MockHttpServletResponse response = new MockHttpServletResponse();
    String actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("true", response.getHeader("isBeta"));
    Assert.assertEquals("isBeta:true, direct read: true", response.getContentAsString());
    // if direct read is false
    when(PropertyUtil.isDirectRead()).thenReturn(false);
    File file = tempFolder.newFile("test.txt");
    when(DiskUtil.targetBetaFile(anyString(), anyString(), anyString())).thenReturn(file);
    response = new MockHttpServletResponse();
    actualValue = configServletInner.doGetConfig(request, response, "test", "test", "test", "", "true", "localhost");
    Assert.assertEquals(HttpServletResponse.SC_OK + "", actualValue);
    Assert.assertEquals("true", response.getHeader("isBeta"));
    Assert.assertEquals("", response.getContentAsString());
    diskUtil.close();
    configCacheService.close();
    propertyUtil.close();
}