feature[ats_25LS9]: Update presenter tests

2 files changed, 21 insertions, 7 deletions

diff --git a/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockArtifactProvider.java b/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockArtifactProvider.java
index 4dd85676d0e..c3a70586d5c 100644
--- a/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockArtifactProvider.java
+++ b/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockArtifactProvider.java
@@ -60,9 +60,9 @@ public class MockArtifactProvider implements ArtifactProvider {
    public List<ReadableArtifact> getRelatedArtifacts(ReadableArtifact art, IRelationTypeSide relationTypeSide) {
       if (art instanceof MockArtifact) {
          MockArtifact mArt = (MockArtifact) art;
-         return null;
+         return (List<ReadableArtifact>) mArt.getRelatedArtifacts(relationTypeSide);
       } else {
-         return null;
+         return Collections.emptyList();
       }
    }
 
@@ -70,7 +70,7 @@ public class MockArtifactProvider implements ArtifactProvider {
    public ReadableArtifact getRelatedArtifact(ReadableArtifact art, IRelationTypeSide relationTypeSide) {
       if (art instanceof MockArtifact) {
          MockArtifact mArt = (MockArtifact) art;
-         return null;
+         return mArt.getRelatedArtifacts(relationTypeSide).iterator().next();
       } else {
          return null;
       }
@@ -89,8 +89,7 @@ public class MockArtifactProvider implements ArtifactProvider {
    @Override
    public Collection<RelationType> getValidRelationTypes(ReadableArtifact art) {
       if (art instanceof MockArtifact) {
-         MockArtifact mArt = (MockArtifact) art;
-         return null;
+         return ((MockArtifact) art).getValidRelationTypes();
       } else {
          return Collections.emptyList();
       }
diff --git a/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockRelationComponent.java b/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockRelationComponent.java
index 652f363ae74..1403b341e6d 100644
--- a/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockRelationComponent.java
+++ b/plugins/org.eclipse.osee.display.presenter.mocks/src/org/eclipse/osee/display/presenter/mocks/MockRelationComponent.java
@@ -26,6 +26,15 @@ public class MockRelationComponent implements RelationComponent {
    private final List<ViewArtifact> leftRelations = new LinkedList<ViewArtifact>();
    private final List<ViewArtifact> rightRelations = new LinkedList<ViewArtifact>();
    private String errorMessage;
+   private String leftName, rightName;
+
+   public String getLeftName() {
+      return leftName;
+   }
+
+   public String getRightName() {
+      return rightName;
+   }
 
    public ViewArtifact getArtifact() {
       return artifact;
@@ -76,20 +85,26 @@ public class MockRelationComponent implements RelationComponent {
 
    @Override
    public void addLeftRelated(ViewArtifact id) {
-      leftRelations.add(id);
+      if (id != null) {
+         leftRelations.add(id);
+      }
    }
 
    @Override
    public void addRightRelated(ViewArtifact id) {
-      rightRelations.add(id);
+      if (id != null) {
+         rightRelations.add(id);
+      }
    }
 
    @Override
    public void setLeftName(String name) {
+      leftName = name;
    }
 
    @Override
    public void setRightName(String name) {
+      rightName = name;
    }
 
 }
\ No newline at end of file