From 0d16c9ac992426b561be499ba1cef70b6a6941e4 Mon Sep 17 00:00:00 2001
From: Simon Spinner <simon.spinner@uni-wuerzburg.de>
Date: Tue, 21 Jun 2016 17:08:19 +0200
Subject: [PATCH] Add assembly contexts to list of undeployed after removal of
 container.

---
 .../prisma/agent/wildfly/service/ModelExtractionService.java   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools.descartes.prisma.agent.wildfly/src/main/java/tools/descartes/prisma/agent/wildfly/service/ModelExtractionService.java b/tools.descartes.prisma.agent.wildfly/src/main/java/tools/descartes/prisma/agent/wildfly/service/ModelExtractionService.java
index cb13ec2..2661f94 100644
--- a/tools.descartes.prisma.agent.wildfly/src/main/java/tools/descartes/prisma/agent/wildfly/service/ModelExtractionService.java
+++ b/tools.descartes.prisma.agent.wildfly/src/main/java/tools/descartes/prisma/agent/wildfly/service/ModelExtractionService.java
@@ -264,6 +264,9 @@ public class ModelExtractionService extends AgentController implements Service<M
 			synchronized (skeleton) {
 				// Remove all obsolete deployment contexts
 				skeleton.getContainers().remove(wildFlyContainer);
+				for (DeploymentContext ctx : skeleton.getDeployment().getDeploymentContexts()) {
+					undeployedAssemblyContexts.add(ctx.getAssemblyContext());
+				}
 				skeleton.getDeployment().getDeploymentContexts().clear();
 				apply(skeleton);
 				wildFlyContainer = null;
-- 
GitLab