ceremonyclient/pebble/internal/metamorphic/metarunner/main.go

67 lines
1.7 KiB
Go
Raw Normal View History

2024-01-03 07:31:42 +00:00
// Copyright 2023 The LevelDB-Go and Pebble Authors. All rights reserved. Use
// of this source code is governed by a BSD-style license that can be found in
// the LICENSE file.
// metarunner is a utility which runs metamorphic.RunOnce or Compare. It is
// equivalent to executing `internal/metamorphic.TestMeta` with `--run-dir` or
// `--compare`. It is used for code coverage instrumentation.
package main
import (
"flag"
"fmt"
"os"
"path/filepath"
"strings"
"github.com/cockroachdb/pebble/internal/metamorphic/metaflags"
"github.com/cockroachdb/pebble/metamorphic"
)
var runOnceFlags = metaflags.InitRunOnceFlags()
var _ = flag.String("test.run", "", `ignored; used for compatibility with TestMeta`)
func main() {
flag.Parse()
onceOpts := runOnceFlags.MakeRunOnceOptions()
t := &mockT{}
switch {
case runOnceFlags.Compare != "":
runDirs := strings.Split(runOnceFlags.Compare, ",")
metamorphic.Compare(t, runOnceFlags.Dir, runOnceFlags.Seed, runDirs, onceOpts...)
case runOnceFlags.RunDir != "":
// The --run-dir flag is specified either in the child process (see
// runOptions() below) or the user specified it manually in order to re-run
// a test.
metamorphic.RunOnce(t, runOnceFlags.RunDir, runOnceFlags.Seed, filepath.Join(runOnceFlags.RunDir, "history"), onceOpts...)
default:
t.Errorf("--compare or --run-dir must be used")
}
if t.Failed() {
// Make sure we return an error code.
t.FailNow()
}
}
type mockT struct {
failed bool
}
var _ metamorphic.TestingT = (*mockT)(nil)
func (t *mockT) Errorf(format string, args ...interface{}) {
t.failed = true
fmt.Fprintf(os.Stderr, format+"\n", args...)
}
func (t *mockT) FailNow() {
os.Exit(2)
}
func (t *mockT) Failed() bool {
return t.failed
}