bug[ats_TW10511]: Fix Publish STRS to include requirements with name changes

Change-Id: Ia0c2e9e18b10b0018fa1cc1ed10c4a35744bc4b8

2 files changed, 14 insertions, 1 deletions

diff --git a/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/RequirementTraceabilityData.java b/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/RequirementTraceabilityData.java
index cea2cd73466..4d0182954b9 100644
--- a/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/RequirementTraceabilityData.java
+++ b/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/RequirementTraceabilityData.java
@@ -154,6 +154,10 @@ public class RequirementTraceabilityData {
       return this.requirementData.getRequirementFromTraceMarkIncludeStructuredRequirements(traceMark);
    }
 
+   public Artifact getRequirementFromArtifactId(ArtifactId artId) {
+      return this.requirementData.getRequirementFromArtifactId(artId);
+   }
+
    /**
     * @return the requirementsToCodeUnits
     */
diff --git a/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/RequirementData.java b/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/RequirementData.java
index 37dcea10102..eb51a6b4ffb 100644
--- a/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/RequirementData.java
+++ b/plugins/org.eclipse.osee.define/src/org/eclipse/osee/define/traceability/data/RequirementData.java
@@ -22,8 +22,8 @@ import java.util.Set;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.osee.define.traceability.TraceabilityExtractor;
 import org.eclipse.osee.framework.core.data.ArtifactId;
-import org.eclipse.osee.framework.core.data.IArtifactType;
 import org.eclipse.osee.framework.core.data.BranchId;
+import org.eclipse.osee.framework.core.data.IArtifactType;
 import org.eclipse.osee.framework.core.enums.CoreArtifactTypes;
 import org.eclipse.osee.framework.jdk.core.type.Pair;
 import org.eclipse.osee.framework.skynet.core.artifact.Artifact;
@@ -111,6 +111,15 @@ public class RequirementData extends BaseTraceDataCache {
       return allRequirementsMap.get(extractor.getCanonicalRequirementName(traceMark));
    }
 
+   public Artifact getRequirementFromArtifactId(ArtifactId artId) {
+      for (Artifact art : directRequirements) {
+         if (art.equals(artId)) {
+            return art;
+         }
+      }
+      return null;
+   }
+
    /**
     * Get Requirement Artifact based on traceMark mark if it fails, check if trace mark is a structured requirement and
     * try again