如何在maven中测试之前取消设置环境变量

时间:2016-03-24 02:17:44

标签: java maven unit-testing testing

我有一些单元测试,它取决于环境变量。

我想在使用maven进行测试之前取消设置这些环境变量。

问题:我该如何实现?

1 个答案:

答案 0 :(得分:1)

不幸的是,在这种情况下,您无法使用environmentVariablesMaven Surefire Plugin选项,主要是因为它只能用于添加新环境变量但不能覆盖(或重置) ,实际上等于覆盖空值)现有变量。

另请注意:在Maven中包含ant run并在测试之前执行也不会有效。

建议的解决方案基于Java,特别是在另一篇SO帖子中提出的this approach。虽然不建议使用应用程序代码,但 hack 可能适用于测试代码。

您可以将以下类添加到测试代码中(在src/test/java下):

package com.sample;

import java.lang.reflect.Field;
import java.util.Map;

public class EnvHack {

    @SuppressWarnings("unchecked")
    public static void resetEnvironmentVariable(String name, String value) throws Exception {
        Class<?> processEnvironmentClass = Class.forName("java.lang.ProcessEnvironment");
        Field theEnvironmentField = processEnvironmentClass.getDeclaredField("theEnvironment");
        theEnvironmentField.setAccessible(true);
        Map<String, String> env = (Map<String, String>) theEnvironmentField.get(null);
        env.put(name, value);
        Field theCaseInsensitiveEnvironmentField = processEnvironmentClass
                .getDeclaredField("theCaseInsensitiveEnvironment");
        theCaseInsensitiveEnvironmentField.setAccessible(true);
        Map<String, String> cienv = (Map<String, String>) theCaseInsensitiveEnvironmentField.get(null);
        cienv.put(name, value);
    }

    public static void main(String[] args) throws Exception {
        resetEnvironmentVariable("test_prop", "test_value");
    }
}

上面的类基本上是攻击Java API来改变内存环境变量的值。因此,它们可以设置为不同的值,重置甚至实际上未设置(将其从地图中删除)。

由于该类现在是测试代码的一部分,因此您有以下几种选择:

  • 在某个JUnit测试用例的@Before方法(或@BeforeClasswith a certain difference)中使用该类(在相关类的每个JUnit方法之前)
  • 在JUnit测试方法中使用它(自定义和缩小用法)
  • 在任何执行的JUnit测试之前运行它的main方法(以更全局的方式),如下所述(并且可能回答这个问题,即使其他场景也值得一提,imho)。

让我们看看每种可能的解决方案。

在某个JUnit测试用例的@Before方法中使用该类

package com.sample;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class MainTest {

    @Before
    public void init() throws Exception {
        EnvHack.resetEnvironmentVariable("test_prop", "test_value");
    }

    @Test
    public void testEnvProperty() throws Exception {
        String s = System.getenv("test_prop");
        Assert.assertEquals(s, "test_value");
    }

}

这个解决方案可以在每个测试类中使用,并且当一组测试(方法)共享相同的要求时(建议:如果它们没有,则可能是一个提示,可能应该移出一些方法)。

在JUnit测试方法中使用

package com.sample;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class MainTest {

    @Before
    public void init() throws Exception {
        EnvHack.resetEnvironmentVariable("test_prop", "test_value");
    }

    @Test
    public void testEnvProperty() throws Exception {
        EnvHack.resetEnvironmentVariable("test_prop", "test_value2");
        String s = System.getenv("test_prop");
        Assert.assertEquals(s, "test_value2");
    }

}

此解决方案提供最高的自由度:您可以准确地在需要的地方使用相关变量,尽管可能会遇到代码重复,但它也可能会强制执行测试隔离。

在执行任何JUnit测试之前运行其main方法

<plugin>
    <groupId>org.codehaus.mojo</groupId>
    <artifactId>exec-maven-plugin</artifactId>
    <version>1.4.0</version>
    <executions>
        <execution>
            <phase>process-test-classes</phase>
            <goals>
                <goal>java</goal>
            </goals>
            <configuration>
                <mainClass>com.sample.EnvHack</mainClass>
                <classpathScope>test</classpathScope>
            </configuration>
        </execution>
    </executions>
</plugin>

请注意我们在这种情况下正在做的事情:

  • 我们正在调用javaExec Maven Plugin目标来实际调用 env hack 类的mainClass
  • 我们在classPathScope设置为test的情况下调用它,以使其对Enforcer插件可见
  • 我们将它作为process-test-classes阶段的一部分运行,以确保它在test阶段之前执行,因此在任何测试之前执行。

此解决方案集中了整个准备环境过程,适用于所有测试。

另一方面,您也可以考虑在测试中使用模拟。这不是一个集中的选项(除非你编码),但可以提供进一步的提示,因此值得一提。以下是通过PowerMock重置环境变量的示例代码

package com.sample;

import org.junit.*;
import org.junit.runner.RunWith;
import org.mockito.Mockito;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
@PrepareForTest(System.class)
public class EnvTest {

    @Before
    public void init() {
        PowerMockito.mockStatic(System.class);
        Mockito.when(System.getenv("TEST_PROP")).thenReturn("TEST_VALUE");
    }

    @Test
    public void test() {
        String var = System.getenv("TEST_PROP");
        Assert.assertEquals("TEST_VALUE", var);
    }
}

这类似于上面提出的第一种和第二种方法。

请注意,要使其工作,您需要将以下依赖项添加到POM中:

<dependencies>
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.11</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-api-mockito</artifactId>
        <version>1.5</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-core</artifactId>
        <version>1.5</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-module-junit4</artifactId>
        <version>1.5</version>
        <scope>test</scope>
    </dependency>
</dependencies>

一般说明:确实不建议使用未完全隔离且取决于环境变量是否存在(或不存在)的测试。您可能会遇到维护问题,或者为同事或您自己的未来进行更糟糕的故障排除。所以,如果你真的需要它,最好正确记录它。