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