1 files changed, 12 insertions, 4 deletions
diff --git a/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/WorkflowRepositoryTest.java b/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/WorkflowRepositoryTest.java
index 4f31e4b38c..7d1b8d0d96 100644
--- a/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/WorkflowRepositoryTest.java
+++ b/mso-catalog-db/src/test/java/org/onap/so/db/catalog/data/repository/WorkflowRepositoryTest.java
@@ -15,8 +15,6 @@
 package org.onap.so.db.catalog.data.repository;
 
 import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertTrue;
 import java.util.List;
 import org.junit.Assert;
 import org.junit.Test;
@@ -33,7 +31,7 @@ public class WorkflowRepositoryTest extends BaseTest {
     public void findByArtifactUuid_ValidUuid_ExpectedOutput() throws Exception {
         Workflow workflow = workflowRepository.findByArtifactUUID("5b0c4322-643d-4c9f-b184-4516049e99b1");
 
-        assertEquals("artifactName", "testingWorkflow", workflow.getArtifactName());
+        assertEquals("artifactName", "testingWorkflow.bpmn", workflow.getArtifactName());
     }
 
     @Test
@@ -43,7 +41,17 @@ public class WorkflowRepositoryTest extends BaseTest {
         Assert.assertTrue(workflows != null);
         Assert.assertTrue(workflows.size() != 0);
 
-        Assert.assertTrue("testingWorkflow".equals(workflows.get(0).getArtifactName()));
+        Assert.assertTrue("testingWorkflow.bpmn".equals(workflows.get(0).getArtifactName()));
+    }
+
+    @Test
+    public void findBySourceTest() throws Exception {
+        List<Workflow> workflows = workflowRepository.findBySource("sdc");
+
+        Assert.assertTrue(workflows != null);
+        Assert.assertTrue(workflows.size() != 0);
+
+        Assert.assertTrue("testingWorkflow.bpmn".equals(workflows.get(0).getArtifactName()));
     }
 
 }