Add mock for getting transcription results

NDLANO · Dec 16, 2024 · e0109cd · e0109cd
1 parent 7d7e95e
commit e0109cd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/audio-api/src/test/scala/no/ndla/audioapi/service/TranscriptionServiceTest.scala b/audio-api/src/test/scala/no/ndla/audioapi/service/TranscriptionServiceTest.scala
@@ -47,7 +47,7 @@ class TranscriptionServiceTest extends UnitSuite with TestEnvironment {
 
     val result = transcriptionService.getTranscription(videoId, language)
 
-    result should be(Success(fakeJobStatus.toString))
+    result should be(Success(Left(fakeJobStatus.toString)))
   }
 
   test("transcribeVideo returns Success when transcription is started") {
@@ -56,6 +56,7 @@ class TranscriptionServiceTest extends UnitSuite with TestEnvironment {
     val maxSpeakers        = 2
     val fakeS3Object       = mock[NdlaS3Object]
     val fakeTranscribeMock = mock[StartTranscriptionJobResponse]
+    when(transcribeClient.getTranscriptionJob(any)).thenReturn(Success(mock[GetTranscriptionJobResponse]))
     when(s3TranscribeClient.getObject(any)).thenReturn(Success(fakeS3Object))
     when(transcriptionService.getAudioExtractionStatus(videoId, language)).thenReturn(Success(()))
     when(transcribeClient.startTranscriptionJob(any, any, any, any, any, any, any, any))