This is an automated email from the ASF dual-hosted git repository.

heneveld pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/brooklyn-server.git

commit c79b72ae412bc676220011978ca21aa1163a80c6
Author: Alex Heneveld <[email protected]>
AuthorDate: Tue Dec 13 01:33:57 2022 +0000

    allow AbstractBlueprintTest to let mgmt context be decorated with custom 
types
---
 .../brooklyn/launcher/blueprints/AbstractBlueprintTest.java    | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git 
a/launcher/src/test/java/org/apache/brooklyn/launcher/blueprints/AbstractBlueprintTest.java
 
b/launcher/src/test/java/org/apache/brooklyn/launcher/blueprints/AbstractBlueprintTest.java
index fda84338cd..c5bf51cf04 100644
--- 
a/launcher/src/test/java/org/apache/brooklyn/launcher/blueprints/AbstractBlueprintTest.java
+++ 
b/launcher/src/test/java/org/apache/brooklyn/launcher/blueprints/AbstractBlueprintTest.java
@@ -319,7 +319,7 @@ public abstract class AbstractBlueprintTest {
                 newMgmt)
                 .setConfigKeyAtManagmentContext();
 
-        return newMgmt;
+        return decorateManagementContext(newMgmt);
     }
 
     // -----
@@ -332,12 +332,16 @@ public abstract class AbstractBlueprintTest {
     /** @return A started management context -- with or without rebind 
depending on the value of {@link #isRebindEnabled()} */
     protected ManagementContext createOrigManagementContext() {
         if (isRebindEnabled()) {
-            return createBuilderForRebindingManagementContext().buildStarted();
+            return 
decorateManagementContext(createBuilderForRebindingManagementContext().buildStarted());
         } else {
-            return LocalManagementContextForTests.newInstance();
+            return 
decorateManagementContext(LocalManagementContextForTests.newInstance());
         }
     }
 
+    protected ManagementContext decorateManagementContext(ManagementContext 
mgmt) {
+        return mgmt;
+    }
+
     protected Application createAndStartApplication(String input) throws 
Exception {
         Application app = this.createApplicationUnstarted(input);
         app.invoke(Startable.START, MutableMap.of()).get();

Reply via email to