src/chromiumos/tast/local/graphics/glbench/cros_config.go

// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package glbench

import (
	"context"
	"path/filepath"

	"chromiumos/tast/common/testexec"
	"chromiumos/tast/errors"
	"chromiumos/tast/local/upstart"
	"chromiumos/tast/shutil"
)

// CrosConfig is the config to run in ChromeOS.
type CrosConfig struct {
	Hasty bool // If hasty is true, glbench will run in hasty mode.
}

// IsHasty returns true if the given run should run in hasty mode.
func (config *CrosConfig) IsHasty() bool {
	return config.Hasty == true
}

// SetUp initialized the environment to run glbench in ChromeOS.
func (config *CrosConfig) SetUp(ctx context.Context) error {
	// If UI is running, we must stop it and restore later.
	if err := upstart.StopJob(ctx, "ui"); err != nil {
		return errors.Wrap(err, "failed to stop ui")
	}
	return nil
}

// Run runs glbench and returns the output.
func (config *CrosConfig) Run(ctx context.Context, preValue interface{}, outDir string) (string, error) {
	args := []string{"-outdir=" + outDir, "-save", "-notemp"}
	if config.IsHasty() {
		args = append(args, "-hasty")
	}

	// glbench is the executable for performance testing.
	glbench := filepath.Join(glbenchDir, "bin/glbench")
	cmd := testexec.CommandContext(ctx, glbench, args...)
	b, err := cmd.Output(testexec.DumpLogOnError)
	if err != nil {
		return "", errors.Wrapf(err, "failed to run %s", shutil.EscapeSlice(cmd.Args))
	}
	return string(b), nil
}

// TearDown teardown the environment.
func (config *CrosConfig) TearDown(ctx context.Context) error {
	if err := upstart.EnsureJobRunning(ctx, "ui"); err != nil {
		return errors.Wrap(err, "failed to start ui")
	}
	return nil
}