Fixes #120 - Adding and fixing test cases for db function

AbsaOSS · Apr 11, 2024 · 5de083a · 5de083a
1 parent 9fd6069
commit 5de083a
Showing 1 changed file with 3 additions and 13 deletions.
diff --git a/server/src/test/scala/za/co/absa/atum/server/api/controller/PartitioningControllerSpec.scala b/server/src/test/scala/za/co/absa/atum/server/api/controller/PartitioningControllerSpec.scala
@@ -35,14 +35,9 @@ class PartitioningControllerSpec extends ZIOSpecDefault with TestData {
 
   private val partitioningServiceMock = mock(classOf[PartitioningService])
 
-  when(partitioningServiceMock.createPartitioningIfNotExists(partitioningSubmitDTO1)).thenReturn(ZIO.right(AtumContextDTO(
-    partitioning = partitioningSubmitDTO1.partitioning,
-    measures = Set(MeasureDTO("count", Seq("*"))),
-    additionalData = Map.empty
-  )))
-  when(partitioningServiceMock.createPartitioningIfNotExists(partitioningSubmitDTO2))
-    .thenReturn(ZIO.left(ErrorInDataException(FunctionStatus(50, "error in data"))))
-  when(partitioningServiceMock.createPartitioningIfNotExists(partitioningSubmitDTO3))
+  when(partitioningServiceMock.returnAtumContext(partitioningSubmitDTO1))
+    .thenReturn(ZIO.succeed(atumContextDTO1))
+  when(partitioningServiceMock.returnAtumContext(partitioningSubmitDTO2))
     .thenReturn(ZIO.fail(ServiceError("boom!")))
 
   private val partitioningServiceMockLayer = ZLayer.succeed(partitioningServiceMock)
@@ -61,11 +56,6 @@ class PartitioningControllerSpec extends ZIOSpecDefault with TestData {
             result <- PartitioningController.createPartitioningIfNotExists(partitioningSubmitDTO1)
           } yield assertTrue(result == expectedAtumContextDTO)
         },
-        test("Returns expected InternalServerErrorResponse") {
-          assertZIO(PartitioningController.createPartitioningIfNotExists(partitioningSubmitDTO3).exit)(
-            failsWithA[InternalServerErrorResponse]
-          )
-        },
         test("Returns expected InternalServerErrorResponse") {
           assertZIO(PartitioningController.createPartitioningIfNotExists(partitioningSubmitDTO2).exit)(
             failsWithA[InternalServerErrorResponse]