workflow-designer-be/src/test/java/org/onap/sdc/workflow/api/types/VersionStatesFormatterTest.java

   1 package org.onap.sdc.workflow.api.types;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import static org.junit.Assert.assertNull;
   5 import static org.onap.sdc.workflow.services.types.WorkflowVersionState.CERTIFIED;
   6 import static org.onap.sdc.workflow.services.types.WorkflowVersionState.DRAFT;
   7
   8 import java.util.Collections;
   9 import java.util.stream.Collectors;
  10 import java.util.stream.Stream;
  11 import org.junit.Test;
  12 import org.junit.runner.RunWith;
  13 import org.mockito.InjectMocks;
  14 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  15
  16 @RunWith(SpringJUnit4ClassRunner.class)
  17 public class VersionStatesFormatterTest {
  18
  19     @InjectMocks
  20     private VersionStatesFormatter versionStateSet;
  21
  22     @Test
  23     public void setVersionStateInvalid() {
  24         versionStateSet.setVersionState("aaa");
  25         assertEquals(Collections.emptySet() ,versionStateSet.getVersionStates());
  26     }
  27
  28     @Test
  29     public void setVersionStateDraft() {
  30         versionStateSet.setVersionState("DRAFT");
  31         assertEquals(Collections.singleton(DRAFT), versionStateSet.getVersionStates());
  32     }
  33
  34     @Test
  35     public void setVersionStateCertified() {
  36         versionStateSet.setVersionState("CERTIFIED");
  37         assertEquals(Collections.singleton(CERTIFIED), versionStateSet.getVersionStates());
  38     }
  39
  40     @Test
  41     public void setVersionStateBoth() {
  42         versionStateSet.setVersionState("DRAFT,CERTIFIED");
  43         assertEquals(Stream.of(DRAFT, CERTIFIED).collect(Collectors.toSet()), versionStateSet.getVersionStates());
  44     }
  45 }