1 files changed, 93 insertions, 0 deletions
diff --git a/plugins/org.eclipse.osee.orcs.core.test/src/org/eclipse/osee/orcs/core/internal/relation/impl/RelationNodeLoaderImplTest.java b/plugins/org.eclipse.osee.orcs.core.test/src/org/eclipse/osee/orcs/core/internal/relation/impl/RelationNodeLoaderImplTest.java
new file mode 100644
index 00000000000..3c00ceb8c77
--- /dev/null
+++ b/plugins/org.eclipse.osee.orcs.core.test/src/org/eclipse/osee/orcs/core/internal/relation/impl/RelationNodeLoaderImplTest.java
@@ -0,0 +1,93 @@
+/*******************************************************************************
+ * Copyright (c) 2013 Boeing.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     Boeing - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.osee.orcs.core.internal.relation.impl;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static org.mockito.MockitoAnnotations.initMocks;
+import java.util.Arrays;
+import java.util.Collection;
+import org.eclipse.osee.framework.core.data.IOseeBranch;
+import org.eclipse.osee.framework.core.enums.CoreBranches;
+import org.eclipse.osee.framework.core.enums.LoadLevel;
+import org.eclipse.osee.framework.core.exception.OseeCoreException;
+import org.eclipse.osee.orcs.OrcsSession;
+import org.eclipse.osee.orcs.core.ds.DataLoader;
+import org.eclipse.osee.orcs.core.ds.DataLoaderFactory;
+import org.eclipse.osee.orcs.core.internal.artifact.Artifact;
+import org.eclipse.osee.orcs.core.internal.graph.GraphBuilder;
+import org.eclipse.osee.orcs.core.internal.graph.GraphBuilderFactory;
+import org.eclipse.osee.orcs.core.internal.graph.GraphData;
+import org.eclipse.osee.orcs.core.internal.relation.RelationNode;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.mockito.Mock;
+
+/**
+ * @author Megumi Telles
+ */
+public class RelationNodeLoaderImplTest {
+
+   @Rule
+   public ExpectedException thrown = ExpectedException.none();
+
+   // @formatter:off
+   @Mock private DataLoaderFactory dataLoaderFactory;
+   @Mock private GraphBuilderFactory graphBuilderFactory;
+   
+   @Mock private OrcsSession session;
+   @Mock private GraphData graph;
+   @Mock private GraphBuilder builder;
+   @Mock private DataLoader loader;
+   
+   @Mock private Artifact artifact;
+   // @formatter:on
+
+   private static final IOseeBranch BRANCH = CoreBranches.COMMON;
+   private static final int TRANSACTION_ID = 231214214;
+   private static final Collection<Integer> ids = Arrays.asList(4, 5, 6, 7);
+
+   private RelationNodeLoaderImpl relationNode;
+
+   @Before
+   public void setUp() throws Exception {
+      initMocks(this);
+
+      relationNode = new RelationNodeLoaderImpl(dataLoaderFactory, graphBuilderFactory);
+
+      when(graph.getBranch()).thenReturn(BRANCH);
+      when(graph.getTransaction()).thenReturn(TRANSACTION_ID);
+   }
+
+   @Test
+   public void testLoadNodes() throws OseeCoreException {
+      Iterable<Artifact> artifacts = Arrays.asList(artifact);
+
+      when(dataLoaderFactory.fromBranchAndArtifactIds(session, BRANCH, ids)).thenReturn(loader);
+      when(graphBuilderFactory.createBuilderForGraph(graph)).thenReturn(builder);
+      when(builder.getArtifacts()).thenReturn(artifacts);
+
+      Iterable<RelationNode> actual = relationNode.loadNodes(session, graph, ids, LoadLevel.FULL);
+
+      verify(dataLoaderFactory).fromBranchAndArtifactIds(session, BRANCH, ids);
+      verify(graphBuilderFactory).createBuilderForGraph(graph);
+
+      verify(loader).setLoadLevel(LoadLevel.FULL);
+      verify(loader).fromTransaction(TRANSACTION_ID);
+      verify(loader).load(null, builder);
+      verify(builder).getArtifacts();
+
+      assertEquals(artifacts, actual);
+   }
+}