fix: add use of sourceId after flag.parse()

uc-cdis · Oct 3, 2024 · 8c0e3e5 · 8c0e3e5
1 parent 6507dae
commit 8c0e3e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/data_generator/datagenerator.go b/tests/data_generator/datagenerator.go
@@ -325,8 +325,8 @@ func main() {
 	environment := flag.String("e", "development", "Environment/prefix of config .yaml file name")
 	testData := flag.String("d", "models_tests_data_config", "Prefix of test data config .yaml file name")
 	sourceId := flag.Int("s", 1, "Source id for Omop DB")
-	tests.SetTestSourceId(*sourceId)
 	flag.Parse()
+	tests.SetTestSourceId(*sourceId)
 	Init(*environment, *sourceId)
 	log.Printf("\n\n=============== GENERATING TEST DATA BASED ON CONFIG ============================\n\n")
 	RunDataGeneration(*testData)