53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
|
package raftutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
raftboltdb "github.com/hashicorp/raft-boltdb"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
// TestRaftStateInfo_InUse asserts that commands that inspect raft
|
||
|
|
// state such as "nomad operator raft info" and "nomad operator raft
|
||
|
|
// logs" fail with a helpful error message when called on an inuse
|
||
|
|
// database.
|
||
|
|
func TestRaftStateInfo_InUse(t *testing.T) {
|
||
|
|
t.Parallel() // since there's a 1s timeout.
|
||
|
|
|
||
|
|
// First create an empty raft db
|
||
|
|
dir := filepath.Join(t.TempDir(), "raft.db")
|
||
|
|
|
||
|
|
fakedb, err := raftboltdb.NewBoltStore(dir)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Next try to read the db without closing it
|
||
|
|
s, _, _, err := RaftStateInfo(dir)
|
||
|
|
assert.Nil(t, s)
|
||
|
|
require.EqualError(t, err, errAlreadyOpen.Error())
|
||
|
|
|
||
|
|
// LogEntries should produce the same error
|
||
|
|
_, _, err = LogEntries(dir)
|
||
|
|
require.EqualError(t, err, "failed to open raft logs: "+errAlreadyOpen.Error())
|
||
|
|
|
||
|
|
// Commands should work once the db is closed
|
||
|
|
require.NoError(t, fakedb.Close())
|
||
|
|
|
||
|
|
s, _, _, err = RaftStateInfo(dir)
|
||
|
|
assert.NotNil(t, s)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, s.Close())
|
||
|
|
|
||
|
|
logCh, errCh, err := LogEntries(dir)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Consume entries to cleanly close db
|
||
|
|
for closed := false; closed; {
|
||
|
|
select {
|
||
|
|
case _, closed = <-logCh:
|
||
|
|
case <-errCh:
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|