refactor: Add allowDeleted method to DeletionFlag

2 files changed, 14 insertions, 0 deletions

diff --git a/plugins/org.eclipse.osee.framework.core.test/src/org/eclipse/osee/framework/core/test/enums/DeletionFlagTest.java b/plugins/org.eclipse.osee.framework.core.test/src/org/eclipse/osee/framework/core/test/enums/DeletionFlagTest.java
index d288eadb118..6dd75ac7d10 100644
--- a/plugins/org.eclipse.osee.framework.core.test/src/org/eclipse/osee/framework/core/test/enums/DeletionFlagTest.java
+++ b/plugins/org.eclipse.osee.framework.core.test/src/org/eclipse/osee/framework/core/test/enums/DeletionFlagTest.java
@@ -31,5 +31,15 @@ public class DeletionFlagTest {
    public void testExcludeDeletion() {
       DeletionFlag flag = DeletionFlag.EXCLUDE_DELETED;
       Assert.assertFalse(flag.areDeletedAllowed());
+
+   }
+
+   @Test
+   public void testAllowDeleted() {
+      DeletionFlag actual = DeletionFlag.allowDeleted(true);
+      Assert.assertEquals(DeletionFlag.INCLUDE_DELETED, actual);
+
+      actual = DeletionFlag.allowDeleted(false);
+      Assert.assertEquals(DeletionFlag.EXCLUDE_DELETED, actual);
    }
 }
diff --git a/plugins/org.eclipse.osee.framework.core/src/org/eclipse/osee/framework/core/enums/DeletionFlag.java b/plugins/org.eclipse.osee.framework.core/src/org/eclipse/osee/framework/core/enums/DeletionFlag.java
index bbbb2f699da..c5178dbeb6d 100644
--- a/plugins/org.eclipse.osee.framework.core/src/org/eclipse/osee/framework/core/enums/DeletionFlag.java
+++ b/plugins/org.eclipse.osee.framework.core/src/org/eclipse/osee/framework/core/enums/DeletionFlag.java
@@ -21,4 +21,8 @@ public enum DeletionFlag {
    public boolean areDeletedAllowed() {
       return this == INCLUDE_DELETED;
    }
+
+   public static DeletionFlag allowDeleted(boolean areAllowed) {
+      return areAllowed ? DeletionFlag.INCLUDE_DELETED : DeletionFlag.EXCLUDE_DELETED;
+   }
 }