mirror of
https://source.quilibrium.com/quilibrium/ceremonyclient.git
synced 2024-11-10 18:25:17 +00:00
50 lines
1.2 KiB
Go
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)
|
||
|
|
})
|
||
|
|
}
|