df96b89958
are removed from the event buffer. Wire up event buffer size config, use pointers for structs.Events instead of copying.
125 lines
3.3 KiB
Go
125 lines
3.3 KiB
Go
package state
|
|
|
|
import (
|
|
testing "github.com/mitchellh/go-testing-interface"
|
|
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
|
"github.com/hashicorp/nomad/helper/uuid"
|
|
"github.com/hashicorp/nomad/nomad/mock"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
)
|
|
|
|
func TestStateStore(t testing.T) *StateStore {
|
|
config := &StateStoreConfig{
|
|
Logger: testlog.HCLogger(t),
|
|
Region: "global",
|
|
}
|
|
state, err := NewStateStore(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
if state == nil {
|
|
t.Fatalf("missing state")
|
|
}
|
|
return state
|
|
}
|
|
|
|
func TestStateStorePublisher(t testing.T) *StateStoreConfig {
|
|
return &StateStoreConfig{
|
|
Logger: testlog.HCLogger(t),
|
|
Region: "global",
|
|
EnablePublisher: true,
|
|
}
|
|
}
|
|
func TestStateStoreCfg(t testing.T, cfg *StateStoreConfig) *StateStore {
|
|
state, err := NewStateStore(cfg)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if state == nil {
|
|
t.Fatalf("missing state")
|
|
}
|
|
return state
|
|
}
|
|
|
|
// CreateTestCSIPlugin is a helper that generates the node + fingerprint results necessary
|
|
// to create a CSIPlugin by directly inserting into the state store. The plugin requires a
|
|
// controller.
|
|
func CreateTestCSIPlugin(s *StateStore, id string) func() {
|
|
return createTestCSIPlugin(s, id, true)
|
|
}
|
|
|
|
// CreateTestCSIPluginNodeOnly is a helper that generates the node + fingerprint results
|
|
// necessary to create a CSIPlugin by directly inserting into the state store. The plugin
|
|
// does not require a controller. In tests that exercise volume registration, this prevents
|
|
// an error attempting to RPC the node.
|
|
func CreateTestCSIPluginNodeOnly(s *StateStore, id string) func() {
|
|
return createTestCSIPlugin(s, id, false)
|
|
}
|
|
|
|
func createTestCSIPlugin(s *StateStore, id string, requiresController bool) func() {
|
|
// Create some nodes
|
|
ns := make([]*structs.Node, 3)
|
|
for i := range ns {
|
|
n := mock.Node()
|
|
n.Attributes["nomad.version"] = "0.11.0"
|
|
ns[i] = n
|
|
}
|
|
|
|
// Install healthy plugin fingerprinting results
|
|
ns[0].CSIControllerPlugins = map[string]*structs.CSIInfo{
|
|
id: {
|
|
PluginID: id,
|
|
AllocID: uuid.Generate(),
|
|
Healthy: true,
|
|
HealthDescription: "healthy",
|
|
RequiresControllerPlugin: requiresController,
|
|
RequiresTopologies: false,
|
|
ControllerInfo: &structs.CSIControllerInfo{
|
|
SupportsReadOnlyAttach: true,
|
|
SupportsAttachDetach: true,
|
|
SupportsListVolumes: true,
|
|
SupportsListVolumesAttachedNodes: false,
|
|
},
|
|
},
|
|
}
|
|
|
|
// Install healthy plugin fingerprinting results
|
|
for _, n := range ns[1:] {
|
|
n.CSINodePlugins = map[string]*structs.CSIInfo{
|
|
id: {
|
|
PluginID: id,
|
|
AllocID: uuid.Generate(),
|
|
Healthy: true,
|
|
HealthDescription: "healthy",
|
|
RequiresControllerPlugin: requiresController,
|
|
RequiresTopologies: false,
|
|
NodeInfo: &structs.CSINodeInfo{
|
|
ID: n.ID,
|
|
MaxVolumes: 64,
|
|
RequiresNodeStageVolume: true,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// Insert them into the state store
|
|
index := uint64(999)
|
|
for _, n := range ns {
|
|
index++
|
|
s.UpsertNode(index, n)
|
|
}
|
|
|
|
ids := make([]string, len(ns))
|
|
for i, n := range ns {
|
|
ids[i] = n.ID
|
|
}
|
|
|
|
// Return cleanup function that deletes the nodes
|
|
return func() {
|
|
index++
|
|
s.DeleteNode(index, ids)
|
|
}
|
|
}
|