model_test.go

		t.Fatalf("Action should have failed")
	}
	ue_data = dataModel.NodeDataUnion{}
	ue_node = dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue_data}
	nodes = []dataModel.ScenarioNode{ue_node}
	_, err = m.RemoveScenarioNodes(&nodes)
	if err == nil {
		t.Fatalf("Action should have failed")
	}
	ue_pl = dataModel.PhysicalLocation{Id: "ue-id", Type_: NodeTypeUE}
	ue_data = dataModel.NodeDataUnion{PhysicalLocation: &ue_pl}
	ue_node = dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue_data}
	nodes = []dataModel.ScenarioNode{ue_node}
	_, err = m.RemoveScenarioNodes(&nodes)
	if err == nil {
		t.Fatalf("Action should have failed")
	}
	ue_pl = dataModel.PhysicalLocation{Id: "ue-id", Name: "ue"}
	ue_data = dataModel.NodeDataUnion{PhysicalLocation: &ue_pl}
	ue_node = dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue_data}
	nodes = []dataModel.ScenarioNode{ue_node}
	_, err = m.RemoveScenarioNodes(&nodes)
	if err == nil {
		t.Fatalf("Action should have failed")
	}

	fmt.Println("Add ue3")
	ue3_pl := dataModel.PhysicalLocation{Id: "ue3-id", Name: "ue3", Type_: NodeTypeUE}
	ue3_data := dataModel.NodeDataUnion{PhysicalLocation: &ue3_pl}
	ue3_node := dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue3_data, Parent: "zone1-poa1"}
	nodes = []dataModel.ScenarioNode{ue3_node}
	names, err := m.AddScenarioNodes(&nodes)
	if err != nil || len(names) != 1 {
		t.Fatalf("Error adding nodes")
	}
	n := m.nodeMap.FindByName("ue3")
	if n == nil || n.name != "ue3" {
		t.Fatalf("Failed to add nodes")
	}
	p := n.parent.(*dataModel.NetworkLocation)
	if p == nil || p.Name != "zone1-poa1" {
		t.Fatalf("Failed to add nodes")
	}
	d := n.object.(*dataModel.PhysicalLocation)
	if d == nil || d.Name != "ue3" {
		t.Fatalf("Failed to add nodes")
	}

	fmt.Println("Add ue4 & ue5")
	ue4_pl := dataModel.PhysicalLocation{Id: "ue4-id", Name: "ue4", Type_: NodeTypeUE}
	ue4_data := dataModel.NodeDataUnion{PhysicalLocation: &ue4_pl}
	ue4_node := dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue4_data, Parent: "zone1-poa1"}
	ue5_pl := dataModel.PhysicalLocation{Id: "ue5-id", Name: "ue5", Type_: NodeTypeUE}
	ue5_data := dataModel.NodeDataUnion{PhysicalLocation: &ue5_pl}
	ue5_node := dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue5_data, Parent: "zone1-poa1"}
	nodes = []dataModel.ScenarioNode{ue4_node, ue5_node}
	names, err = m.AddScenarioNodes(&nodes)
	if err != nil || len(names) != 2 {
		t.Fatalf("Error adding nodes")
	}
	n = m.nodeMap.FindByName("ue4")
	if n == nil || n.name != "ue4" {
		t.Fatalf("Failed to add nodes")
	}
	p = n.parent.(*dataModel.NetworkLocation)
	if p == nil || p.Name != "zone1-poa1" {
		t.Fatalf("Failed to add nodes")
	}
	d = n.object.(*dataModel.PhysicalLocation)
	if d == nil || d.Name != "ue4" {
		t.Fatalf("Failed to add nodes")
	}
	n = m.nodeMap.FindByName("ue5")
	if n == nil || n.name != "ue5" {
		t.Fatalf("Failed to add nodes")
	}
	p = n.parent.(*dataModel.NetworkLocation)
	if p == nil || p.Name != "zone1-poa1" {
		t.Fatalf("Failed to add nodes")
	}
	d = n.object.(*dataModel.PhysicalLocation)
	if d == nil || d.Name != "ue5" {
		t.Fatalf("Failed to add nodes")
	}

	fmt.Println("Remove ue4")
	ue4_node = dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue4_data}
	nodes = []dataModel.ScenarioNode{ue4_node}
	names, err = m.RemoveScenarioNodes(&nodes)
	if err != nil || len(names) != 1 {
		t.Fatalf("Error removing nodes")
	}
	n = m.nodeMap.FindByName("ue4")
	if n != nil {
		t.Fatalf("Failed to remove nodes")
	}

	fmt.Println("Remove ue3 & ue5")
	ue3_node = dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue3_data}
	ue5_node = dataModel.ScenarioNode{Type_: NodeTypeUE, NodeDataUnion: &ue5_data}
	nodes = []dataModel.ScenarioNode{ue3_node, ue5_node}
	names, err = m.RemoveScenarioNodes(&nodes)
	if err != nil || len(names) != 2 {
		t.Fatalf("Error removing nodes")
	}
	n = m.nodeMap.FindByName("ue3")
	if n != nil {
		t.Fatalf("Failed to remove nodes")
	}
	n = m.nodeMap.FindByName("ue5")
	if n != nil {
		t.Fatalf("Failed to remove nodes")
	}
}

func validateNodeContext(nodeCtx *NodeContext, deployment, domain, zone, netLoc, phyLoc string) bool {
	if nodeCtx.Parents[Deployment] != deployment ||
		nodeCtx.Parents[Domain] != domain ||
		nodeCtx.Parents[Zone] != zone ||
		nodeCtx.Parents[NetLoc] != netLoc ||
		nodeCtx.Parents[PhyLoc] != phyLoc {
		return false
	}
	return true
}

func TestValidateScenario(t *testing.T) {
	fmt.Println("--- ", t.Name())
	log.MeepTextLogInit(t.Name())

	// Incompatible scenarios
	fmt.Println("Validate empty scenario")
	validJsonScenario, status, err := ValidateScenario([]byte(""))
	if validJsonScenario != nil || status != ValidatorStatusError || err == nil {
		t.Fatalf("Empty scenario should not be valid")
	}

	fmt.Println("Validate scenario: scenarioVer[none] < validatorVer[2.0.0]")
	ValidatorVersion = semver.Version{Major: 2, Minor: 0, Patch: 0}
	validJsonScenario, status, err = ValidateScenario([]byte(testScenario))
	if validJsonScenario != nil || status != ValidatorStatusError || err == nil {
		t.Fatalf("validJsonScenario should not be compatible")
	}

	fmt.Println("Validate scenario: scenarioVer[1.0.0] < validatorVer[2.0.0]")
	ValidatorVersion = semver.Version{Major: 2, Minor: 0, Patch: 0}
	validJsonScenario, status, err = ValidateScenario([]byte(testScenario_v1_0_0))
	if validJsonScenario != nil || status != ValidatorStatusError || err == nil {
		t.Fatalf("validJsonScenario should not be compatible")
	}

	// Compatible Scenarios
	fmt.Println("Validate scenario: scenarioVer[none] < validatorVer[1.3.0]")
	ValidatorVersion = semver.Version{Major: 1, Minor: 3, Patch: 0}
	validJsonScenario, status, err = ValidateScenario([]byte(testScenario))
	if validJsonScenario == nil || status != ValidatorStatusUpdated || err != nil {
		t.Fatalf("validJsonScenario should not be nil")
	}
	if string(validJsonScenario) != testScenario_v1_3_0 {
		t.Fatalf("validJsonScenario != testScenario_v1_3_0")
	}

	fmt.Println("Validate scenario: scenarioVer[1.0.0] < validatorVer[1.3.0]")
	ValidatorVersion = semver.Version{Major: 1, Minor: 3, Patch: 0}
	validJsonScenario, status, err = ValidateScenario([]byte(testScenario_v1_0_0))
	if validJsonScenario == nil || status != ValidatorStatusUpdated || err != nil {
		t.Fatalf("validJsonScenario should not be nil")
	}
	if string(validJsonScenario) != testScenario_v1_3_0 {
		t.Fatalf("validJsonScenario != testScenario_v1_3_0")
	}

	fmt.Println("Validate scenario: scenarioVer[1.3.0] == validatorVer[1.3.0]")
	ValidatorVersion = semver.Version{Major: 1, Minor: 3, Patch: 0}
	validJsonScenario, status, err = ValidateScenario([]byte(testScenario_v1_3_0))
	if validJsonScenario == nil || status != ValidatorStatusValid || err != nil {
		t.Fatalf("validJsonScenario should not be nil")
	}
	if string(validJsonScenario) != testScenario_v1_3_0 {
		t.Fatalf("validJsonScenario != testScenario_v1_3_0")
	}
}