Fixed the Junit main test, also added custom configuration support too.
Fixed the Junit main test, also added custom configuration support too.