support test env

Added support to test environment.