ceremonyclient/pebble/internal/metamorphic/crossversion/crossversion_test.go
Cassandra Heart 2e2a1e4789
v1.2.0 (#31)
2024-01-03 01:31:42 -06:00

410 lines
12 KiB
Go

// Copyright 2022 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.
// Package crossversion builds on the metamorphic testing implemented in
// internal/metamorphic, performing metamorphic testing across versions of
// Pebble. This improves test coverage of upgrade and migration code paths.
package crossversion
import (
"bytes"
"context"
"flag"
"fmt"
"io"
"math/rand"
"os"
"os/exec"
"path/filepath"
"strconv"
"strings"
"sync"
"testing"
"time"
"unicode"
"github.com/cockroachdb/errors"
"github.com/cockroachdb/pebble/metamorphic"
"github.com/cockroachdb/pebble/vfs"
"github.com/stretchr/testify/require"
)
var (
factor int
seed int64
versions pebbleVersions
artifactsDir string
streamOutput bool
)
func init() {
// NB: If you add new command-line flags, you should update the
// reproductionCommand function.
flag.Int64Var(&seed, "seed", 0,
`a pseudorandom number generator seed`)
flag.IntVar(&factor, "factor", 10,
`the number of data directories to carry forward
from one version's run to the subsequent version's runs.`)
flag.Var(&versions, "version",
`a comma-separated 3-tuple defining a Pebble version to test.
The expected format is <label>,<SHA>,<test-binary-path>.
The label should be a human-readable label describing the
version, for example, 'CRDB-22.1'. The SHA indicates the
exact commit sha of the version, and may be abbreviated.
The test binary path must point to a test binary of the
internal/metamorphic package built on the indicated SHA.
A test binary may be built with 'go test -c'.
This flag should be provided multiple times to indicate
the set of versions to test. The order of the versions
is significant and database states generated from earlier
versions will be used to initialize runs of subsequent
versions.`)
flag.StringVar(&artifactsDir, "artifacts", "",
`the path to a directory where test artifacts should be
moved on failure. Defaults to the current working directory.`)
flag.BoolVar(&streamOutput, "stream-output", false,
`stream TestMeta output to standard output`)
}
func reproductionCommand() string {
return fmt.Sprintf(
"SEED=%d FACTOR=%d ./scripts/run-crossversion-meta.sh %s\n",
seed, factor, versions.String(),
)
}
// TestMetaCrossVersion performs cross-version metamorphic testing.
//
// It runs tests against the internal/metamorphic test binaries specified with
// multiple instances of the -version flag, exercising upgrade and migration
// code paths.
//
// More specifically, assume we are passed the following versions:
//
// --version 22.2,<sha>,meta-22-2.test --version 23.1,<sha>,meta-23-1.test
//
// TestMetaCrossVersion will:
// - run TestMeta on meta-22-2.test;
// - retain a random subset of the resulting directories (each directory is a
// store after a sequence of operations);
// - run TestMeta on meta-23.1.test once for every retained directory from the
// previous version (using it as initial state).
func TestMetaCrossVersion(t *testing.T) {
if seed == 0 {
seed = time.Now().UnixNano()
}
tempDir := t.TempDir()
t.Logf("Test directory: %s\n", tempDir)
t.Logf("Reproduction:\n %s\n", reproductionCommand())
// Print all the versions supplied and ensure all the test binaries
// actually exist before proceeding.
for i, v := range versions {
if len(v.SHA) > 8 {
// Use shortened SHAs for readability.
versions[i].SHA = versions[i].SHA[:8]
}
absPath, err := filepath.Abs(v.TestBinaryPath)
if err != nil {
t.Fatal(err)
}
fi, err := os.Stat(absPath)
if err != nil {
t.Fatal(err)
}
versions[i].TestBinaryPath = absPath
t.Logf("%d: %s (Mode = %s)", i, v.String(), fi.Mode())
}
// All randomness should be derived from `seed`. This makes reproducing a
// failure locally easier.
ctx := context.Background()
require.NoError(t, runCrossVersion(ctx, t, tempDir, versions, seed, factor))
}
type pebbleVersion struct {
Label string
SHA string
TestBinaryPath string
}
type initialState struct {
desc string
path string
}
func (s initialState) String() string {
if s.desc == "" {
return "<empty>"
}
return s.desc
}
func runCrossVersion(
ctx context.Context,
t *testing.T,
tempDir string,
versions pebbleVersions,
seed int64,
factor int,
) error {
prng := rand.New(rand.NewSource(seed))
// Use prng to derive deterministic seeds to provide to the child
// metamorphic runs. The same seed is used for all runs on a particular
// Pebble version.
versionSeeds := make([]uint64, len(versions))
for i := range versions {
versionSeeds[i] = prng.Uint64()
}
rootDir := filepath.Join(tempDir, strconv.FormatInt(seed, 10))
if err := os.MkdirAll(rootDir, os.ModePerm); err != nil {
return err
}
// When run with test parallelism, multiple tests may fail concurrently.
// Only one should actually run the test failure logic which copies the root
// dir into the artifacts directory.
var fatalOnce sync.Once
// The outer for loop executes once per version being tested. It takes a
// list of initial states, populated by the previous version. The inner loop
// executes once per initial state, running the metamorphic test against the
// initial state.
//
// The number of states that are carried forward from one version to the
// next is fixed by `factor`.
initialStates := []initialState{{}}
for i := range versions {
t.Logf("Running tests with version %s with %d initial state(s).", versions[i].SHA, len(initialStates))
histories, nextInitialStates, err := runVersion(ctx, t, &fatalOnce, rootDir, versions[i], versionSeeds[i], initialStates)
if err != nil {
return err
}
// All the initial states described the same state and all of this
// version's metamorphic runs used the same seed, so all of the
// resulting histories should be identical.
if h, diff := metamorphic.CompareHistories(t, histories); h > 0 {
fatalf(t, &fatalOnce, rootDir, "Metamorphic test divergence between %q and %q:\nDiff:\n%s",
nextInitialStates[0].desc, nextInitialStates[h].desc, diff)
}
// Prune the set of initial states we collected for this version, using
// the deterministic randomness of prng to pick which states we keep.
if len(nextInitialStates) > factor {
prng.Shuffle(len(nextInitialStates), func(i, j int) {
nextInitialStates[i], nextInitialStates[j] = nextInitialStates[j], nextInitialStates[i]
})
// Delete the states that we're not going to use.
for _, s := range nextInitialStates[factor:] {
require.NoError(t, os.RemoveAll(s.path))
}
nextInitialStates = nextInitialStates[:factor]
}
initialStates = nextInitialStates
}
return nil
}
func runVersion(
ctx context.Context,
t *testing.T,
fatalOnce *sync.Once,
rootDir string,
vers pebbleVersion,
seed uint64,
initialStates []initialState,
) (histories []string, nextInitialStates []initialState, err error) {
// mu guards histories and nextInitialStates. The subtests may be run in
// parallel (via t.Parallel()).
var mu sync.Mutex
// The outer 'execution-<label>' subtest will block until all of the
// individual subtests have completed.
t.Run(fmt.Sprintf("execution-%s", vers.Label), func(t *testing.T) {
for j, s := range initialStates {
j, s := j, s // re-bind loop vars to scope
runID := fmt.Sprintf("%s_%s_%d_%03d", vers.Label, vers.SHA, seed, j)
r := metamorphicTestRun{
seed: seed,
dir: filepath.Join(rootDir, runID),
vers: vers,
initialState: s,
testBinaryPath: vers.TestBinaryPath,
}
t.Run(s.desc, func(t *testing.T) {
t.Parallel()
require.NoError(t, os.MkdirAll(r.dir, os.ModePerm))
var buf bytes.Buffer
var out io.Writer = &buf
if streamOutput {
out = io.MultiWriter(out, os.Stderr)
}
t.Logf(" Running test with version %s with initial state %s.",
vers.SHA, s)
if err := r.run(ctx, out); err != nil {
fatalf(t, fatalOnce, rootDir, "Metamorphic test failed: %s\nOutput:%s\n", err, buf.String())
}
// dir is a directory containing the ops file and subdirectories for
// each run with a particular set of OPTIONS. For example:
//
// dir/
// ops
// random-000/
// random-001/
// ...
// standard-000/
// standard-001/
// ...
dir := getRunDir(t, r.dir)
// subrunDirs contains the names of all dir's subdirectories.
subrunDirs := getDirs(t, dir)
mu.Lock()
defer mu.Unlock()
for _, subrunDir := range subrunDirs {
// Record the subrun as an initial state for the next version.
nextInitialStates = append(nextInitialStates, initialState{
path: filepath.Join(dir, subrunDir),
desc: fmt.Sprintf("sha=%s-seed=%d-opts=%s(%s)", vers.SHA, seed, subrunDir, s.String()),
})
histories = append(histories, filepath.Join(dir, subrunDir, "history"))
}
})
}
})
return histories, nextInitialStates, err
}
func fatalf(t testing.TB, fatalOnce *sync.Once, dir string, msg string, args ...interface{}) {
fatalOnce.Do(func() {
if artifactsDir == "" {
var err error
artifactsDir, err = os.Getwd()
require.NoError(t, err)
}
// When run with test parallelism, other subtests may still be running
// within subdirectories of `dir`. We copy instead of rename so that those
// substests don't also fail when we remove their files out from under them.
// Those additional failures would confuse the test output.
dst := filepath.Join(artifactsDir, filepath.Base(dir))
t.Logf("Copying test dir %q to %q.", dir, dst)
_, err := vfs.Clone(vfs.Default, vfs.Default, dir, dst, vfs.CloneTryLink)
if err != nil {
t.Error(err)
}
t.Fatalf(msg, args...)
})
}
type metamorphicTestRun struct {
seed uint64
dir string
vers pebbleVersion
initialState initialState
testBinaryPath string
}
func (r *metamorphicTestRun) run(ctx context.Context, output io.Writer) error {
args := []string{
"-test.run", "TestMeta$",
"-seed", strconv.FormatUint(r.seed, 10),
"-keep",
// Use an op-count distribution that includes a low lower bound, so that
// some intermediary versions do very little work besides opening the
// database. This helps exercise state from version n that survives to
// versions ≥ n+2.
"-ops", "uniform:1-10000",
// Explicitly specify the location of the _meta directory. In Cockroach
// CI when built using bazel, the subprocesses may be given a different
// current working directory than the one provided below. To ensure we
// can find this run's artifacts, explicitly pass the intended dir.
"-dir", filepath.Join(r.dir, "_meta"),
}
// Propagate the verbose flag, if necessary.
if testing.Verbose() {
args = append(args, "-test.v")
}
if r.initialState.path != "" {
args = append(args,
"--initial-state", r.initialState.path,
"--initial-state-desc", r.initialState.desc)
}
cmd := exec.CommandContext(ctx, r.testBinaryPath, args...)
cmd.Dir = r.dir
cmd.Stderr = output
cmd.Stdout = output
// Print the command itself before executing it.
if testing.Verbose() {
fmt.Fprintln(output, cmd)
}
return cmd.Run()
}
func (v pebbleVersion) String() string {
return fmt.Sprintf("%s,%s,%s", v.Label, v.SHA, v.TestBinaryPath)
}
// pebbleVersions implements flag.Value for the -version flag.
type pebbleVersions []pebbleVersion
var _ flag.Value = (*pebbleVersions)(nil)
// String returns the SHAs of the versions.
func (f *pebbleVersions) String() string {
var buf bytes.Buffer
for i, v := range *f {
if i > 0 {
fmt.Fprint(&buf, " ")
}
fmt.Fprintf(&buf, v.SHA)
}
return buf.String()
}
// Set is part of the flag.Value interface; it is called once for every
// occurrence of the version flag.
func (f *pebbleVersions) Set(value string) error {
// Expected format is `<label>,<sha>,<path>`.
fields := strings.FieldsFunc(value, func(r rune) bool { return r == ',' || unicode.IsSpace(r) })
if len(fields) != 3 {
return errors.Newf("unable to parse version %q", value)
}
*f = append(*f, pebbleVersion{
Label: fields[0],
SHA: fields[1],
TestBinaryPath: fields[2],
})
return nil
}
func getDirs(t testing.TB, dir string) (names []string) {
dirents, err := os.ReadDir(dir)
if err != nil {
t.Fatal(err)
}
for _, dirent := range dirents {
if dirent.IsDir() {
names = append(names, dirent.Name())
}
}
return names
}
func getRunDir(t testing.TB, dir string) string {
metaDir := filepath.Join(dir, "_meta")
dirs := getDirs(t, metaDir)
if len(dirs) != 1 {
t.Fatalf("expected 1 directory, found %d", len(dirs))
}
return filepath.Join(metaDir, dirs[0])
}