diff --git a/locode_generate_test.go b/locode_generate_test.go new file mode 100644 index 0000000..e568756 --- /dev/null +++ b/locode_generate_test.go @@ -0,0 +1,30 @@ +package main + +import ( + "flag" + "testing" +) + +var ( + in = flag.String(locodeGenerateInputFlag, "", "List of paths to UN/LOCODE tables (csv)") + subdiv = flag.String(locodeGenerateSubDivFlag, "", "Path to UN/LOCODE subdivision database (csv)") + airports = flag.String(locodeGenerateAirportsFlag, "", "Path to OpenFlights airport database (csv)") + countries = flag.String(locodeGenerateCountriesFlag, "", "Path to OpenFlights country database (csv)") + continents = flag.String(locodeGenerateContinentsFlag, "", "Path to continent polygons (GeoJSON)") + out = flag.String(locodeGenerateOutputFlag, "", "Target path for generated database") +) + +func BenchmarkLocodeGenerate(b *testing.B) { + locodeGenerateInPaths = append(locodeGenerateInPaths, *in) + locodeGenerateSubDivPath = *subdiv + locodeGenerateAirportsPath = *airports + locodeGenerateCountriesPath = *countries + locodeGenerateContinentsPath = *continents + locodeGenerateOutPath = *out + + b.ResetTimer() + + for i := 0; i < b.N; i++ { + locodeGenerateCmd.Run(locodeGenerateCmd, []string{}) + } +}