如何在Junit PowerMockito中使用System.getenv

Question

我可以從Junit中模擬System.getenv值，但是當我執行測試用例時-在我的服務類中，System.getevn值作為null出現。 不知道我在做什么錯。 請找到我的測試服務類和junit類。

可以幫我解決這個問題嗎-為什么在我的實際服務類別中未設置該值？

TestService.java

public class TestService {

    public TestService() throws Exception {

        loadTestMethod();

    }

    private void loadTestMethod() {

        System.out.println("Environment vairlable : " + System.getenv("my_key_name"));
        System.setProperty("app.key", System.getenv("my_key_name"));

    }

}

TestServiceTest.java

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PowerMockIgnore;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
@PrepareForTest(value = { System.class })
@PowerMockIgnore("javax.management.*")

public class TestServiceTest {

    @Mock
    TestService testService;

    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);

        PowerMockito.mockStatic(System.class);
        PowerMockito.when(System.getenv("my_key_name")).thenReturn("Testing");
        System.out.println("Junit Environment vairlable : " + System.getenv("my_key_name"));
        testService = new TestService();

    }

    @Test
    public void myServiceTest() {

    }
}

Answer 1

僅僅因為PowerMockito允許我們模擬靜態並不意味着我們應該這樣做。

您的類依賴於靜態實現問題，在大多數情況下，這些問題使孤立進行單元測試變得困難。

考慮遵循Explicit Dependency Principle

方法和類應明確要求（通常通過方法參數或構造函數參數）它們需要的任何協作對象才能正常運行。

創建所需功能的抽象

public interface SystemWrapper {
    //The "standard" output stream
    PrintStream out;
    //Gets the value of the specified environment variable
    string getenv(string name);
    //Sets the system property indicated by the specified key.
    string setProperty(String key, String value);
}

該實現將封裝對靜態系統類的實際調用

public class SystemWrapperImplementation implements SystemWrapper {
    //The "standard" output stream
    public final PrintStream out = System.out;

    //Gets the value of the specified environment variable
    public string getenv(string name) {
        return System.getenv(name);
    }

    //Sets the system property indicated by the specified key.
    public string setProperty(String key, String value) {
        return System.setProperty(key, value);
    }
}

然后，您的依賴類將需要重構以包含抽象

public class TestService {
    private SystemWrapper system;

    public TestService(SystemWrapper system) throws Exception {
        this.system = system;
        string key = "app.key";
        string name = "my_key_name";
        loadTestMethod(key, name);
    }

    private void loadTestMethod(string key, string name) {
        string environmentVariable = system.getenv(name);
        system.out.println("Environment variable : " + environmentVariable);
        system.setProperty(key, environmentVariable);
    }
}

現在進行測試，您可以根據需要模擬必要的依賴關系，而不會產生任何不利影響。 然后，在調用實際代碼時將在生產中使用該實現。

最后，我建議不要讓您的構造函數拋出異常。 構造函數應主要用於變量分配。