jmokitによる環境変数(System.getenv())の追加

jmockit_System_getenv.java

package mymock.test;

import junit.framework.TestCase;
import mockit.Mocked;
import mockit.NonStrictExpectations;
import java.util.Map;

public class SystemTest extends TestCase {
	private Map<String, String> envs = null;
	
	public void test() {
		envs = System.getenv();
		
		new NonStrictExpectations() {
			@Mocked("getenv") System system; // getenv()のみ部分モック
			{
				System.getenv();
				returns(envs);
				
				for (Map.Entry<String, String> env : envs.entrySet()) {
					System.getenv(env.getKey());
					returns(env.getValue());
				}
				
				// 環境変数追加
				System.getenv("hoge");
				returns("hoge");
		
			}
		};
		assertEquals("hoge", System.getenv("hoge"); // 追加した環境変数確認
		for (Map.Entry<String, String> env : envs.entrySet()) { // 既存の環境変数確認
			System.out.println(env.getKey());
			System.out.println(System.getenv(env.getKey()));
			System.out.println();
		}
	}
}