T3MP3ST public launch build
This commit is contained in:
t3mp3st contributors committed
1 parent
972247b881
commit
184c7d4e59
1589 files changed
+57150
-506
No files matched your search
@@ -0,0 +1,315 @@
|
||||
// Copyright IBM Corp. 2015, 2025
|
||||
// SPDX-License-Identifier: BUSL-1.1
|
||||
|
||||
package nomad
|
||||
|
||||
import (
|
||||
"net"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/hashicorp/nomad/ci"
|
||||
"github.com/hashicorp/nomad/client"
|
||||
"github.com/hashicorp/nomad/client/config"
|
||||
"github.com/hashicorp/nomad/helper/uuid"
|
||||
"github.com/hashicorp/nomad/nomad/structs"
|
||||
"github.com/hashicorp/nomad/testutil"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
type namedConnWrapper struct {
|
||||
net.Conn
|
||||
name string
|
||||
}
|
||||
|
||||
type namedAddr string
|
||||
|
||||
func (n namedAddr) String() string { return string(n) }
|
||||
func (n namedAddr) Network() string { return string(n) }
|
||||
|
||||
func (n namedConnWrapper) LocalAddr() net.Addr {
|
||||
return namedAddr(n.name)
|
||||
}
|
||||
|
||||
func TestServer_removeNodeConn_differentAddrs(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, nil)
|
||||
defer cleanupS1()
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
|
||||
p1, p2 := net.Pipe()
|
||||
w1 := namedConnWrapper{
|
||||
Conn: p1,
|
||||
name: "a",
|
||||
}
|
||||
w2 := namedConnWrapper{
|
||||
Conn: p2,
|
||||
name: "b",
|
||||
}
|
||||
|
||||
// Add the connections
|
||||
nodeID := uuid.Generate()
|
||||
ctx1 := &RPCContext{
|
||||
Conn: w1,
|
||||
NodeID: nodeID,
|
||||
}
|
||||
ctx2 := &RPCContext{
|
||||
Conn: w2,
|
||||
NodeID: nodeID,
|
||||
}
|
||||
|
||||
s1.addNodeConn(ctx1)
|
||||
s1.addNodeConn(ctx2)
|
||||
require.Len(s1.connectedNodes(), 1)
|
||||
require.Len(s1.nodeConns[nodeID], 2)
|
||||
|
||||
// Check that the value is the second conn.
|
||||
state, ok := s1.getNodeConn(nodeID)
|
||||
require.True(ok)
|
||||
require.Equal(state.Ctx.Conn.LocalAddr().String(), w2.name)
|
||||
|
||||
// Delete the first
|
||||
s1.removeNodeConn(ctx1)
|
||||
require.Len(s1.connectedNodes(), 1)
|
||||
require.Len(s1.nodeConns[nodeID], 1)
|
||||
|
||||
// Check that the value is the second conn.
|
||||
state, ok = s1.getNodeConn(nodeID)
|
||||
require.True(ok)
|
||||
require.Equal(state.Ctx.Conn.LocalAddr().String(), w2.name)
|
||||
|
||||
// Delete the second
|
||||
s1.removeNodeConn(ctx2)
|
||||
require.Len(s1.connectedNodes(), 0)
|
||||
|
||||
_, ok = s1.getNodeConn(nodeID)
|
||||
require.False(ok)
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_NoPath(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 2
|
||||
})
|
||||
defer cleanupS1()
|
||||
s2, cleanupS2 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 2
|
||||
})
|
||||
defer cleanupS2()
|
||||
TestJoin(t, s1, s2)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
testutil.WaitForLeader(t, s2.RPC)
|
||||
|
||||
nodeID := uuid.Generate()
|
||||
srv, err := s1.serverWithNodeConn(nodeID, s1.Region())
|
||||
require.Nil(srv)
|
||||
require.EqualError(err, structs.ErrNoNodeConn.Error())
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_NoPath_Region(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, nil)
|
||||
defer cleanupS1()
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
|
||||
nodeID := uuid.Generate()
|
||||
srv, err := s1.serverWithNodeConn(nodeID, "fake-region")
|
||||
require.Nil(srv)
|
||||
require.EqualError(err, structs.ErrNoRegionPath.Error())
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_Path(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 2
|
||||
})
|
||||
defer cleanupS1()
|
||||
s2, cleanupS2 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 2
|
||||
})
|
||||
defer cleanupS2()
|
||||
TestJoin(t, s1, s2)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
testutil.WaitForLeader(t, s2.RPC)
|
||||
|
||||
// Create a fake connection for the node on server 2
|
||||
nodeID := uuid.Generate()
|
||||
s2.addNodeConn(&RPCContext{
|
||||
NodeID: nodeID,
|
||||
})
|
||||
|
||||
srv, err := s1.serverWithNodeConn(nodeID, s1.Region())
|
||||
require.NotNil(srv)
|
||||
require.Equal(srv.Addr.String(), s2.config.RPCAddr.String())
|
||||
require.Nil(err)
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_Path_Region(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, nil)
|
||||
defer cleanupS1()
|
||||
s2, cleanupS2 := TestServer(t, func(c *Config) {
|
||||
c.Region = "two"
|
||||
})
|
||||
defer cleanupS2()
|
||||
TestJoin(t, s1, s2)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
testutil.WaitForLeader(t, s2.RPC)
|
||||
|
||||
// Create a fake connection for the node on server 2
|
||||
nodeID := uuid.Generate()
|
||||
s2.addNodeConn(&RPCContext{
|
||||
NodeID: nodeID,
|
||||
})
|
||||
|
||||
srv, err := s1.serverWithNodeConn(nodeID, s2.Region())
|
||||
require.NotNil(srv)
|
||||
require.Equal(srv.Addr.String(), s2.config.RPCAddr.String())
|
||||
require.Nil(err)
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_Path_Newest(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS1()
|
||||
s2, cleanupS2 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS2()
|
||||
s3, cleanupS3 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS3()
|
||||
TestJoin(t, s1, s2, s3)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
testutil.WaitForLeader(t, s2.RPC)
|
||||
testutil.WaitForLeader(t, s3.RPC)
|
||||
|
||||
// Create a fake connection for the node on server 2 and 3
|
||||
nodeID := uuid.Generate()
|
||||
s2.addNodeConn(&RPCContext{
|
||||
NodeID: nodeID,
|
||||
})
|
||||
s3.addNodeConn(&RPCContext{
|
||||
NodeID: nodeID,
|
||||
})
|
||||
|
||||
srv, err := s1.serverWithNodeConn(nodeID, s1.Region())
|
||||
require.NotNil(srv)
|
||||
require.Equal(srv.Addr.String(), s3.config.RPCAddr.String())
|
||||
require.Nil(err)
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_PathAndErr(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS1()
|
||||
s2, cleanupS2 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS2()
|
||||
s3, cleanupS3 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS3()
|
||||
TestJoin(t, s1, s2, s3)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
testutil.WaitForLeader(t, s2.RPC)
|
||||
testutil.WaitForLeader(t, s3.RPC)
|
||||
|
||||
// Create a fake connection for the node on server 2
|
||||
nodeID := uuid.Generate()
|
||||
s2.addNodeConn(&RPCContext{
|
||||
NodeID: nodeID,
|
||||
})
|
||||
|
||||
// Shutdown the RPC layer for server 3
|
||||
s3.rpcListener.Close()
|
||||
|
||||
srv, err := s1.serverWithNodeConn(nodeID, s1.Region())
|
||||
require.NotNil(srv)
|
||||
require.Equal(srv.Addr.String(), s2.config.RPCAddr.String())
|
||||
require.Nil(err)
|
||||
}
|
||||
|
||||
func TestServerWithNodeConn_NoPathAndErr(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
|
||||
s1, cleanupS1 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS1()
|
||||
s2, cleanupS2 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS2()
|
||||
s3, cleanupS3 := TestServer(t, func(c *Config) {
|
||||
c.BootstrapExpect = 3
|
||||
})
|
||||
defer cleanupS3()
|
||||
TestJoin(t, s1, s2, s3)
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
testutil.WaitForLeader(t, s2.RPC)
|
||||
testutil.WaitForLeader(t, s3.RPC)
|
||||
|
||||
// Shutdown the RPC layer for server 3
|
||||
s3.rpcListener.Close()
|
||||
|
||||
srv, err := s1.serverWithNodeConn(uuid.Generate(), s1.Region())
|
||||
require.Nil(srv)
|
||||
require.NotNil(err)
|
||||
|
||||
// the exact error seems to be dependent on timing and raft protocol version
|
||||
if !strings.Contains(err.Error(), "failed querying") && !strings.Contains(err.Error(), "No path to node") {
|
||||
require.Contains(err.Error(), "failed querying")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNodeStreamingRpc_badEndpoint(t *testing.T) {
|
||||
ci.Parallel(t)
|
||||
require := require.New(t)
|
||||
s1, cleanupS1 := TestServer(t, nil)
|
||||
defer cleanupS1()
|
||||
testutil.WaitForLeader(t, s1.RPC)
|
||||
|
||||
c, cleanupC := client.TestClient(t, func(c *config.Config) {
|
||||
c.Servers = []string{s1.config.RPCAddr.String()}
|
||||
})
|
||||
defer cleanupC()
|
||||
|
||||
// Wait for the client to connect
|
||||
testutil.WaitForResult(func() (bool, error) {
|
||||
nodes := s1.connectedNodes()
|
||||
return len(nodes) == 1, nil
|
||||
}, func(err error) {
|
||||
t.Fatalf("should have a clients")
|
||||
})
|
||||
|
||||
state, ok := s1.getNodeConn(c.NodeID())
|
||||
require.True(ok)
|
||||
|
||||
conn, err := NodeStreamingRpc(state.Session, "Bogus")
|
||||
require.Nil(conn)
|
||||
require.NotNil(err)
|
||||
require.Contains(err.Error(), "Bogus")
|
||||
require.True(structs.IsErrUnknownMethod(err))
|
||||
}
|
||||
Reference in new issue
Block a user