blob: 9d3bb11dfd2b9b522858eb9678c50dd1a56eeaf5 [file] [log] [blame]
// 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 (
// 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