refactor: add import benchmark profile filter

Author: cmilesio

internal/wire/import_bench_test.go

@@ -21,6 +21,7 @@ const (
 	importBenchBreakdown  = "WIRE_IMPORT_BENCH_BREAKDOWN"
 	importBenchScenarios  = "WIRE_IMPORT_BENCH_SCENARIOS"
 	importBenchScenarioBD = "WIRE_IMPORT_BENCH_SCENARIO_BREAKDOWN"
+	importBenchProfile    = "WIRE_IMPORT_BENCH_PROFILE"
 	stockWireCommit       = "9c25c9016f6825302537c4efdd5e897976f9c826"
 	stockWireModulePath   = "github.com/google/wire"
 	currentWireModulePath = "github.com/goforj/wire"
@@ -145,6 +146,18 @@ func TestPrintImportScenarioBenchmarkTable(t *testing.T) {
 		{localPkgs: 10, depPkgs: 25, external: true, label: "external"},
 		{localPkgs: 10, depPkgs: 100, external: true, label: "external"},
 	}
+	if filter := os.Getenv(importBenchProfile); filter != "" {
+		filtered := make([]appBenchProfile, 0, len(profiles))
+		for _, profile := range profiles {
+			if profile.label == filter {
+				filtered = append(filtered, profile)
+			}
+		}
+		if len(filtered) == 0 {
+			t.Fatalf("%s=%q did not match any benchmark profile", importBenchProfile, filter)
+		}
+		profiles = filtered
+	}
 	rows := make([]importBenchScenarioRow, 0, len(profiles)*6)
 	for _, profile := range profiles {
 		shapeFixture := createAppShapeBenchFixture(t, profile.localPkgs, profile.depPkgs, profile.external, currentWireModulePath, repoRoot)

scripts/import-benchmarks.sh

@@ -12,12 +12,13 @@ usage() {
   cat <<'EOF'
 Usage:
   scripts/import-benchmarks.sh table
-  scripts/import-benchmarks.sh scenarios
+  scripts/import-benchmarks.sh scenarios [profile]
   scripts/import-benchmarks.sh breakdown
 
 Commands:
   table      Print the 10/100/1000 import stock-vs-current benchmark table.
   scenarios  Print the stock-vs-current change-type scenario table.
+             Optional profiles: local, local-high, external.
   breakdown  Print a focused 1000-import cold/unchanged breakdown.
 EOF
 }
@@ -27,7 +28,11 @@ case "${1:-}" in
     WIRE_IMPORT_BENCH_TABLE=1 go test ./internal/wire -run TestPrintImportScaleBenchmarkTable -count=1 -v
     ;;
   scenarios)
-    WIRE_IMPORT_BENCH_SCENARIOS=1 go test ./internal/wire -run TestPrintImportScenarioBenchmarkTable -count=1 -v
+    if [[ -n "${2:-}" ]]; then
+      WIRE_IMPORT_BENCH_SCENARIOS=1 WIRE_IMPORT_BENCH_PROFILE="${2}" go test ./internal/wire -run TestPrintImportScenarioBenchmarkTable -count=1 -v
+    else
+      WIRE_IMPORT_BENCH_SCENARIOS=1 go test ./internal/wire -run TestPrintImportScenarioBenchmarkTable -count=1 -v
+    fi
     ;;
   breakdown)
     WIRE_IMPORT_BENCH_BREAKDOWN=1 go test ./internal/wire -run TestPrintImportScaleBenchmarkBreakdown -count=1 -v