93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
|
package agent
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/nomad/helper/uuid"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestHTTP_rpcHandlerForAlloc(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
require := require.New(t)
|
||
|
|
agent := NewTestAgent(t, t.Name(), nil)
|
||
|
|
|
||
|
|
a := mockFSAlloc(agent.client.NodeID(), nil)
|
||
|
|
addAllocToClient(agent, a, terminalClientAlloc)
|
||
|
|
|
||
|
|
// Case 1: Client has allocation
|
||
|
|
// Outcome: Use local client
|
||
|
|
lc, rc, s := agent.Server.rpcHandlerForAlloc(a.ID)
|
||
|
|
require.True(lc)
|
||
|
|
require.False(rc)
|
||
|
|
require.False(s)
|
||
|
|
|
||
|
|
// Case 2: Client doesn't have allocation and there is a server
|
||
|
|
// Outcome: Use server
|
||
|
|
lc, rc, s = agent.Server.rpcHandlerForAlloc(uuid.Generate())
|
||
|
|
require.False(lc)
|
||
|
|
require.False(rc)
|
||
|
|
require.True(s)
|
||
|
|
|
||
|
|
// Case 3: Client doesn't have allocation and there is no server
|
||
|
|
// Outcome: Use client RPC to server
|
||
|
|
srv := agent.server
|
||
|
|
agent.server = nil
|
||
|
|
lc, rc, s = agent.Server.rpcHandlerForAlloc(uuid.Generate())
|
||
|
|
require.False(lc)
|
||
|
|
require.True(rc)
|
||
|
|
require.False(s)
|
||
|
|
agent.server = srv
|
||
|
|
|
||
|
|
// Case 4: No client
|
||
|
|
// Outcome: Use server
|
||
|
|
client := agent.client
|
||
|
|
agent.client = nil
|
||
|
|
lc, rc, s = agent.Server.rpcHandlerForAlloc(uuid.Generate())
|
||
|
|
require.False(lc)
|
||
|
|
require.False(rc)
|
||
|
|
require.True(s)
|
||
|
|
agent.client = client
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHTTP_rpcHandlerForNode(t *testing.T) {
|
||
|
|
t.Parallel()
|
||
|
|
require := require.New(t)
|
||
|
|
agent := NewTestAgent(t, t.Name(), nil)
|
||
|
|
cID := agent.client.NodeID()
|
||
|
|
|
||
|
|
// Case 1: Node running, no node ID given
|
||
|
|
// Outcome: Use local node
|
||
|
|
lc, rc, s := agent.Server.rpcHandlerForNode("")
|
||
|
|
require.True(lc)
|
||
|
|
require.False(rc)
|
||
|
|
require.False(s)
|
||
|
|
|
||
|
|
// Case 2: Node running, it's ID given
|
||
|
|
// Outcome: Use local node
|
||
|
|
lc, rc, s = agent.Server.rpcHandlerForNode(cID)
|
||
|
|
require.True(lc)
|
||
|
|
require.False(rc)
|
||
|
|
require.False(s)
|
||
|
|
|
||
|
|
// Case 3: Local node but wrong ID and there is no server
|
||
|
|
// Outcome: Use client RPC to server
|
||
|
|
srv := agent.server
|
||
|
|
agent.server = nil
|
||
|
|
lc, rc, s = agent.Server.rpcHandlerForNode(uuid.Generate())
|
||
|
|
require.False(lc)
|
||
|
|
require.True(rc)
|
||
|
|
require.False(s)
|
||
|
|
agent.server = srv
|
||
|
|
|
||
|
|
// Case 4: No client
|
||
|
|
// Outcome: Use server
|
||
|
|
client := agent.client
|
||
|
|
agent.client = nil
|
||
|
|
lc, rc, s = agent.Server.rpcHandlerForNode(uuid.Generate())
|
||
|
|
require.False(lc)
|
||
|
|
require.False(rc)
|
||
|
|
require.True(s)
|
||
|
|
agent.client = client
|
||
|
|
}
|