ceremonyclient/pebble/record/rotation_test.go
Cassandra Heart 2e2a1e4789
v1.2.0 (#31)
2024-01-03 01:31:42 -06:00

50 lines
1.2 KiB
Go

// 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)
})
}