Ensure sdf raw zip exists for tests to work

pixlise · Nov 5, 2024 · c075e98 · c075e98
1 parent 5b0937f
commit c075e98
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/api/dataimport/sdfToRSI/scanSDF_test.go b/api/dataimport/sdfToRSI/scanSDF_test.go
@@ -3,6 +3,8 @@ package sdfToRSI
 import "fmt"
 
 func Example_scanSDF() {
+	ensureSDFRawExists()
+
 	refs, err := scanSDF("./test-data/BadPath.txt")
 	fmt.Printf("%v|%v\n", len(refs), err)
 

diff --git a/api/dataimport/sdfToRSI/sdfToRSI_test.go b/api/dataimport/sdfToRSI/sdfToRSI_test.go
@@ -9,6 +9,18 @@ import (
 )
 
 func Example_ConvertSDFtoRSI() {
+	ensureSDFRawExists()
+
+	fmt.Printf("Mkdir output: %v\n", os.MkdirAll("./output", 0644))
+	files, rtts, err := ConvertSDFtoRSIs("./test-data/sdf_raw.txt", "./output/")
+	fmt.Printf("%v, %v: %v\n", files, rtts, err)
+
+	// Output:
+	// Mkdir output: <nil>
+	// [RSI-208536069.csv RSI-208601602.csv], [208536069 208601602]: <nil>
+}
+
+func ensureSDFRawExists() {
 	// If it's not here, unzip it
 	_, err := os.Stat("./test-data/sdf_raw.txt")
 	if err != nil {
@@ -39,12 +51,4 @@ func Example_ConvertSDFtoRSI() {
 			}
 		}
 	}
-
-	fmt.Printf("Mkdir output: %v", os.MkdirAll("./output", 0644))
-	files, rtts, err := ConvertSDFtoRSIs("./test-data/sdf_raw.txt", "./output/")
-	fmt.Printf("%v, %v: %v\n", files, rtts, err)
-
-	// Output:
-	// Mkdir output: <nil>
-	// [RSI-208536069.csv RSI-208601602.csv], [208536069 208601602]: <nil>
 }