func TestState_ClusterExists(t *testing.T) {
t.Parallel()
testPrefix := "TestState_ClusterExists"
testutil.ResetEtcd(t, testPrefix)
logger := testutil.NewTestLogger(testPrefix, t)
state, err := NewStateEtcdWithPrefix(testutil.LocalEtcdConfig, testPrefix, logger)
if err != nil {
t.Fatalf("Could not create state", err)
}
clusterID := structs.ClusterID(uuid.New())
planID := uuid.New()
clusterState := structs.ClusterState{
InstanceID: clusterID,
OrganizationGUID: "OrganizationGUID",
PlanID: planID,
ServiceID: "ServiceID",
SpaceGUID: "SpaceGUID",
}
err = state.SaveCluster(clusterState)
if err != nil {
t.Fatalf("SaveCluster failed %s", err)
}
if !state.ClusterExists(clusterID) {
t.Fatalf("Cluster %s should exist", clusterID)
}
if state.ClusterExists("fakeID") {
t.Fatalf("Cluster %s should not exist", "fakeID")
}
}
func TestState_LoadCluster(t *testing.T) {
t.Parallel()
testPrefix := "TestState_LoadClusterState"
testutil.ResetEtcd(t, testPrefix)
logger := testutil.NewTestLogger(testPrefix, t)
state, err := NewStateEtcdWithPrefix(testutil.LocalEtcdConfig, testPrefix, logger)
if err != nil {
t.Fatalf("Could not create state", err)
}
instanceID := structs.ClusterID(uuid.New())
planID := uuid.New()
clusterState := structs.ClusterState{
InstanceID: instanceID,
OrganizationGUID: "OrganizationGUID",
PlanID: planID,
ServiceID: "ServiceID",
SpaceGUID: "SpaceGUID",
}
err = state.SaveCluster(clusterState)
if err != nil {
t.Fatalf("SaveCluster failed %s", err)
}
loadedState, err := state.LoadCluster(instanceID)
if !reflect.DeepEqual(clusterState, loadedState) {
t.Fatalf("Failed to load ClusterState")
}
}
func TestAddNode_PrioritizeCells_FirstNode(t *testing.T) {
t.Parallel()
testPrefix := "TestAddNode_PrioritizeCells_FirstNode"
testutil.ResetEtcd(t, testPrefix)
logger := testutil.NewTestLogger(testPrefix, t)
clusterLoader := &FakeClusterLoader{
Clusters: []*structs.ClusterState{
&structs.ClusterState{
Nodes: []*structs.Node{
&structs.Node{CellGUID: "cell-n1-z1"},
&structs.Node{CellGUID: "cell-n3-z2"},
},
},
&structs.ClusterState{
Nodes: []*structs.Node{
&structs.Node{CellGUID: "cell-n1-z1"},
&structs.Node{CellGUID: "cell-n3-z2"},
},
},
&structs.ClusterState{
Nodes: []*structs.Node{
&structs.Node{CellGUID: "cell-n1-z1"},
&structs.Node{CellGUID: "cell-n2-z1"},
},
},
},
}
availableCells := cells.NewCells([]*config.Cell{
&config.Cell{GUID: "cell-n1-z1", AvailabilityZone: "z1"},
&config.Cell{GUID: "cell-n2-z1", AvailabilityZone: "z1"},
&config.Cell{GUID: "cell-n3-z2", AvailabilityZone: "z2"},
&config.Cell{GUID: "cell-n4-z2", AvailabilityZone: "z2"},
}, clusterLoader)
currentClusterNodes := []*structs.Node{}
step := AddNode{logger: logger, availableCells: availableCells}
cellsToTry, _ := step.prioritizeCellsToTry(currentClusterNodes)
cellIDs := []string{}
for _, cell := range cellsToTry {
cellIDs = append(cellIDs, cell.GUID)
}
expectedPriority := []string{"cell-n4-z2", "cell-n2-z1", "cell-n3-z2", "cell-n1-z1"}
if !reflect.DeepEqual(cellIDs, expectedPriority) {
t.Fatalf("Expected prioritized cells %v to be %v", cellIDs, expectedPriority)
}
}
func TestState_DeleteCluster(t *testing.T) {
t.Parallel()
testPrefix := "TestState_DeleteClusterState"
etcdApi := testutil.ResetEtcd(t, testPrefix)
logger := testutil.NewTestLogger(testPrefix, t)
state, err := NewStateEtcdWithPrefix(testutil.LocalEtcdConfig, testPrefix, logger)
if err != nil {
t.Fatalf("Could not create state", err)
}
instanceID := structs.ClusterID(uuid.New())
planID := uuid.New()
clusterState := structs.ClusterState{
InstanceID: instanceID,
OrganizationGUID: "OrganizationGUID",
PlanID: planID,
ServiceID: "ServiceID",
SpaceGUID: "SpaceGUID",
}
err = state.SaveCluster(clusterState)
if err != nil {
t.Fatalf("SaveCluster failed %s", err)
}
err = state.DeleteCluster(instanceID)
if err != nil {
t.Fatalf("DeleteClusterState failed %s", err)
}
key := fmt.Sprintf("%s/service/%s/state", testPrefix, instanceID)
_, err = etcdApi.Get(context.Background(), key, &etcd.GetOptions{})
if err == nil {
t.Fatalf("Was expecting error 'Key not found'")
} else {
notFoundRegExp, _ := regexp.Compile("Key not found")
if notFoundRegExp.FindString(err.Error()) != "Key not found" {
t.Fatalf("An error other than 'Key not found' occured %s", err)
}
}
}
func TestState_LoadCluster(t *testing.T) {
t.Parallel()
testPrefix := "TestState_LoadClusterState"
etcdApi := testutil.ResetEtcd(t, testPrefix)
logger := testutil.NewTestLogger(testPrefix, t)
state, err := NewStateEtcdWithPrefix(testutil.LocalEtcdConfig, testPrefix, logger)
if err != nil {
t.Fatalf("Could not create state", err)
}
instanceID := structs.ClusterID(uuid.New())
planID := uuid.New()
node := structs.Node{ID: "node_id", CellGUID: "cell_guid"}
clusterState := structs.ClusterState{
InstanceID: instanceID,
OrganizationGUID: "OrganizationGUID",
PlanID: planID,
ServiceID: "ServiceID",
SpaceGUID: "SpaceGUID",
SchedulingInfo: structs.SchedulingInfo{
Status: structs.SchedulingStatusInProgress,
},
}
clusterState.Nodes = []*structs.Node{&node}
err = state.SaveCluster(clusterState)
if err != nil {
t.Fatalf("SaveCluster failed %s", err)
}
data, err := json.Marshal(node)
key := fmt.Sprintf(
"/%s/service/%s/nodes/%s", testPrefix, clusterState.InstanceID, node.ID)
etcdApi.Set(context.Background(), key, string(data), &etcd.SetOptions{})
loadedState, err := state.LoadCluster(instanceID)
if !reflect.DeepEqual(clusterState, loadedState) {
t.Fatalf("Failed to load ClusterState. Expected: %v, actual: %v", clusterState, loadedState)
}
}
func TestState_SaveCluster(t *testing.T) {
t.Parallel()
testPrefix := "TestState_SaveCluster"
etcdApi := testutil.ResetEtcd(t, testPrefix)
logger := testutil.NewTestLogger(testPrefix, t)
state, err := NewStateEtcdWithPrefix(testutil.LocalEtcdConfig, testPrefix, logger)
if err != nil {
t.Fatalf("Could not create state", err)
}
clusterID := structs.ClusterID(uuid.New())
planID := uuid.New()
clusterState := structs.ClusterState{
InstanceID: clusterID,
OrganizationGUID: "OrganizationGUID",
PlanID: planID,
ServiceID: "ServiceID",
SpaceGUID: "SpaceGUID",
}
err = state.SaveCluster(clusterState)
if err != nil {
t.Fatalf("SaveCluster failed %s", err)
}
resp, err := etcdApi.Get(context.Background(), fmt.Sprintf("%s/service/%s/state", testPrefix, clusterID), &etcd.GetOptions{})
if err != nil {
t.Fatalf("Could not load state from etcd %s", err)
}
retrievedState := structs.ClusterState{}
json.Unmarshal([]byte(resp.Node.Value), &retrievedState)
if !reflect.DeepEqual(clusterState, retrievedState) {
t.Fatalf("Retrieved State does not match. Want %v, Got %v", clusterState, retrievedState)
}
}