Вы можете различать тип сборки по коду, например
public static final boolean IS_TESTING =
BuildConfig.BUILD_TYPE.equalsIgnoreCase("variantName1")
public static final boolean IS_PRODUCTION =
BuildConfig.BUILD_TYPE.equalsIgnoreCase("variantName2")
затем на основе вашего типа сборки создайте свой класс.
if(IS_TESTING){
// create a version of your class
}else if(IS_PRODUCTION){
// create another version of your class
}