pinpoint/go/run_benchmark/benchmark_test_factory.go

package run_benchmark

import (
	"strings"

	"go.skia.org/infra/go/skerr"

	bc "go.skia.org/infra/pinpoint/go/bot_configs"
)

type BenchmarkTest interface {
	GetCommand() []string
}

// NewBenchmarkTest returns a BenchmarkTest based on the request parameters.
// The Configuration (bot) is used alongside the benchmark to determine the
// isolate target for that combination. Based on the isolate target,
func NewBenchmarkTest(commit, botConfig, browser, benchmark, story, storyTags string, extraArgs []string) (BenchmarkTest, error) {
	config, err := bc.GetBotConfig(botConfig, false)
	if err != nil {
		return nil, skerr.Wrapf(err, "Failed to fetch bot configs to create benchmark test")
	}
	target, err := bc.GetIsolateTarget(botConfig, benchmark)
	if err != nil {
		return nil, skerr.Wrapf(err, "Failed to get isolate target to create the benchmark test")
	}

	// Few targets could have suffixes, especially for Android.
	// For example, 'performance_test_suite_android_clank_monochrome_64_32_bundle'
	if strings.Contains(target, "performance_test_suite") || strings.Contains(target, "telemetry_perf_tests") || target == "performance_webview_test_suite" {
		return &telemetryTest{
			benchmark: benchmark,
			browser:   config.Browser,
			commit:    commit,
			story:     story,
			storyTags: storyTags,
			extraArgs: extraArgs,
		}, nil
	}
	return nil, skerr.Fmt("Unsupported test target %s", target)
}