05c7373a28
set -euo pipefail
unset CDPATH
cd "$(dirname "$0")"
for f in $(git grep '\brequire := require\.New(' | cut -d':' -f1 | sort -u); do
echo "=== require: $f ==="
sed -i '/require := require.New(t)/d' $f
# require.XXX(blah) but not require.XXX(tblah) or require.XXX(rblah)
sed -i 's/\brequire\.\([a-zA-Z0-9_]*\)(\([^tr]\)/require.\1(t,\2/g' $f
# require.XXX(tblah) but not require.XXX(t, blah)
sed -i 's/\brequire\.\([a-zA-Z0-9_]*\)(\(t[^,]\)/require.\1(t,\2/g' $f
# require.XXX(rblah) but not require.XXX(r, blah)
sed -i 's/\brequire\.\([a-zA-Z0-9_]*\)(\(r[^,]\)/require.\1(t,\2/g' $f
gofmt -s -w $f
done
for f in $(git grep '\bassert := assert\.New(' | cut -d':' -f1 | sort -u); do
echo "=== assert: $f ==="
sed -i '/assert := assert.New(t)/d' $f
# assert.XXX(blah) but not assert.XXX(tblah) or assert.XXX(rblah)
sed -i 's/\bassert\.\([a-zA-Z0-9_]*\)(\([^tr]\)/assert.\1(t,\2/g' $f
# assert.XXX(tblah) but not assert.XXX(t, blah)
sed -i 's/\bassert\.\([a-zA-Z0-9_]*\)(\(t[^,]\)/assert.\1(t,\2/g' $f
# assert.XXX(rblah) but not assert.XXX(r, blah)
sed -i 's/\bassert\.\([a-zA-Z0-9_]*\)(\(r[^,]\)/assert.\1(t,\2/g' $f
gofmt -s -w $f
done
118 lines
2.1 KiB
Go
118 lines
2.1 KiB
Go
package api
|
|
|
|
import (
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
)
|
|
|
|
func TestAPI_StatusLeader(t *testing.T) {
|
|
t.Parallel()
|
|
c, s := makeClient(t)
|
|
defer s.Stop()
|
|
s.WaitForSerfCheck(t)
|
|
|
|
status := c.Status()
|
|
|
|
leader, err := status.Leader()
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if leader == "" {
|
|
t.Fatalf("Expected leader, found empty string")
|
|
}
|
|
}
|
|
|
|
func TestAPI_StatusLeaderWithQueryOptions(t *testing.T) {
|
|
t.Parallel()
|
|
c, s := makeClient(t)
|
|
defer s.Stop()
|
|
s.WaitForSerfCheck(t)
|
|
|
|
status := c.Status()
|
|
|
|
opts := QueryOptions{
|
|
Datacenter: "dc1",
|
|
}
|
|
|
|
leader, err := status.LeaderWithQueryOptions(&opts)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if leader == "" {
|
|
t.Fatalf("Expected leader, found empty string")
|
|
}
|
|
}
|
|
|
|
func TestAPI_StatusPeers(t *testing.T) {
|
|
t.Parallel()
|
|
c, s := makeClient(t)
|
|
defer s.Stop()
|
|
s.WaitForSerfCheck(t)
|
|
|
|
status := c.Status()
|
|
|
|
peers, err := status.Peers()
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if len(peers) == 0 {
|
|
t.Fatalf("Expected peers, found %d", len(peers))
|
|
}
|
|
}
|
|
|
|
func TestAPI_StatusPeersWithQueryOptions(t *testing.T) {
|
|
t.Parallel()
|
|
c, s := makeClient(t)
|
|
defer s.Stop()
|
|
s.WaitForSerfCheck(t)
|
|
|
|
status := c.Status()
|
|
|
|
opts := QueryOptions{
|
|
Datacenter: "dc1",
|
|
}
|
|
|
|
peers, err := status.PeersWithQueryOptions(&opts)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if len(peers) == 0 {
|
|
t.Fatalf("Expected peers, found %d", len(peers))
|
|
}
|
|
}
|
|
|
|
func TestAPI_StatusLeader_WrongDC(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
c, s := makeClient(t)
|
|
defer s.Stop()
|
|
s.WaitForSerfCheck(t)
|
|
|
|
status := c.Status()
|
|
|
|
opts := QueryOptions{
|
|
Datacenter: "wrong_dc1",
|
|
}
|
|
|
|
_, err := status.LeaderWithQueryOptions(&opts)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), "No path to datacenter")
|
|
}
|
|
|
|
func TestAPI_StatusPeers_WrongDC(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
c, s := makeClient(t)
|
|
defer s.Stop()
|
|
s.WaitForSerfCheck(t)
|
|
|
|
status := c.Status()
|
|
|
|
opts := QueryOptions{
|
|
Datacenter: "wrong_dc1",
|
|
}
|
|
_, err := status.PeersWithQueryOptions(&opts)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), "No path to datacenter")
|
|
}
|