feature[ats_ATS64164]: CPA: Retrieve workflow artifact types from TeamDef

Change-Id: I6606b69667c7d3b7a59cbea3f10ad648cb67b584

1 files changed, 21 insertions, 0 deletions

diff --git a/plugins/org.eclipse.osee.ats.core.client/src/org/eclipse/osee/ats/core/client/team/TeamWorkFlowManager.java b/plugins/org.eclipse.osee.ats.core.client/src/org/eclipse/osee/ats/core/client/team/TeamWorkFlowManager.java
index c59d07caac2..bf29b72ba88 100644
--- a/plugins/org.eclipse.osee.ats.core.client/src/org/eclipse/osee/ats/core/client/team/TeamWorkFlowManager.java
+++ b/plugins/org.eclipse.osee.ats.core.client/src/org/eclipse/osee/ats/core/client/team/TeamWorkFlowManager.java
@@ -42,6 +42,7 @@ import org.eclipse.osee.ats.core.workflow.transition.TransitionHelper;
 import org.eclipse.osee.framework.core.data.IArtifactType;
 import org.eclipse.osee.framework.core.model.type.ArtifactType;
 import org.eclipse.osee.framework.core.util.Result;
+import org.eclipse.osee.framework.jdk.core.type.OseeArgumentException;
 import org.eclipse.osee.framework.jdk.core.type.OseeCoreException;
 import org.eclipse.osee.framework.jdk.core.util.Conditions;
 import org.eclipse.osee.framework.jdk.core.util.Strings;
@@ -312,6 +313,26 @@ public class TeamWorkFlowManager {
 
    public static IArtifactType getTeamWorkflowArtifactType(IAtsTeamDefinition teamDef, Collection<IAtsActionableItem> actionableItems) throws OseeCoreException {
       IArtifactType teamWorkflowArtifactType = AtsArtifactTypes.TeamWorkflow;
+      if (teamDef.getStoreObject() != null) {
+         String artifactTypeName =
+            ((Artifact) teamDef.getStoreObject()).getSoleAttributeValue(AtsAttributeTypes.TeamWorkflowArtifactType,
+               null);
+         if (Strings.isValid(artifactTypeName)) {
+            boolean found = false;
+            for (IArtifactType type : ArtifactTypeManager.getAllTypes()) {
+               if (type.getName().equals(artifactTypeName)) {
+                  teamWorkflowArtifactType = type;
+                  found = true;
+                  break;
+               }
+            }
+            if (!found) {
+               throw new OseeArgumentException(
+                  "Team Workflow Artifact Type name [%s] off Team Definition %s could not be found.", artifactTypeName,
+                  teamDef.toStringWithId());
+            }
+         }
+      }
       ITeamWorkflowProvider teamWorkflowProvider = getTeamWorkflowProvider(teamDef, actionableItems);
       if (teamWorkflowProvider != null) {
          teamWorkflowArtifactType = teamWorkflowProvider.getTeamWorkflowArtifactType(teamDef, actionableItems);