Remove powermock from services (#7860)

This commit is contained in:
Vishesh 2023-08-16 12:35:18 +05:30 committed by GitHub
parent 5559668f12
commit fe70f4d801
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
7 changed files with 45 additions and 49 deletions

View File

@ -35,7 +35,7 @@ import org.mockito.InjectMocks;
import org.mockito.Mock; import org.mockito.Mock;
import org.mockito.Mockito; import org.mockito.Mockito;
import org.mockito.Spy; import org.mockito.Spy;
import org.powermock.modules.junit4.PowerMockRunner; import org.mockito.junit.MockitoJUnitRunner;
import org.springframework.test.util.ReflectionTestUtils; import org.springframework.test.util.ReflectionTestUtils;
import java.util.ArrayList; import java.util.ArrayList;
@ -47,7 +47,7 @@ import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.ArgumentMatchers.nullable; import static org.mockito.ArgumentMatchers.nullable;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
@RunWith(PowerMockRunner.class) @RunWith(MockitoJUnitRunner.class)
public class NonStrictHostAntiAffinityProcessorTest { public class NonStrictHostAntiAffinityProcessorTest {
@Spy @Spy

View File

@ -21,20 +21,20 @@ import static org.mockito.Mockito.when;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.powermock.modules.junit4.PowerMockRunner; import org.mockito.junit.MockitoJUnitRunner;
import org.powermock.reflect.Whitebox;
import org.springframework.test.util.ReflectionTestUtils;
import streamer.Element; import streamer.Element;
import streamer.SocketWrapper; import streamer.SocketWrapper;
@RunWith(PowerMockRunner.class) @RunWith(MockitoJUnitRunner.class)
public class ClientTest { public class ClientTest {
@Test(expected = NullPointerException.class) @Test(expected = NullPointerException.class)
public void testAssemblePipelineWhenMainElementIsNull() throws Exception { public void testAssemblePipelineWhenMainElementIsNull() throws Exception {
SocketWrapper socketMock = mock(SocketWrapper.class); SocketWrapper socketMock = mock(SocketWrapper.class);
when(socketMock.getId()).thenReturn("socket"); when(socketMock.getId()).thenReturn("socket");
Whitebox.setInternalState(Client.class, "socket", socketMock); ReflectionTestUtils.setField(Client.class, "socket", socketMock);
Element main = null; Element main = null;
Client.assemblePipeline(main); Client.assemblePipeline(main);

View File

@ -19,33 +19,30 @@ package com.cloud.consoleproxy;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.mockito.Mock; import org.mockito.MockedConstruction;
import org.mockito.MockedStatic;
import org.mockito.Mockito; import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito; import org.mockito.junit.MockitoJUnitRunner;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import java.util.Map; import java.util.Map;
@RunWith(PowerMockRunner.class) @RunWith(MockitoJUnitRunner.class)
public class ConsoleProxyHttpHandlerHelperTest { public class ConsoleProxyHttpHandlerHelperTest {
@Mock
ConsoleProxyPasswordBasedEncryptor encryptor;
@Test @Test
@PrepareForTest({ConsoleProxy.class, ConsoleProxyHttpHandlerHelper.class})
public void testQueryMapExtraParameter() throws Exception { public void testQueryMapExtraParameter() throws Exception {
PowerMockito.mockStatic(ConsoleProxy.class); try (MockedStatic<ConsoleProxy> ignore = Mockito.mockStatic(ConsoleProxy.class);
PowerMockito.when(ConsoleProxy.getEncryptorPassword()).thenReturn("password"); MockedConstruction<ConsoleProxyPasswordBasedEncryptor> ignored = Mockito.mockConstruction(ConsoleProxyPasswordBasedEncryptor.class, (mock, context) -> {
PowerMockito.whenNew(ConsoleProxyPasswordBasedEncryptor.class).withArguments(Mockito.anyString()).thenReturn(encryptor); Mockito.when(mock.decryptObject(Mockito.eq(ConsoleProxyClientParam.class), Mockito.anyString())).thenReturn(null);
Mockito.when(encryptor.decryptObject(Mockito.eq(ConsoleProxyClientParam.class), Mockito.anyString())).thenReturn(null); });) {
Mockito.when(ConsoleProxy.getEncryptorPassword()).thenReturn("password");
String extraValidationToken = "test-token"; String extraValidationToken = "test-token";
String query = String.format("token=SOME_TOKEN&extra=%s", extraValidationToken); String query = String.format("token=SOME_TOKEN&extra=%s", extraValidationToken);
Map<String, String> queryMap = ConsoleProxyHttpHandlerHelper.getQueryMap(query); Map<String, String> queryMap = ConsoleProxyHttpHandlerHelper.getQueryMap(query);
Assert.assertTrue(queryMap.containsKey("extra")); Assert.assertTrue(queryMap.containsKey("extra"));
Assert.assertEquals(extraValidationToken, queryMap.get("extra")); Assert.assertEquals(extraValidationToken, queryMap.get("extra"));
}
} }
} }

View File

@ -22,54 +22,50 @@ import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doThrow; import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.spy; import static org.mockito.Mockito.spy;
import java.io.BufferedWriter;
import java.io.File; import java.io.File;
import java.io.FileWriter;
import java.io.StringWriter;
import org.apache.cloudstack.storage.command.DeleteCommand; import org.apache.cloudstack.storage.command.DeleteCommand;
import org.apache.cloudstack.storage.to.TemplateObjectTO; import org.apache.cloudstack.storage.to.TemplateObjectTO;
import org.apache.log4j.Level; import org.apache.log4j.Level;
import org.junit.Assert; import org.junit.Assert;
import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.mockito.Mockito; import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito; import org.mockito.Spy;
import org.powermock.core.classloader.annotations.PowerMockIgnore; import org.mockito.junit.MockitoJUnitRunner;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import com.cloud.test.TestAppender; import com.cloud.test.TestAppender;
@RunWith(PowerMockRunner.class) @RunWith(MockitoJUnitRunner.class)
@PowerMockIgnore({ "javax.xml.*", "org.xml.*"})
public class NfsSecondaryStorageResourceTest { public class NfsSecondaryStorageResourceTest {
@Spy
private NfsSecondaryStorageResource resource; private NfsSecondaryStorageResource resource;
@Before
public void setUp() {
resource = new NfsSecondaryStorageResource();
}
@Test @Test
@PrepareForTest(NfsSecondaryStorageResource.class)
public void testSwiftWriteMetadataFile() throws Exception { public void testSwiftWriteMetadataFile() throws Exception {
String filename = "testfile"; String metaFileName = "test_metadata_file";
try { try {
String expected = "uniquename=test\nfilename=" + filename + "\nsize=100\nvirtualsize=1000"; String uniqueName = "test_unique_name";
String filename = "test_filename";
long size = 1024L;
long virtualSize = 2048L;
StringWriter stringWriter = new StringWriter(); File metaFile = resource.swiftWriteMetadataFile(metaFileName, uniqueName, filename, size, virtualSize);
BufferedWriter bufferWriter = new BufferedWriter(stringWriter);
PowerMockito.whenNew(BufferedWriter.class).withArguments(any(FileWriter.class)).thenReturn(bufferWriter);
resource.swiftWriteMetadataFile(filename, "test", filename, 100, 1000); Assert.assertTrue(metaFile.exists());
Assert.assertEquals(metaFileName, metaFile.getName());
Assert.assertEquals(expected, stringWriter.toString()); String expectedContent = "uniquename=" + uniqueName + "\n" +
"filename=" + filename + "\n" +
"size=" + size + "\n" +
"virtualsize=" + virtualSize;
String actualContent = new String(java.nio.file.Files.readAllBytes(metaFile.toPath()));
Assert.assertEquals(expectedContent, actualContent);
} finally { } finally {
File remnance = new File(filename); File metaFile = new File(metaFileName);
remnance.delete(); metaFile.delete();
} }
} }