btcd/integration/prune_test.go

41 lines
1.0 KiB
Go

// Copyright (c) 2023 The btcsuite developers
// Use of this source code is governed by an ISC
// license that can be found in the LICENSE file.
// This file is ignored during the regular tests due to the following build tag.
//go:build rpctest
// +build rpctest
package integration
import (
"testing"
"github.com/btcsuite/btcd/chaincfg"
"github.com/btcsuite/btcd/integration/rpctest"
"github.com/stretchr/testify/require"
)
func TestPrune(t *testing.T) {
t.Parallel()
// Boilerplate code to make a pruned node.
btcdCfg := []string{"--prune=1536"}
r, err := rpctest.New(&chaincfg.SimNetParams, nil, btcdCfg, "")
require.NoError(t, err)
if err := r.SetUp(false, 0); err != nil {
require.NoError(t, err)
}
t.Cleanup(func() { r.TearDown() })
// Check that the rpc call for block chain info comes back correctly.
chainInfo, err := r.Client.GetBlockChainInfo()
require.NoError(t, err)
if !chainInfo.Pruned {
t.Fatalf("expected the node to be pruned but the pruned "+
"boolean was %v", chainInfo.Pruned)
}
}