Java System.exit()処理を無効にして代わりにExceptionを発生させる。

バッチ処理などでSystem.exit()で終わる処理をJUnitでテストするとJVMが終了してしまう。
exit()処理の代わりにExceptionを発生させ、テストを可能にするためのクラス。
基本的にはSystem.exit()など使わずにreturnを使うんだが、古いバッチをメンテナンスするとよくある。

SystemExitStub.java

package testlib.system;

import java.security.Permission;

/**
 * System.exit()処理を無効にし代わりにExitExceptionを発生させる.<br>
 * JUnitでSystem.exitのある処理をテストを実施する場合等に使用する。<br>
 * 本クラスはスレッドセーフではないのでマルチスレッド環境では利用不可。<br>
 * シングルトンクラス。
 *
 * @author boss_ape
 */
public final class SystemExitStub extends SecurityManager
{
    /** シングルトンパターンのためのインスタンス. */
    private static final SystemExitStub INSTANCE_STUB = new SystemExitStub();

    /** 現在使われている変更前のセキュリティマネージャ. */
    private static final SecurityManager INSTANCE_SYSTEM = System.getSecurityManager();

    /**
     * デフォルトコンストラクタの禁止.
     */
    private SystemExitStub() { }

    /**
     * このクラスのインスタンスを取得.
     *
     * @return このクラスのインスタンス
     */
    public static SystemExitStub getlnstance()
    {
        return INSTANCE_STUB;
    }

    @Override
    public void checkPermission(Permission permission)
    {
        if ("exitVM".equals(permission.getName()))
            System.out.println("System.exit[exitVM]が呼ばれた");
    }

    @Override
    public void checkExit(int status)
    {
        throw new ExitException(status);
    }

    /**
     * System.exit()をExitExceptionに変更設定する.
     */
    public void change()
    {
        // JVMを終了させない
        System.setSecurityManager(INSTANCE_STUB);
    }

    /**
     * System.exit()を元に戻す.
     */
    public void reset()
    {
        System.setSecurityManager(INSTANCE_SYSTEM);
    }

    /**
     * JVMの停止の代わりに発生させるException.
     *
     * @author boss_ape
     */
    public class ExitException extends SecurityException
    {
        public int state = 0;

        public ExitException(int state)
        {
            this.state = state;
        }
    }
}

以下、使用サンプル。
Javaバッチとそのテスト。

まずはJavaバッチサンプル

SampleBatch.java

package jp.co.ivynetwork.batch;

/**
 * FugaFugaをFugeFugeするバッチ処理.<br>
 * 0:正常終了<br>
 * 1:異常終了
 *
 * @author boss_ape
 */
public class SampleBatch
{
    public static void main(String[] args)
    {
        try {
            // パラメータが何か指定されていたらException発生
            if (null == args)
                throw new NullPointerException();

        } catch (NullPointerException e) {
            System.exit(1);
        }
        System.exit(0);
    }
}

上記バッチのJUnitテスト

SampleBatchTest.java

package jp.co.ivynetwork.batch;

import static org.hamcrest.CoreMatchers.*;
import static org.junit.Assert.*;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import testlib.system.SystemExitStub;

/**
 * バッチテスト.<br>
 * テスト実行中、System.exit()でJVMが終了しないようにExitExceptionをスローする。
 *
 * @author boss_ape
 */
public class SampleBatchTest
{
    /** System.exit変更管理オブジェクト. */
    private SystemExitStub exit = SystemExitStub.getlnstance();

    /**
     * テストメソッド実行前にSystem.exit()設定を変更.
     */
    @Before
    public void changeSystem()
    {
        // System.exit()をExitExceptionに変更
        exit.change();
    }

    /**
     * テストメソッド実行後にSystem.exit()設定を戻す.
     */
    @After
    public void resetSystem()
    {
        // System.exit()設定を元に戻す
        exit.reset();
    }

    /**
     * main処理のテスト.<br>
     * パラメータnullで実行時、exit(1)で終了する事を確認
     */
    @Test
    public void testMainパラメータにnullを設定()
    {
        try {
            // バッチ実行
            SampleBatch.main(null);

        // 通常のSystem.exit()が実行された場合、JUnitのランナーそのものが終了してしまう
        } catch (SystemExitStub.ExitException e) {
            // System.exit(1)で終了していたら正常
            assertThat(e.state, is(1));
        }
    }

    /**
     * main処理のテスト.<br>
     * パラメータ指定で実行時、exit(0)で終了する事を確認
     */
    @Test
    public void testMainパラメータに値を設定()
    {
        try {
            // バッチ実行
            SampleBatch.main(new String[] { "パラメータ指定" });

        // 通常のSystem.exit()が実行された場合、JUnitのランナーそのものが終了してしまう
        } catch (SystemExitStub.ExitException e) {
            // System.exit(0)で終了していたら正常
            assertThat(e.state, is(0));
        }
    }
}