51 lines
1,002 B
Go
51 lines
1,002 B
Go
|
package nomad
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/nomad/testutil"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRPC_forwardLeader(t *testing.T) {
|
||
|
|
s1 := testServer(t, nil)
|
||
|
|
defer s1.Shutdown()
|
||
|
|
s2 := testServer(t, func(c *Config) {
|
||
|
|
c.DevDisableBootstrap = true
|
||
|
|
})
|
||
|
|
defer s2.Shutdown()
|
||
|
|
testJoin(t, s1, s2)
|
||
|
|
testutil.WaitForLeader(t, s1.RPC)
|
||
|
|
|
||
|
|
var out struct{}
|
||
|
|
err := s1.forwardLeader("Status.Ping", struct{}{}, &out)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = s2.forwardLeader("Status.Ping", struct{}{}, &out)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRPC_forwardRegion(t *testing.T) {
|
||
|
|
s1 := testServer(t, nil)
|
||
|
|
defer s1.Shutdown()
|
||
|
|
s2 := testServer(t, func(c *Config) {
|
||
|
|
c.Region = "region2"
|
||
|
|
})
|
||
|
|
defer s2.Shutdown()
|
||
|
|
testJoin(t, s1, s2)
|
||
|
|
|
||
|
|
var out struct{}
|
||
|
|
err := s1.forwardRegion("region2", "Status.Ping", struct{}{}, &out)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = s2.forwardRegion("region1", "Status.Ping", struct{}{}, &out)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
}
|