Skip to content

Commit c06d256

Browse files
author
Dingane Hlaluku
committed
Add unit test for diagnostics file list builder
1 parent a4c4711 commit c06d256

File tree

1 file changed

+66
-0
lines changed

1 file changed

+66
-0
lines changed
Lines changed: 66 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,66 @@
1+
package org.apache.cloudstack.diagnostics;
2+
3+
import java.util.ArrayList;
4+
import java.util.List;
5+
import java.util.stream.Collectors;
6+
import java.util.stream.Stream;
7+
8+
import com.cloud.vm.VMInstanceVO;
9+
import com.cloud.vm.VirtualMachine;
10+
import org.apache.cloudstack.framework.config.ConfigKey;
11+
import org.junit.After;
12+
import org.junit.Before;
13+
import org.junit.Test;
14+
import org.junit.runner.RunWith;
15+
import org.mockito.InjectMocks;
16+
import org.mockito.Mock;
17+
import org.mockito.Mockito;
18+
import org.mockito.runners.MockitoJUnitRunner;
19+
20+
import static org.junit.Assert.*;
21+
22+
@RunWith(MockitoJUnitRunner.class)
23+
public class DiagnosticsFilesListFactoryTest {
24+
25+
private ConsoleProxyDiagnosticFiles proxyDiagnosticFiles;
26+
27+
@Mock
28+
private VMInstanceVO vmInstance;
29+
30+
@InjectMocks
31+
private DiagnosticsFilesListFactory listFactory = new DiagnosticsFilesListFactory();
32+
33+
@Before
34+
public void setUp() throws Exception {
35+
Mockito.when(vmInstance.getType()).thenReturn(VirtualMachine.Type.ConsoleProxy);
36+
}
37+
38+
@After
39+
public void tearDown() throws Exception {
40+
Mockito.reset(vmInstance);
41+
}
42+
43+
@Test
44+
public void testgetDiagnosticsFilesListCpVmDataTypeList() {
45+
List<String> dataTypeList = new ArrayList<>();
46+
dataTypeList.add("/var/log/auth.log");
47+
dataTypeList.add("/etc/dnsmasq.conf");
48+
dataTypeList.add("[IPTABLES]");
49+
dataTypeList.add("[IFCONFIG]");
50+
51+
List<String> files = listFactory.getDiagnosticsFilesList(dataTypeList, vmInstance).generateFileList();
52+
53+
assertEquals(files, dataTypeList);
54+
55+
}
56+
57+
@Test
58+
public void testDiagnisticsFileListDefaultsCpvm() {
59+
List<String> filesList = listFactory.getDiagnosticsFilesList(null, vmInstance).generateFileList();
60+
61+
ConfigKey configKey = proxyDiagnosticFiles.CpvmDefaultSupportedFiles;
62+
String[] defaultFileArray = configKey.defaultValue().split(",");
63+
64+
assertEquals(filesList.size(), defaultFileArray.length);
65+
}
66+
}

0 commit comments

Comments
 (0)