plugins/org.eclipse.osee.client.integration.tests/src/org/eclipse/osee/client/integration/tests/integration/orcs/rest/RelationEndpointTest.java

/*********************************************************************
 * Copyright (c) 2004, 2007 Boeing
 *
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *     Boeing - initial API and implementation
 **********************************************************************/

package org.eclipse.osee.client.integration.tests.integration.orcs.rest;

import static org.eclipse.osee.framework.core.enums.CoreArtifactTokens.DefaultHierarchyRoot;
import java.util.List;
import org.eclipse.osee.client.demo.DemoChoice;
import org.eclipse.osee.framework.core.client.OseeClient;
import org.eclipse.osee.framework.core.data.ArtifactId;
import org.eclipse.osee.framework.core.data.ArtifactToken;
import org.eclipse.osee.framework.core.enums.CoreArtifactTypes;
import org.eclipse.osee.framework.core.enums.CoreRelationTypes;
import org.eclipse.osee.framework.core.enums.DemoBranches;
import org.eclipse.osee.framework.core.util.OsgiUtil;
import org.eclipse.osee.framework.skynet.core.artifact.search.ArtifactQuery;
import org.eclipse.osee.orcs.rest.model.ArtifactEndpoint;
import org.eclipse.osee.orcs.rest.model.RelationEndpoint;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * @author Hugo Trejo, Torin Grenda, David Miller
 */
public class RelationEndpointTest {

   //@Rule
   //public OseeClientIntegrationRule integration = new OseeClientIntegrationRule(OSEE_CLIENT_DEMO);

   private static RelationEndpoint relationEndpoint;
   private static ArtifactEndpoint artifactEndpoint;

   @BeforeClass
   public static void testSetup() {
      OseeClient oseeclient = OsgiUtil.getService(DemoChoice.class, OseeClient.class);
      relationEndpoint = oseeclient.getRelationEndpoint(DemoBranches.SAW_PL_Working_Branch);
      artifactEndpoint = oseeclient.getArtifactEndpoint(DemoBranches.SAW_PL_Working_Branch);

   }

   /**
    * Simple test for the getRelation rest call. Checking for the following cases <br/>
    * 1. Can create an acceptable relation 2. Cannot create a relation that is not acceptable
    */

   @Test
   public void testGetRelationByType() {
      // Create two artifactIds for softwareRequirements MSWord
      ArtifactId parentArtifact = DefaultHierarchyRoot;
      ArtifactToken sideA = artifactEndpoint.createArtifact(DemoBranches.SAW_PL_Working_Branch,
         CoreArtifactTypes.SoftwareRequirementMsWord, parentArtifact, "requirementA");

      ArtifactToken sideB = artifactEndpoint.createArtifact(DemoBranches.SAW_PL_Working_Branch,
         CoreArtifactTypes.SoftwareRequirementMsWord, parentArtifact, "requirementB");

      // Act
      relationEndpoint.createRelationByType(sideA, sideB, CoreRelationTypes.RequirementTrace);

      // Check if relation was created
      List<ArtifactId> artIdsSideA =
         ArtifactQuery.createQueryBuilder(DemoBranches.SAW_PL_Working_Branch).andId(sideA).andExists(
            CoreRelationTypes.RequirementTrace_HigherLevelRequirement).getIds();
      List<ArtifactId> artIdsSideB =
         ArtifactQuery.createQueryBuilder(DemoBranches.SAW_PL_Working_Branch).andId(sideB).andExists(
            CoreRelationTypes.RequirementTrace_LowerLevelRequirement).getIds();

      boolean relationCreatedSideA = false;
      boolean relationCreatedSideB = false;
      if (artIdsSideA.size() > 0) {
         for (ArtifactId artifactId : artIdsSideA) {
            if (artifactId.equals(sideA)) {
               relationCreatedSideA = true;
            }
         }
      }
      if (artIdsSideB.size() > 0) {
         for (ArtifactId artifactId : artIdsSideB) {
            if (artifactId.equals(sideB)) {
               relationCreatedSideB = true;
            }
         }
      }

      Assert.assertTrue(relationCreatedSideA && relationCreatedSideB);
   }

   @Test
   public void testGetArtifactsByRelationType() {
      ArtifactToken parentArtifact = DefaultHierarchyRoot;
      List<ArtifactToken> arts = relationEndpoint.getRelatedHierarchy(parentArtifact);
      Assert.assertFalse(arts.isEmpty());
   }
}