1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
package client_test
import (
"context"
"fmt"
"testing"
"time"
dqlite "github.com/canonical/go-dqlite/v3"
"github.com/canonical/go-dqlite/v3/client"
"github.com/stretchr/testify/require"
)
func TestMembership(t *testing.T) {
infos, cleanup := setup(t)
defer cleanup()
store := client.NewInmemNodeStore()
store.Set(context.Background(), infos[:1])
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
client, err := client.FindLeader(ctx, store)
require.NoError(t, err)
defer client.Close()
err = client.Add(ctx, infos[1])
require.NoError(t, err)
}
func setup(t *testing.T) ([]client.NodeInfo, func()) {
n := 3
nodes := make([]*dqlite.Node, n)
infos := make([]client.NodeInfo, n)
var cleanups []func()
for i := range nodes {
id := uint64(i + 1)
address := fmt.Sprintf("@test-%d", id)
dir, cleanup := newDir(t)
cleanups = append(cleanups, cleanup)
node, err := dqlite.New(id, address, dir, dqlite.WithBindAddress(address))
require.NoError(t, err)
nodes[i] = node
infos[i].ID = id
infos[i].Address = address
err = node.Start()
require.NoError(t, err)
cleanups = append(cleanups, func() { node.Close() })
}
return infos, func() {
for i := len(cleanups) - 1; i >= 0; i-- {
cleanups[i]()
}
}
}
|
