src/chromiumos/tast/cmd/tast/run_wrapper.go

// Copyright 2018 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 main

import (
	"context"

	"chromiumos/tast/cmd/tast/internal/run"
)

// runWrapper is a wrapper that allows functions from the run package to be stubbed out for testing.
type runWrapper interface {
	// run calls run.Run.
	run(ctx context.Context, cfg *run.Config) (run.Status, []*run.EntityResult)
	// writeResults calls run.WriteResults.
	writeResults(ctx context.Context, cfg *run.Config, results []*run.EntityResult, complete bool) error
}

// realRunWrapper is a runWrapper implementation that calls the real functions in the run package.
type realRunWrapper struct{}

func (w realRunWrapper) run(ctx context.Context, cfg *run.Config) (run.Status, []*run.EntityResult) {
	return run.Run(ctx, cfg)
}

func (w realRunWrapper) writeResults(ctx context.Context, cfg *run.Config, results []*run.EntityResult, complete bool) error {
	return run.WriteResults(ctx, cfg, results, complete)
}