PredictionContainerGenerator extends AbstractApplicationSection { public PredictionContainerGenerator withAddons(AddonsConfig addonsConfig) { this.addonsConfig = addonsConfig; return this; } PredictionContainerGenerator(String applicationCanonicalName, List<PMMLResource> resources); PredictionContainerGenerator withAddons(AddonsConfig addonsConfig); @Override ClassOrInterfaceDeclaration classDeclaration(); void addPredictionRulesMapperClass(String predictionRulesMapperClass);  }
@Test void withAddons() { PredictionContainerGenerator retrieved = predictionContainerGenerator.withAddons(null); assertEquals(retrieved, predictionContainerGenerator); assertNull(predictionContainerGenerator.addonsConfig); predictionContainerGenerator.withAddons(AddonsConfig.DEFAULT); assertEquals(AddonsConfig.DEFAULT, predictionContainerGenerator.addonsConfig); }