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)
}
}
