ceremonyclient/pebble/record/rotation_test.go

50 lines
1.2 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.
package record
import (
"fmt"
"strconv"
"testing"
"github.com/cockroachdb/datadriven"
)
func TestRotation(t *testing.T) {
var rh RotationHelper
datadriven.RunTest(t, "testdata/rotation", func(t *testing.T, td *datadriven.TestData) string {
oneIntArg := func() int64 {
if len(td.CmdArgs) != 1 {
td.Fatalf(t, "expected one integer argument")
}
n, err := strconv.Atoi(td.CmdArgs[0].String())
if err != nil {
td.Fatalf(t, "expected one integer argument")
}
return int64(n)
}
switch td.Cmd {
case "add":
size := oneIntArg()
rh.AddRecord(size)
case "should-rotate":
nextSnapshotSize := oneIntArg()
return fmt.Sprint(rh.ShouldRotate(nextSnapshotSize))
case "rotate":
snapshotSize := oneIntArg()
rh.Rotate(snapshotSize)
default:
td.Fatalf(t, "unknown command %s", td.Cmd)
}
// For commands with no output, show the debug info.
a, b := rh.DebugInfo()
return fmt.Sprintf("last-snapshot-size: %d\nsize-since-last-snapshot: %d", a, b)
})
}