You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
package store
|
|
|
|
import (
|
|
"os"
|
|
"sort"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func Test_NumPeersEnableSingle(t *testing.T) {
|
|
s0 := mustNewStore(true)
|
|
defer os.RemoveAll(s0.Path())
|
|
if err := s0.Open(true); err != nil {
|
|
t.Fatalf("failed to open node for num peers test: %s", err.Error())
|
|
}
|
|
s0.WaitForLeader(5 * time.Second)
|
|
s0.Close(true)
|
|
|
|
j, err := JoinAllowed(s0.Path())
|
|
if err != nil {
|
|
t.Fatalf("failed to check join status of %s: %s", s0.Path(), err.Error())
|
|
}
|
|
if !j {
|
|
t.Fatalf("config files at %s indicate joining is not allowed", s0.Path())
|
|
}
|
|
}
|
|
|
|
func Test_NumPeersDisableSingle(t *testing.T) {
|
|
s0 := mustNewStore(true)
|
|
defer os.RemoveAll(s0.Path())
|
|
if err := s0.Open(false); err != nil {
|
|
t.Fatalf("failed to open node for num peers test: %s", err.Error())
|
|
}
|
|
s0.Close(true)
|
|
|
|
j, err := JoinAllowed(s0.Path())
|
|
if err != nil {
|
|
t.Fatalf("failed to check join status of %s: %s", s0.Path(), err.Error())
|
|
}
|
|
if !j {
|
|
t.Fatalf("config files at %s indicate joining is not allowed", s0.Path())
|
|
}
|
|
}
|
|
|
|
func Test_NumPeersJoin(t *testing.T) {
|
|
s0 := mustNewStore(true)
|
|
defer os.RemoveAll(s0.Path())
|
|
if err := s0.Open(true); err != nil {
|
|
t.Fatalf("failed to open node for num peers test: %s", err.Error())
|
|
}
|
|
s0.WaitForLeader(5 * time.Second)
|
|
|
|
s1 := mustNewStore(true)
|
|
defer os.RemoveAll(s1.Path())
|
|
if err := s1.Open(false); err != nil {
|
|
t.Fatalf("failed to open node for num peers test: %s", err.Error())
|
|
}
|
|
|
|
// Get sorted list of cluster nodes.
|
|
storeNodes := []string{s0.Addr().String(), s1.Addr().String()}
|
|
sort.StringSlice(storeNodes).Sort()
|
|
|
|
// Join the second node to the first.
|
|
if err := s0.Join(s1.Addr().String()); err != nil {
|
|
t.Fatalf("failed to join to node at %s: %s", s0.Addr().String(), err.Error())
|
|
}
|
|
s1.WaitForLeader(5 * time.Second)
|
|
s1.Close(true)
|
|
s0.Close(true)
|
|
|
|
// Check that peers are set as expected.
|
|
m, _ := NumPeers(s0.Path())
|
|
if m != 2 {
|
|
t.Fatalf("got wrong value for number of peers, exp %d, got %d", 2, m)
|
|
}
|
|
|
|
j, err := JoinAllowed(s0.Path())
|
|
if err != nil {
|
|
t.Fatalf("failed to check join status of %s: %s", s0.Path(), err.Error())
|
|
}
|
|
if j {
|
|
t.Fatalf("config files at %s indicate joining is allowed", s0.Path())
|
|
}
|
|
|
|
k, err := JoinAllowed(s1.Path())
|
|
if err != nil {
|
|
t.Fatalf("failed to check join status of %s: %s", s1.Path(), err.Error())
|
|
}
|
|
if k {
|
|
t.Fatalf("config files at %s indicate joining is allowed", s1.Path())
|
|
}
|
|
}
|