FORGE-2316: Added excluded stacks to Stack

forge · Jan 14, 2016 · 8599d68 · 8599d68
1 parent d4fd826
commit 8599d68
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/projects/api/src/main/java/org/jboss/forge/addon/projects/stacks/StackBuilder.java b/projects/api/src/main/java/org/jboss/forge/addon/projects/stacks/StackBuilder.java
@@ -55,6 +55,13 @@ public StackBuilder includes(Stack stack)
       return this;
    }
 
+   public StackBuilder excludes(Stack stack)
+   {
+      excludedFacets.addAll(stack.getIncludedFacets());
+      excludedFacets.addAll(stack.getExcludedFacets());
+      return this;
+   }
+
    public StackBuilder excludes(Class<? extends ProjectFacet> facet)
    {
       excludedFacets.add(facet);

diff --git a/projects/tests/src/test/java/org/jboss/forge/addon/projects/stacks/StackBuilderTest.java b/projects/tests/src/test/java/org/jboss/forge/addon/projects/stacks/StackBuilderTest.java
@@ -66,4 +66,18 @@ public void testIncludesStack()
       Assert.assertTrue(stack.supports(ProjectFacetC.class));
    }
 
+   @Test
+   public void testExcludesStack()
+   {
+      Stack anotherStack = StackBuilder.stack("Another Test Stack")
+               .includes(ProjectFacetC.class);
+      Stack stack = StackBuilder.stack("Test Stack")
+               .includes(ProjectFacetA.class)
+               .includes(anotherStack)
+               .includes(ProjectFacetB.class)
+               .excludes(anotherStack);
+      Assert.assertTrue(stack.supports(ProjectFacetA.class));
+      Assert.assertTrue(stack.supports(ProjectFacetB.class));
+      Assert.assertFalse(stack.supports(ProjectFacetC.class));
+   }
 }