diff --git a/legend-engine-config/legend-engine-extensions-collection-generation/pom.xml b/legend-engine-config/legend-engine-extensions-collection-generation/pom.xml
index 5e5b259bc69..c6943e8b0db 100644
--- a/legend-engine-config/legend-engine-extensions-collection-generation/pom.xml
+++ b/legend-engine-config/legend-engine-extensions-collection-generation/pom.xml
@@ -140,6 +140,14 @@
org.finos.legend.engine
legend-engine-protocol-pure
+
+ org.finos.legend.engine
+ legend-engine-testable
+
+
+ org.finos.legend.engine
+ legend-engine-configuration
+
org.finos.legend.engine
@@ -162,6 +170,11 @@
org.finos.legend.engine
legend-engine-external-shared-format-model
+
+ org.finos.legend.engine
+ legend-engine-test-runner-mapping
+ runtime
+
@@ -391,6 +404,11 @@
org.finos.legend.engine
legend-engine-language-pure-dsl-service-pure
+
+ org.finos.legend.engine
+ legend-engine-test-runner-service
+ runtime
+
diff --git a/legend-engine-config/legend-engine-extensions-collection-generation/src/test/java/org/finos/legend/engine/extensions/collection/generation/TestExtensions.java b/legend-engine-config/legend-engine-extensions-collection-generation/src/test/java/org/finos/legend/engine/extensions/collection/generation/TestExtensions.java
index c1865923cf6..05e82753f7a 100644
--- a/legend-engine-config/legend-engine-extensions-collection-generation/src/test/java/org/finos/legend/engine/extensions/collection/generation/TestExtensions.java
+++ b/legend-engine-config/legend-engine-extensions-collection-generation/src/test/java/org/finos/legend/engine/extensions/collection/generation/TestExtensions.java
@@ -80,7 +80,13 @@
import org.finos.legend.engine.language.sql.grammar.integration.SQLGrammarParserExtension;
import org.finos.legend.engine.language.sql.grammar.integration.SQLPureGrammarComposerExtension;
import org.finos.legend.engine.language.stores.elasticsearch.v7.from.ElasticsearchGrammarParserExtension;
+import org.finos.legend.engine.plan.generation.extension.LegendPlanGeneratorExtension;
+import org.finos.legend.engine.plan.generation.extension.PlanGeneratorExtension;
import org.finos.legend.engine.protocol.bigqueryFunction.metamodel.BigQueryFunctionProtocolExtension;
+import org.finos.legend.engine.testable.extension.TestableRunnerExtension;
+import org.finos.legend.engine.testable.mapping.extension.MappingTestableRunnerExtension;
+import org.finos.legend.engine.testable.persistence.extension.PersistenceTestableRunnerExtension;
+import org.finos.legend.engine.testable.service.extension.ServiceTestableRunnerExtension;
import org.finos.legend.pure.code.core.ElasticsearchPureCoreExtension;
import org.finos.legend.engine.language.stores.elasticsearch.v7.to.ElasticsearchGrammarComposerExtension;
import org.finos.legend.engine.protocol.pure.v1.extension.PureProtocolExtension;
@@ -135,7 +141,7 @@ public void testExpectedCompilerExtensionsArePresent()
@Test
public void testPlanGeneratorExtensionArePresent()
{
- assertHasExtensions(getExpectedPlanGeneratorExtensions(), PureCoreExtension.class);
+ assertHasExtensions(getExpectedPureExtensions(), PureCoreExtension.class);
}
@Test
@@ -162,6 +168,18 @@ public void testExpectedEntitlementServiceExtensionsArePresent()
assertHasExtensions(EntitlementServiceExtensionLoader.extensions(), getExpectedEntitlementServiceExtensions(), EntitlementServiceExtension.class);
}
+ @Test
+ public void testExpectedPlanGeneratorExtension()
+ {
+ assertHasExtensions(getExpectedPlanGeneratorExtensions(), PlanGeneratorExtension.class);
+ }
+
+ @Test
+ public void testExpectedTestableRunnerExtension()
+ {
+ assertHasExtensions(getExpectedTestableExtensions(), TestableRunnerExtension.class);
+ }
+
@Test
public void testCodeRepositories()
{
@@ -295,6 +313,22 @@ protected Iterable extends Class extends PureProtocolExtension>> getExpected
;
}
+ protected Iterable extends Class extends TestableRunnerExtension>> getExpectedTestableExtensions()
+ {
+ // DO NOT DELETE ITEMS FROM THIS LIST (except when replacing them with something equivalent)
+ return Lists.mutable.>empty()
+ .with(ServiceTestableRunnerExtension.class)
+ .with(MappingTestableRunnerExtension.class)
+ .with(PersistenceTestableRunnerExtension.class);
+ }
+
+ protected Iterable extends Class extends PlanGeneratorExtension>> getExpectedPlanGeneratorExtensions()
+ {
+ return Lists.mutable.>empty()
+ .with(LegendPlanGeneratorExtension.class);
+
+ }
+
protected Iterable extends Class extends GenerationExtension>> getExpectedGenerationExtensions()
{
// DO NOT DELETE ITEMS FROM THIS LIST (except when replacing them with something equivalent)
@@ -402,7 +436,7 @@ protected Iterable extends Class extends CompilerExtension>> getExpectedComp
;
}
- protected Iterable extends Class extends PureCoreExtension>> getExpectedPlanGeneratorExtensions()
+ protected Iterable extends Class extends PureCoreExtension>> getExpectedPureExtensions()
{
// DO NOT DELETE ITEMS FROM THIS LIST (except when replacing them with something equivalent)
return Lists.mutable.>empty()
@@ -422,8 +456,7 @@ protected Iterable extends Class extends PureCoreExtension>> getExpectedPlan
.with(XMLJavaBindingPureCoreExtension.class)
.with(ServicePureCoreExtension.class)
.with(RelationalJavaBindingPureCoreExtension.class)
- .with(ArrowPureCoreExtension.class)
- ;
+ .with(ArrowPureCoreExtension.class);
}
protected Iterable extends Class extends ExternalFormatExtension>>> getExpectedExternalFormatExtensions()