From 33e358bfae1486ebeb0c1c888415558e8bd27656 Mon Sep 17 00:00:00 2001
From: George Bartolomey <george@bh4.ru>
Date: Sun, 15 Sep 2024 12:02:27 +0300
Subject: [PATCH] [#14] Add locode_generate benchmark

Signed-off-by: George Bartolomey <george@bh4.ru>
---
 locode_generate_test.go | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 locode_generate_test.go

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{})
+	}
+}