🐛 Resolve NoSuchElementException in MappingExecutionEndpoint test

srdc · Sep 17, 2024 · be7dc82 · be7dc82
1 parent 8469f99
commit be7dc82
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tofhir-server/src/test/scala/io/tofhir/server/endpoint/MappingExecutionEndpointTest.scala b/tofhir-server/src/test/scala/io/tofhir/server/endpoint/MappingExecutionEndpointTest.scala
@@ -374,12 +374,10 @@ class MappingExecutionEndpointTest extends BaseEndpointTest with OnFhirTestConta
         val success = waitForCondition(45) {
           fsSinkFolder.listFiles.exists(_.getName.contains("results.csv")) && {
             // read the csv file created in the file system
-            val csvFile: File = fsSinkFolder.listFiles.find(_.getName.contains("results.csv"))
-              .get.listFiles().find(file => file.getName.endsWith(".csv"))
-              .get
+            val outputFolder: File = fsSinkFolder.listFiles.find(_.getName.contains("results.csv")).get
             val results = sparkSession.read
               .option("header","true")
-              .csv(csvFile.getPath)
+              .csv(outputFolder.getPath)
             // verify the row count
             results.count() == 1
             // verify the content of row