cleanup prune integration test

This commit is contained in:
Michael Eischer 2023-05-08 18:36:42 +02:00
parent 7a268e4aba
commit 692f81ede8

View file

@ -1,10 +1,8 @@
package main package main
import ( import (
"bytes"
"context" "context"
"encoding/json" "encoding/json"
"os"
"path/filepath" "path/filepath"
"testing" "testing"
@ -77,21 +75,15 @@ func createPrunableRepo(t *testing.T, env *testEnvironment) {
} }
func testRunForgetJSON(t testing.TB, gopts GlobalOptions, args ...string) { func testRunForgetJSON(t testing.TB, gopts GlobalOptions, args ...string) {
buf := bytes.NewBuffer(nil) buf, err := withCaptureStdout(func() error {
oldJSON := gopts.JSON
gopts.stdout = buf
gopts.JSON = true gopts.JSON = true
defer func() {
gopts.stdout = os.Stdout
gopts.JSON = oldJSON
}()
opts := ForgetOptions{ opts := ForgetOptions{
DryRun: true, DryRun: true,
Last: 1, Last: 1,
} }
return runForget(context.TODO(), opts, gopts, args)
rtest.OK(t, runForget(context.TODO(), opts, gopts, args)) })
rtest.OK(t, err)
var forgets []*ForgetGroup var forgets []*ForgetGroup
rtest.OK(t, json.Unmarshal(buf.Bytes(), &forgets)) rtest.OK(t, json.Unmarshal(buf.Bytes(), &forgets))