@@ -121,28 +121,33 @@ protected ProcessType getProcessType() {
121121 * @return the additional initialization
122122 */
123123 public static AdditionalInitialization withCapabilities (final String ... capabilities ) {
124- return new ManagementAdditionalInitialization () {
125-
126- @ Override
127- protected void initializeExtraSubystemsAndModel (ExtensionRegistry extensionRegistry , Resource rootResource , ManagementResourceRegistration rootRegistration , RuntimeCapabilityRegistry capabilityRegistry ) {
128- super .initializeExtraSubystemsAndModel (extensionRegistry , rootResource , rootRegistration , capabilityRegistry );
129- registerCapabilities (capabilityRegistry , capabilities );
130- }
131- };
124+ return withCapabilities (Stability .DEFAULT , capabilities );
132125 }
133126
134127 /**
135128 * Creates a {@link org.jboss.as.subsystem.test.AdditionalInitialization.ManagementAdditionalInitialization} with
136129 * the given {@link org.jboss.as.controller.capability.RuntimeCapability capabilities} registered, making it
137130 * possible for subsystems under test to require them. No runtime API will be available, but that should not
138131 * be needed for a {@link org.jboss.as.controller.RunningMode#ADMIN_ONLY} test.
139- *
132+ * @param schema a subsystem schema
140133 * @param capabilities the capabilities
141134 * @return the additional initialization
142135 */
143136 public static <S extends SubsystemSchema <S >> AdditionalInitialization withCapabilities (S schema , String ... capabilities ) {
144- return new ManagementAdditionalInitialization (schema ) {
137+ return withCapabilities (schema .getStability (), capabilities );
138+ }
145139
140+ /**
141+ * Creates a {@link org.jboss.as.subsystem.test.AdditionalInitialization.ManagementAdditionalInitialization} with
142+ * the given {@link org.jboss.as.controller.capability.RuntimeCapability capabilities} registered, making it
143+ * possible for subsystems under test to require them. No runtime API will be available, but that should not
144+ * be needed for a {@link org.jboss.as.controller.RunningMode#ADMIN_ONLY} test.
145+ * @param stability the stability level of the test process
146+ * @param capabilities the capabilities
147+ * @return the additional initialization
148+ */
149+ public static <S extends SubsystemSchema <S >> AdditionalInitialization withCapabilities (Stability stability , String ... capabilities ) {
150+ return new ManagementAdditionalInitialization (stability ) {
146151 @ Override
147152 protected void initializeExtraSubystemsAndModel (ExtensionRegistry extensionRegistry , Resource rootResource , ManagementResourceRegistration rootRegistration , RuntimeCapabilityRegistry capabilityRegistry ) {
148153 super .initializeExtraSubystemsAndModel (extensionRegistry , rootResource , rootRegistration , capabilityRegistry );
0 commit comments