Set environment for unit test