检测是否在运行JUnit单元测试

  在写单元测试时遇到一个问题：同一份代码，在单元测试时，我希望他连接到测试数据库；在正式运行时，我希望他连接到正式数据库。

  因此，我要判断当前运行的是测试还是生产环境。

 

  首先，我配置了两套数据库连接环境，一套用于单元测试，一套用于生产环境。

  然后，通过运行时判断是否在运行单元测试，而决定读取哪一套的配置，判断代码为：

public class JUnitUtils {
    private static Boolean isRunningTest = null;

    /**
     * 检测是否在运行单元测试
     * @return
     */
    public static Boolean IsRunningTest() {
        if (null == isRunningTest) {
            StackTraceElement[] stackTrace = Thread.currentThread().getStackTrace();
            List statckList = Arrays.asList(stackTrace);
            for (Iterator i = statckList.iterator(); i.hasNext(); ) {
                String stackString = i.next().toString();
                if (stackString.lastIndexOf("junit.runners") > -1) {
                    isRunningTest = true;
                    return isRunningTest;
                }
            }
            isRunningTest = false;
            return isRunningTest;
        } else {
            return isRunningTest;
        }
    }
}

 

评论

15 楼 siyuan 2012-11-20  

这个真的可以profile或-D参数就可以解决了

14 楼 aaa1aaaaa 2012-11-07  

使用Derby嘛，也就是Java DB，不久解决了啊

13 楼 zhukewen_java 2012-11-06  

cavenfeng 写道

好像我描述的不太清楚，再完整描述一下这个需求的来源。

首先，我们的工程是Maven的工程，那默认的情况下，package（打包）项目的时候是会运行单元测试的，对吧？

问题就出在这里了，开发人员package项目，有时候会忘记把数据库连接从开发库改回单元测试库的，因此单元测试就会把一些开发库的数据删除，或者插入一些单元测试的数据，这就是我们痛苦的地方。

因此，我就想把单元测试和命令行RUN两种环境完全分开，用不相见！

maven的更好弄了，本身就可以支持不同的profile.当然，人为要破坏是不能够防止的

12 楼 cavenfeng 2012-11-06  

好像我描述的不太清楚，再完整描述一下这个需求的来源。

首先，我们的工程是Maven的工程，那默认的情况下，package（打包）项目的时候是会运行单元测试的，对吧？

问题就出在这里了，开发人员package项目，有时候会忘记把数据库连接从开发库改回单元测试库的，因此单元测试就会把一些开发库的数据删除，或者插入一些单元测试的数据，这就是我们痛苦的地方。

因此，我就想把单元测试和命令行RUN两种环境完全分开，用不相见！

11 楼 mfkvfn 2012-11-06  

cavenfeng 写道

mfkvfn 写道

就搞个配置文件不就行了。平时测试时使用测试数据库。
正式部署时修改下配置文件的内容。

单元测试在持续集成环境自动跑，不能随便修改配置呢

集成环境中配置文件连接测试环境。
正式部署运行时先修改一下配置文件再启动运行不就行了。

10 楼 javatomccsss 2012-11-06  

写得不全啊！

9 楼 zhukewen_java 2012-11-05  

其实可以搞两套配置环境的，单元测试的时候加载用于测试的那一套环境

8 楼 cavenfeng 2012-11-05  

nhyjq 写道

关系搞反了
读哪套配置应该是你执行程序（无论是单元测试还是生产运行）时同时指定的

无论框架级的spring，还是辅助工具ant,maven之类都有执行时指定配置的方法

最不济你执行程序时还能传个环境参数吧，test/production之类，不就好了？

是这样的，被测试的代码里面包含数据库的操作，ORM使用IBatis，IBatis的配置在代码中指向了某个ibatis的配置。
那我需要测试这些代码，在单元测试中没办法修改IBatis的配置呢

7 楼 cavenfeng 2012-11-05  

gosure 写道

框架不应该侵入代码！

是有点bad smell

6 楼 cavenfeng 2012-11-05  

mfkvfn 写道

就搞个配置文件不就行了。平时测试时使用测试数据库。
正式部署时修改下配置文件的内容。

单元测试在持续集成环境自动跑，不能随便修改配置呢

5 楼 cavenfeng 2012-11-05  

shenliuyang 写道

崩溃、、、、     还写几套数据源、、、、   maven中有Profile 这个好东西。

同学，ORM使用IBatis，请问Maven Profile怎么实现JUnit和生产环境读取不同的Profile？

4 楼 shenliuyang 2012-11-05  

崩溃、、、、     还写几套数据源、、、、   maven中有Profile 这个好东西。

3 楼 mfkvfn 2012-11-05  

就搞个配置文件不就行了。平时测试时使用测试数据库。
正式部署时修改下配置文件的内容。

2 楼 nhyjq 2012-11-05  

关系搞反了
读哪套配置应该是你执行程序（无论是单元测试还是生产运行）时同时指定的

无论框架级的spring，还是辅助工具ant,maven之类都有执行时指定配置的方法

最不济你执行程序时还能传个环境参数吧，test/production之类，不就好了？

1 楼 gosure 2012-11-05  

框架不应该侵入代码！