Skip to content
model_test.go 59.1 KiB
Newer Older
Michel Roy's avatar
Michel Roy committed
/*
 * Copyright (c) 2019  InterDigital Communications, Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package model

import (
	"fmt"
	"testing"
Michel Roy's avatar
Michel Roy committed
	"time"
Michel Roy's avatar
Michel Roy committed

	dataModel "github.com/InterDigitalInc/AdvantEDGE/go-packages/meep-data-model"
Michel Roy's avatar
Michel Roy committed
	log "github.com/InterDigitalInc/AdvantEDGE/go-packages/meep-logger"
	"github.com/blang/semver"
Michel Roy's avatar
Michel Roy committed
)

const modelRedisAddr string = "localhost:30380"
const modelRedisTestTable = 0
Michel Roy's avatar
Michel Roy committed
const modelName string = "test-model"
const moduleName string = "test-module"
const testScenario string = `{"_id":"demo1","_rev":"5-905df5009b54170401d47031711afff7","name":"demo1","deployment":{"interDomainLatency":50,"interDomainLatencyVariation":5,"interDomainThroughput":1000,"domains":[{"id":"PUBLIC","name":"PUBLIC","type":"PUBLIC","interZoneLatency":6,"interZoneLatencyVariation":2,"interZoneThroughput":1000000,"zones":[{"id":"PUBLIC-COMMON","name":"PUBLIC-COMMON","type":"COMMON","interFogLatency":2,"interFogLatencyVariation":1,"interFogThroughput":1000000,"interEdgeLatency":3,"interEdgeLatencyVariation":1,"interEdgeThroughput":1000000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000000,"networkLocations":[{"id":"PUBLIC-COMMON-DEFAULT","name":"PUBLIC-COMMON-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"cloud1","name":"cloud1","type":"DC","processes":[{"id":"cloud1-iperf","name":"cloud1-iperf","type":"CLOUD-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"cloud1-iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"cloud1-svc","name":"cloud1-svc","type":"CLOUD-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=cloud1-svc, MGM_APP_ID=cloud1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"cloud1-svc","ports":[{"protocol":"TCP","port":80}]}}]}]}]}]},{"id":"operator1","name":"operator1","type":"OPERATOR","interZoneLatency":15,"interZoneLatencyVariation":3,"interZoneThroughput":1000,"zones":[{"id":"operator1-COMMON","name":"operator1-COMMON","type":"COMMON","interFogLatency":2,"interFogLatencyVariation":1,"interFogThroughput":1000000,"interEdgeLatency":3,"interEdgeLatencyVariation":1,"interEdgeThroughput":1000000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000000,"networkLocations":[{"id":"operator1-COMMON-DEFAULT","name":"operator1-COMMON-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1}]},{"id":"zone1","name":"zone1","type":"ZONE","interFogLatency":10,"interFogLatencyVariation":2,"interFogThroughput":1000,"interEdgeLatency":12,"interEdgeLatencyVariation":2,"interEdgeThroughput":1000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000,"networkLocations":[{"id":"zone1-DEFAULT","name":"zone1-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"zone1-edge1","name":"zone1-edge1","type":"EDGE","processes":[{"id":"zone1-edge1-iperf","name":"zone1-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone1-edge1-svc","name":"zone1-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]}]},{"id":"zone1-poa1","name":"zone1-poa1","type":"POA","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":1000,"physicalLocations":[{"id":"zone1-fog1","name":"zone1-fog1","type":"FOG","processes":[{"id":"zone1-fog1-iperf","name":"zone1-fog1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-fog1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone1-fog1-svc","name":"zone1-fog1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-fog1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-fog1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]},{"id":"ue1","name":"ue1","type":"UE","processes":[{"id":"ue1-iperf","name":"ue1-iperf","type":"UE-APP","image":"gophernet/iperf-client","commandArguments":"-c, export; iperf -u -c $IPERF_SERVICE_HOST -p $IPERF_SERVICE_PORT -t 3600 -b 50M;","commandExe":"/bin/bash"}]},{"id":"ue2-ext","name":"ue2-ext","type":"UE","isExternal":true,"processes":[{"id":"ue2-svc","name":"ue2-svc","type":"UE-APP","isExternal":true,"externalConfig":{"ingressServiceMap":[{"name":"svc","port":80,"externalPort":31111,"protocol":"TCP"},{"name":"iperf","port":80,"externalPort":31222,"protocol":"UDP"},{"name":"cloud1-svc","port":80,"externalPort":31112,"protocol":"TCP"},{"name":"cloud1-iperf","port":80,"externalPort":31223,"protocol":"UDP"}]}}]}]},{"id":"zone1-poa2","name":"zone1-poa2","type":"POA","terminalLinkLatency":10,"terminalLinkLatencyVariation":2,"terminalLinkThroughput":50}]},{"id":"zone2","name":"zone2","type":"ZONE","interFogLatency":10,"interFogLatencyVariation":2,"interFogThroughput":1000,"interEdgeLatency":12,"interEdgeLatencyVariation":2,"interEdgeThroughput":1000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000,"networkLocations":[{"id":"zone2-DEFAULT","name":"zone2-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"zone2-edge1","name":"zone2-edge1","type":"EDGE","processes":[{"id":"zone2-edge1-iperf","name":"zone2-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone2-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone2-edge1-svc","name":"zone2-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone2-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone2-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]}]},{"id":"zone2-poa1","name":"zone2-poa1","type":"POA","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":20}]}]}]}}`
const testScenario_v1_0_0 string = `{"version":"1.0.0","name":"demo1","deployment":{"interDomainLatency":50,"interDomainLatencyVariation":5,"interDomainThroughput":1000,"domains":[{"id":"PUBLIC","name":"PUBLIC","type":"PUBLIC","interZoneLatency":6,"interZoneLatencyVariation":2,"interZoneThroughput":1000000,"zones":[{"id":"PUBLIC-COMMON","name":"PUBLIC-COMMON","type":"COMMON","interFogLatency":2,"interFogLatencyVariation":1,"interFogThroughput":1000000,"interEdgeLatency":3,"interEdgeLatencyVariation":1,"interEdgeThroughput":1000000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000000,"networkLocations":[{"id":"PUBLIC-COMMON-DEFAULT","name":"PUBLIC-COMMON-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"cloud1","name":"cloud1","type":"DC","processes":[{"id":"cloud1-iperf","name":"cloud1-iperf","type":"CLOUD-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"cloud1-iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"cloud1-svc","name":"cloud1-svc","type":"CLOUD-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=cloud1-svc, MGM_APP_ID=cloud1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"cloud1-svc","ports":[{"protocol":"TCP","port":80}]}}]}]}]}]},{"id":"operator1","name":"operator1","type":"OPERATOR","interZoneLatency":15,"interZoneLatencyVariation":3,"interZoneThroughput":1000,"zones":[{"id":"operator1-COMMON","name":"operator1-COMMON","type":"COMMON","interFogLatency":2,"interFogLatencyVariation":1,"interFogThroughput":1000000,"interEdgeLatency":3,"interEdgeLatencyVariation":1,"interEdgeThroughput":1000000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000000,"networkLocations":[{"id":"operator1-COMMON-DEFAULT","name":"operator1-COMMON-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1}]},{"id":"zone1","name":"zone1","type":"ZONE","interFogLatency":10,"interFogLatencyVariation":2,"interFogThroughput":1000,"interEdgeLatency":12,"interEdgeLatencyVariation":2,"interEdgeThroughput":1000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000,"networkLocations":[{"id":"zone1-DEFAULT","name":"zone1-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"zone1-edge1","name":"zone1-edge1","type":"EDGE","processes":[{"id":"zone1-edge1-iperf","name":"zone1-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone1-edge1-svc","name":"zone1-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]}]},{"id":"zone1-poa1","name":"zone1-poa1","type":"POA","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":1000,"physicalLocations":[{"id":"zone1-fog1","name":"zone1-fog1","type":"FOG","processes":[{"id":"zone1-fog1-iperf","name":"zone1-fog1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-fog1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone1-fog1-svc","name":"zone1-fog1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-fog1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-fog1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]},{"id":"ue1","name":"ue1","type":"UE","processes":[{"id":"ue1-iperf","name":"ue1-iperf","type":"UE-APP","image":"gophernet/iperf-client","commandArguments":"-c, export; iperf -u -c $IPERF_SERVICE_HOST -p $IPERF_SERVICE_PORT -t 3600 -b 50M;","commandExe":"/bin/bash"}]},{"id":"ue2-ext","name":"ue2-ext","type":"UE","isExternal":true,"processes":[{"id":"ue2-svc","name":"ue2-svc","type":"UE-APP","isExternal":true,"externalConfig":{"ingressServiceMap":[{"name":"svc","port":80,"externalPort":31111,"protocol":"TCP"},{"name":"iperf","port":80,"externalPort":31222,"protocol":"UDP"},{"name":"cloud1-svc","port":80,"externalPort":31112,"protocol":"TCP"},{"name":"cloud1-iperf","port":80,"externalPort":31223,"protocol":"UDP"}]}}]}]},{"id":"zone1-poa2","name":"zone1-poa2","type":"POA","terminalLinkLatency":10,"terminalLinkLatencyVariation":2,"terminalLinkThroughput":50}]},{"id":"zone2","name":"zone2","type":"ZONE","interFogLatency":10,"interFogLatencyVariation":2,"interFogThroughput":1000,"interEdgeLatency":12,"interEdgeLatencyVariation":2,"interEdgeThroughput":1000,"edgeFogLatency":5,"edgeFogLatencyVariation":1,"edgeFogThroughput":1000,"networkLocations":[{"id":"zone2-DEFAULT","name":"zone2-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"zone2-edge1","name":"zone2-edge1","type":"EDGE","processes":[{"id":"zone2-edge1-iperf","name":"zone2-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone2-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone2-edge1-svc","name":"zone2-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone2-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone2-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]}]},{"id":"zone2-poa1","name":"zone2-poa1","type":"POA","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":20}]}]}]}}`
const testScenario_v1_3_0 string = `{"version":"1.3.0","name":"demo1","deployment":{"interDomainLatency":50,"interDomainLatencyVariation":5,"interDomainThroughput":1000,"domains":[{"id":"PUBLIC","name":"PUBLIC","type":"PUBLIC","interZoneLatency":6,"interZoneLatencyVariation":2,"interZoneThroughput":1000000,"zones":[{"id":"PUBLIC-COMMON","name":"PUBLIC-COMMON","type":"COMMON","netChar":{"latency":5,"latencyVariation":1,"throughput":1000000},"networkLocations":[{"id":"PUBLIC-COMMON-DEFAULT","name":"PUBLIC-COMMON-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"cloud1","name":"cloud1","type":"DC","processes":[{"id":"cloud1-iperf","name":"cloud1-iperf","type":"CLOUD-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"cloud1-iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"cloud1-svc","name":"cloud1-svc","type":"CLOUD-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=cloud1-svc, MGM_APP_ID=cloud1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"cloud1-svc","ports":[{"protocol":"TCP","port":80}]}}]}]}]}]},{"id":"operator1","name":"operator1","type":"OPERATOR","interZoneLatency":15,"interZoneLatencyVariation":3,"interZoneThroughput":1000,"zones":[{"id":"operator1-COMMON","name":"operator1-COMMON","type":"COMMON","netChar":{"latency":5,"latencyVariation":1,"throughput":1000000},"networkLocations":[{"id":"operator1-COMMON-DEFAULT","name":"operator1-COMMON-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1}]},{"id":"zone1","name":"zone1","type":"ZONE","netChar":{"latency":5,"latencyVariation":1,"throughput":1000},"networkLocations":[{"id":"zone1-DEFAULT","name":"zone1-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"zone1-edge1","name":"zone1-edge1","type":"EDGE","processes":[{"id":"zone1-edge1-iperf","name":"zone1-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone1-edge1-svc","name":"zone1-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]}]},{"id":"zone1-poa1","name":"zone1-poa1","type":"POA","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":1000,"physicalLocations":[{"id":"zone1-fog1","name":"zone1-fog1","type":"FOG","processes":[{"id":"zone1-fog1-iperf","name":"zone1-fog1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-fog1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone1-fog1-svc","name":"zone1-fog1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-fog1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-fog1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]},{"id":"ue1","name":"ue1","type":"UE","processes":[{"id":"ue1-iperf","name":"ue1-iperf","type":"UE-APP","image":"gophernet/iperf-client","commandArguments":"-c, export; iperf -u -c $IPERF_SERVICE_HOST -p $IPERF_SERVICE_PORT -t 3600 -b 50M;","commandExe":"/bin/bash"}]},{"id":"ue2-ext","name":"ue2-ext","type":"UE","isExternal":true,"processes":[{"id":"ue2-svc","name":"ue2-svc","type":"UE-APP","isExternal":true,"externalConfig":{"ingressServiceMap":[{"name":"svc","port":80,"externalPort":31111,"protocol":"TCP"},{"name":"iperf","port":80,"externalPort":31222,"protocol":"UDP"},{"name":"cloud1-svc","port":80,"externalPort":31112,"protocol":"TCP"},{"name":"cloud1-iperf","port":80,"externalPort":31223,"protocol":"UDP"}]}}]}]},{"id":"zone1-poa2","name":"zone1-poa2","type":"POA","terminalLinkLatency":10,"terminalLinkLatencyVariation":2,"terminalLinkThroughput":50}]},{"id":"zone2","name":"zone2","type":"ZONE","netChar":{"latency":5,"latencyVariation":1,"throughput":1000},"networkLocations":[{"id":"zone2-DEFAULT","name":"zone2-DEFAULT","type":"DEFAULT","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":50000,"terminalLinkPacketLoss":1,"physicalLocations":[{"id":"zone2-edge1","name":"zone2-edge1","type":"EDGE","processes":[{"id":"zone2-edge1-iperf","name":"zone2-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone2-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]}},{"id":"zone2-edge1-svc","name":"zone2-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone2-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone2-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]}}]}]},{"id":"zone2-poa1","name":"zone2-poa1","type":"POA","terminalLinkLatency":1,"terminalLinkLatencyVariation":1,"terminalLinkThroughput":20}]}]}]}}`
Michel Roy's avatar
Michel Roy committed

const testScenario_v1_5_0 string = `{"version":"1.5.0","name":"demo1","deployment":{"netChar":{"latency":50,"latencyVariation":5,"latencyDistribution":"Normal","throughputDl":1000,"throughputUl":1000},"domains":[{"id":"PUBLIC","name":"PUBLIC","type":"PUBLIC","netChar":{"latency":6,"latencyVariation":2,"throughputDl":1000000,"throughputUl":1000000},"zones":[{"id":"PUBLIC-COMMON","name":"PUBLIC-COMMON","type":"COMMON","netChar":{"latency":5,"latencyVariation":1},"networkLocations":[{"id":"PUBLIC-COMMON-DEFAULT","name":"PUBLIC-COMMON-DEFAULT","type":"DEFAULT","netChar":{"latency":1,"latencyVariation":1,"throughputDl":50000,"throughputUl":50000,"packetLoss":1},"physicalLocations":[{"id":"cloud1","name":"cloud1","type":"DC","processes":[{"id":"cloud1-iperf","name":"cloud1-iperf","type":"CLOUD-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"cloud1-iperf","ports":[{"protocol":"UDP","port":80}]},"netChar":{}},{"id":"cloud1-svc","name":"cloud1-svc","type":"CLOUD-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=cloud1-svc, MGM_APP_ID=cloud1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"cloud1-svc","ports":[{"protocol":"TCP","port":80}]},"netChar":{}}],"netChar":{}}]}]}]},{"id":"operator1","name":"operator1","type":"OPERATOR","netChar":{"latency":15,"latencyVariation":3,"throughputDl":1000,"throughputUl":1000},"zones":[{"id":"operator1-COMMON","name":"operator1-COMMON","type":"COMMON","netChar":{"latency":5,"latencyVariation":1},"networkLocations":[{"id":"operator1-COMMON-DEFAULT","name":"operator1-COMMON-DEFAULT","type":"DEFAULT","netChar":{"latency":1,"latencyVariation":1,"throughputDl":50000,"throughputUl":50000,"packetLoss":1}}]},{"id":"zone1","name":"zone1","type":"ZONE","netChar":{"latency":5,"latencyVariation":1},"networkLocations":[{"id":"zone1-DEFAULT","name":"zone1-DEFAULT","type":"DEFAULT","netChar":{"latency":1,"latencyVariation":1,"throughputDl":50000,"throughputUl":50000,"packetLoss":1},"physicalLocations":[{"id":"zone1-edge1","name":"zone1-edge1","type":"EDGE","processes":[{"id":"zone1-edge1-iperf","name":"zone1-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]},"netChar":{}},{"id":"zone1-edge1-svc","name":"zone1-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]},"netChar":{}}],"netChar":{}}]},{"id":"zone1-poa1","name":"zone1-poa1","type":"POA","netChar":{"latency":1,"latencyVariation":1,"throughputDl":1000,"throughputUl":1000},"physicalLocations":[{"id":"zone1-fog1","name":"zone1-fog1","type":"FOG","processes":[{"id":"zone1-fog1-iperf","name":"zone1-fog1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone1-fog1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]},"netChar":{}},{"id":"zone1-fog1-svc","name":"zone1-fog1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone1-fog1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone1-fog1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]},"netChar":{}}],"netChar":{}},{"id":"ue1","name":"ue1","type":"UE","processes":[{"id":"ue1-iperf","name":"ue1-iperf","type":"UE-APP","image":"gophernet/iperf-client","commandArguments":"-c, export; iperf -u -c $IPERF_SERVICE_HOST -p $IPERF_SERVICE_PORT -t 3600 -b 50M;","commandExe":"/bin/bash","netChar":{}}],"netChar":{}},{"id":"ue2-ext","name":"ue2-ext","type":"UE","isExternal":true,"processes":[{"id":"ue2-svc","name":"ue2-svc","type":"UE-APP","isExternal":true,"externalConfig":{"ingressServiceMap":[{"name":"svc","port":80,"externalPort":31111,"protocol":"TCP"},{"name":"iperf","port":80,"externalPort":31222,"protocol":"UDP"},{"name":"cloud1-svc","port":80,"externalPort":31112,"protocol":"TCP"},{"name":"cloud1-iperf","port":80,"externalPort":31223,"protocol":"UDP"}]},"netChar":{}}],"netChar":{}}]},{"id":"zone1-poa2","name":"zone1-poa2","type":"POA","netChar":{"latency":10,"latencyVariation":2,"throughputDl":50,"throughputUl":50}}]},{"id":"zone2","name":"zone2","type":"ZONE","netChar":{"latency":5,"latencyVariation":1},"networkLocations":[{"id":"zone2-DEFAULT","name":"zone2-DEFAULT","type":"DEFAULT","netChar":{"latency":1,"latencyVariation":1,"throughputDl":50000,"throughputUl":50000,"packetLoss":1},"physicalLocations":[{"id":"zone2-edge1","name":"zone2-edge1","type":"EDGE","processes":[{"id":"zone2-edge1-iperf","name":"zone2-edge1-iperf","type":"EDGE-APP","image":"gophernet/iperf-server","commandArguments":"-c, export; iperf -s -p $IPERF_SERVICE_PORT;","commandExe":"/bin/bash","serviceConfig":{"name":"zone2-edge1-iperf","meSvcName":"iperf","ports":[{"protocol":"UDP","port":80}]},"netChar":{}},{"id":"zone2-edge1-svc","name":"zone2-edge1-svc","type":"EDGE-APP","image":"meep-docker-registry:30001/demo-server","environment":"MGM_GROUP_NAME=svc, MGM_APP_ID=zone2-edge1-svc, MGM_APP_PORT=80","serviceConfig":{"name":"zone2-edge1-svc","meSvcName":"svc","ports":[{"protocol":"TCP","port":80}]},"netChar":{}}],"netChar":{}}]},{"id":"zone2-poa1","name":"zone2-poa1","type":"POA","netChar":{"latency":1,"latencyVariation":1,"throughputDl":20,"throughputUl":20}}]}]}]}}`

Michel Roy's avatar
Michel Roy committed
func TestNewModel(t *testing.T) {
	fmt.Println("--- ", t.Name())
	log.MeepTextLogInit(t.Name())

Michel Roy's avatar
Michel Roy committed
	// Switch to a different table for testing
	redisTable = modelRedisTestTable

Michel Roy's avatar
Michel Roy committed
	// Keep this one first...
	fmt.Println("Invalid Redis DB address")
	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: "ExpectedFailure-InvalidDbLocation"}
	_, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err == nil {
		t.Fatalf("Should report error on invalid Redis db")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Invalid module")
	cfg = ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "", DbAddr: modelRedisAddr}
	_, err = NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err == nil {
		t.Fatalf("Should report error on invalid module")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Create normal")
	cfg = ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: modelRedisAddr}
	_, err = NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Create no name")
	cfg = ModelCfg{Name: "", Module: "test-mod", DbAddr: modelRedisAddr}
	_, err = NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err == nil {
		t.Fatalf("Should not allow creating model without a name")
Michel Roy's avatar
Michel Roy committed
	}
}

Michel Roy's avatar
Michel Roy committed
func TestGetSetScenario(t *testing.T) {
Michel Roy's avatar
Michel Roy committed
	fmt.Println("--- ", t.Name())
	log.MeepTextLogInit(t.Name())

Michel Roy's avatar
Michel Roy committed
	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: modelRedisAddr}
	m, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}

Michel Roy's avatar
Michel Roy committed
	fmt.Println("GetSvcMap - error case")
Michel Roy's avatar
Michel Roy committed
	svcMap := m.GetServiceMaps()
Michel Roy's avatar
Michel Roy committed
	if len(*svcMap) != 0 {
		t.Fatalf("Service map unexpected")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Set Model")
Michel Roy's avatar
Michel Roy committed
	err = m.SetScenario([]byte(testScenario))
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Error setting model")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Name != "demo1" {
		t.Fatalf("SetScenario failed")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Get Model")
Michel Roy's avatar
Michel Roy committed
	s, err := m.GetScenario()
	if err != nil {
		t.Fatalf("Error getting scenario")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	if s == nil {
		t.Fatalf("Error getting scenario")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	// if s.Name != "demo1" {
	// 	t.Fatalf("GetModel failed")
Michel Roy's avatar
Michel Roy committed
	// }
Michel Roy's avatar
Michel Roy committed

	fmt.Println("GetSvcMap - existing")
	svcMap = m.GetServiceMaps()
	if svcMap == nil {
		t.Fatalf("Service map expected")
Michel Roy's avatar
Michel Roy committed
	}
}

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

Michel Roy's avatar
Michel Roy committed
	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: modelRedisAddr}
	m, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	fmt.Println("Set model")
Michel Roy's avatar
Michel Roy committed
	err = m.SetScenario([]byte(testScenario))
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Error setting model")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Name != "demo1" {
		t.Fatalf("SetScenario failed")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Activate model")
	err = m.Activate()
	if err != nil {
		t.Fatalf("Error activating model")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	fmt.Println("Set model")
Michel Roy's avatar
Michel Roy committed
	err = m.SetScenario([]byte(testScenario))
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Error updating model")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Deactivate model")
	err = m.Deactivate()
	if err != nil {
		t.Fatalf("Error deactivating model")
Michel Roy's avatar
Michel Roy committed
	}
}

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

Michel Roy's avatar
Michel Roy committed
	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: modelRedisAddr}
	m, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Set Model")
Michel Roy's avatar
Michel Roy committed
	err = m.SetScenario([]byte(testScenario))
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Error setting model")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Name != "demo1" {
		t.Fatalf("SetScenario failed")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue1")
	old, new, err := m.MoveNode("ue1", "zone2-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone1-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone2-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue2-ext")
	old, new, err = m.MoveNode("ue2-ext", "zone2-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone1-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone2-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue1 back")
	old, new, err = m.MoveNode("ue1", "zone1-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone2-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone1-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue2-ext back")
	old, new, err = m.MoveNode("ue2-ext", "zone1-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone2-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone1-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue2-ext again")
	old, new, err = m.MoveNode("ue2-ext", "zone2-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone1-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone2-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue1")
	old, new, err = m.MoveNode("ue1", "zone2-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone1-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone2-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move ue1 back again")
	old, new, err = m.MoveNode("ue1", "zone1-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone2-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone1-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Move zone1-edge1-iperf")
	_, _, err = m.MoveNode("zone1-edge1-iperf", "zone2-edge2")
	if err == nil {
		t.Fatalf("Moving Edge-App part of mobility group should not be allowed")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Move zone1-edge1-iperf")
	// Remove mobility group
	node := m.nodeMap.FindByName("zone1-edge1-iperf")
	if node == nil {
		t.Fatalf("unable to find node")
Michel Roy's avatar
Michel Roy committed
	}
	proc := node.object.(*dataModel.Process)
Michel Roy's avatar
Michel Roy committed
	proc.ServiceConfig.MeSvcName = ""
	old, new, err = m.MoveNode("zone1-edge1-iperf", "zone2-edge1")
	if err != nil {
		t.Fatalf("Error moving Edge-App")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone1-edge1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone2-edge1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Move Node - not a UE")
	_, _, err = m.MoveNode("Not-a-UE", "zone1-poa1")
	if err == nil {
		t.Fatalf("Error moving UE - inexisting UE")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Move Node - not a PoA")
	_, _, err = m.MoveNode("ue1", "Not-a-poa")
	if err == nil {
		t.Fatalf("Error moving UE - inexisting PoA")
Michel Roy's avatar
Michel Roy committed
	}

}

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

Michel Roy's avatar
Michel Roy committed
	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: modelRedisAddr}
	m, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Set Model")
Michel Roy's avatar
Michel Roy committed
	err = m.SetScenario([]byte(testScenario))
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Error setting model")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Name != "demo1" {
		t.Fatalf("SetScenario failed")
Michel Roy's avatar
Michel Roy committed
	}

	var nc dataModel.EventNetworkCharacteristicsUpdate
Michel Roy's avatar
Michel Roy committed
	nc.ElementName = "demo1"
	nc.ElementType = "SCENARIO"
	var netChar dataModel.NetworkCharacteristics
	netChar.Latency = 1
	netChar.LatencyVariation = 2
	netChar.LatencyDistribution = "Normal"
	netChar.ThroughputDl = 3
	netChar.ThroughputUl = 5
	netChar.PacketLoss = 4
	nc.NetChar = &netChar
Michel Roy's avatar
Michel Roy committed
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Deployment.NetChar.Latency != 1 {
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Deployment.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Deployment.NetChar.LatencyDistribution != "Normal" {
		t.Fatalf("Update " + nc.ElementType + " distribution failed")
	}
	if m.scenario.Deployment.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
Michel Roy's avatar
Michel Roy committed
	}
	if m.scenario.Deployment.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
	}
	if m.scenario.Deployment.NetChar.PacketLoss != 4 {
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "operator1"
	nc.ElementType = "OPERATOR"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n := m.nodeMap.FindByName(nc.ElementName)
	d := n.object.(*dataModel.Domain)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if d.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if d.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughput failed")
	}
	if d.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughput failed")
Michel Roy's avatar
Michel Roy committed
	}
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "zone1"
Michel Roy's avatar
Michel Roy committed
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	z := n.object.(*dataModel.Zone)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if z.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if z.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if z.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
	if z.NetChar.PacketLoss != 4 {
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "zone1-poa1"
	nc.ElementType = "POA"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	nl := n.object.(*dataModel.NetworkLocation)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if nl.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if nl.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if nl.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "zone1-fog1"
	nc.ElementType = "FOG"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	pl := n.object.(*dataModel.PhysicalLocation)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if pl.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "zone1-edge1"
	nc.ElementType = "EDGE"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	pl = n.object.(*dataModel.PhysicalLocation)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
	}
	if pl.NetChar.PacketLoss != 4 {
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "ue1"
	nc.ElementType = "UE"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	pl = n.object.(*dataModel.PhysicalLocation)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
	}
	if pl.NetChar.PacketLoss != 4 {
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "cloud1"
	nc.ElementType = "DISTANT CLOUD"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	pl = n.object.(*dataModel.PhysicalLocation)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if pl.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if pl.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "zone1-edge1-iperf"
	nc.ElementType = "EDGE APPLICATION"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	proc := n.object.(*dataModel.Process)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if proc.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "ue1-iperf"
	nc.ElementType = "UE APPLICATION"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	proc = n.object.(*dataModel.Process)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if proc.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "cloud1-iperf"
	nc.ElementType = "CLOUD APPLICATION"
	err = m.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	n = m.nodeMap.FindByName(nc.ElementName)
	proc = n.object.(*dataModel.Process)
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
Michel Roy's avatar
Michel Roy committed
	}
	if proc.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
	}
	if proc.NetChar.PacketLoss != 4 {
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "Not-a-Name"
	nc.ElementType = "POA"
	err = m.UpdateNetChar(&nc)
	if err == nil {
		t.Fatalf("Update " + nc.ElementType + " should fail")
Michel Roy's avatar
Michel Roy committed
	}

	nc.ElementName = "ue1"
	nc.ElementType = "Not-a-Type"
	err = m.UpdateNetChar(&nc)
		t.Fatalf("Unsupported type should fail")
Michel Roy's avatar
Michel Roy committed

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

	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	fmt.Println("Create Publisher")
	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: moduleName + "-Pub", DbAddr: modelRedisAddr}
	mPub, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	if mPub.GetScenarioName() != "" {
		t.Fatalf("Scenario name should be empty")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed

Michel Roy's avatar
Michel Roy committed
	fmt.Println("Activate")
		t.Fatalf("Failed to activate model")
Michel Roy's avatar
Michel Roy committed
	time.Sleep(50 * time.Millisecond)

	fmt.Println("Set Model")
	err = mPub.SetScenario([]byte(testScenario))
	time.Sleep(50 * time.Millisecond)
	if err != nil {
		t.Fatalf("Error setting model")
Michel Roy's avatar
Michel Roy committed
	}
	if mPub.GetScenarioName() != "demo1" {
		t.Fatalf("Scenario name should be demo1")
Michel Roy's avatar
Michel Roy committed
	}

	// create listener after model has been published to test initialization
Michel Roy's avatar
Michel Roy committed
	fmt.Println("Create Listener")
	cfg = ModelCfg{Name: "Active", Namespace: moduleNamespace, Module: moduleName + "-Lis", DbAddr: modelRedisAddr}
	mLis, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	if mLis.GetScenarioName() != "" {
		t.Fatalf("Scenario name should be empty")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed

	fmt.Println("Sync listener")
	mLis.UpdateScenario()
Michel Roy's avatar
Michel Roy committed
	lis, _ := mLis.GetScenario()
	pub, _ := mPub.GetScenario()
Michel Roy's avatar
Michel Roy committed
	if string(lis) != string(pub) {
		t.Fatalf("Published model different than received one")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	if mLis.GetScenarioName() != "demo1" {
		t.Fatalf("Scenario name should be demo1")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed

	// MoveNode
	fmt.Println("Move ue1")
	old, new, err := mPub.MoveNode("ue1", "zone2-poa1")
	if err != nil {
		t.Fatalf("Error moving UE")
Michel Roy's avatar
Michel Roy committed
	}
	if old != "zone1-poa1" {
		t.Fatalf("Move Node - wrong origin Location " + old)
Michel Roy's avatar
Michel Roy committed
	}
	if new != "zone2-poa1" {
		t.Fatalf("Move Node - wrong destination Location " + new)
Michel Roy's avatar
Michel Roy committed
	}
	mLis.UpdateScenario()
Michel Roy's avatar
Michel Roy committed
	n := mLis.nodeMap.FindByName("ue1")
	parent := n.parent.(*dataModel.NetworkLocation)
Michel Roy's avatar
Michel Roy committed
	if parent.Name != "zone2-poa1" {
		t.Fatalf("Published model not as expected")
Michel Roy's avatar
Michel Roy committed
	}

	//UpdateNetChar
	fmt.Println("Update net-char")
	var nc dataModel.EventNetworkCharacteristicsUpdate
Michel Roy's avatar
Michel Roy committed
	nc.ElementName = "demo1"
	nc.ElementType = "SCENARIO"
	var netChar dataModel.NetworkCharacteristics
	netChar.Latency = 1
	netChar.LatencyVariation = 2
	netChar.LatencyDistribution = "Normal"
	netChar.ThroughputDl = 3
	netChar.ThroughputUl = 5
	netChar.PacketLoss = 4
	nc.NetChar = &netChar
Michel Roy's avatar
Michel Roy committed
	err = mPub.UpdateNetChar(&nc)
	if err != nil {
		t.Fatalf("Update " + nc.ElementType + " failed")
Michel Roy's avatar
Michel Roy committed
	}
	mLis.UpdateScenario()
	if mLis.scenario.Deployment.NetChar.Latency != 1 {
		t.Fatalf("Update " + nc.ElementType + " latency failed")
Michel Roy's avatar
Michel Roy committed
	}
	if mLis.scenario.Deployment.NetChar.LatencyVariation != 2 {
		t.Fatalf("Update " + nc.ElementType + " jitter failed")
Michel Roy's avatar
Michel Roy committed
	}
	if mLis.scenario.Deployment.NetChar.LatencyDistribution != "Normal" {
		t.Fatalf("Update " + nc.ElementType + " distribution failed")
	}
	if mLis.scenario.Deployment.NetChar.ThroughputDl != 3 {
		t.Fatalf("Update " + nc.ElementType + " throughputDl failed")
	}
	if mLis.scenario.Deployment.NetChar.ThroughputUl != 5 {
		t.Fatalf("Update " + nc.ElementType + " throughputUl failed")
Michel Roy's avatar
Michel Roy committed
	}
	if mLis.scenario.Deployment.NetChar.PacketLoss != 4 {
		t.Fatalf("Update " + nc.ElementType + " packet loss failed")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Dectivate")
		t.Fatalf("Failed to deactivate")
Michel Roy's avatar
Michel Roy committed
	}
	mLis.UpdateScenario()
Michel Roy's avatar
Michel Roy committed
	lis, _ = mLis.GetScenario()
	if string(lis) != "{}" {
		t.Fatalf("Deployment should be nil")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
	if mPub.GetScenarioName() != "demo1" {
		t.Fatalf("Scenario name should be demo1")
Michel Roy's avatar
Michel Roy committed
	}
	if mLis.GetScenarioName() != "" {
		t.Fatalf("Scenario name should be empty")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Re-Activate")
		t.Fatalf("Failed to activate")
Michel Roy's avatar
Michel Roy committed
	}
	mLis.UpdateScenario()
Michel Roy's avatar
Michel Roy committed
	if mPub.GetScenarioName() != "demo1" {
		t.Fatalf("Scenario name should be demo1")
Michel Roy's avatar
Michel Roy committed
	}
	if mLis.GetScenarioName() != "demo1" {
		t.Fatalf("Scenario name should be demo1")
Michel Roy's avatar
Michel Roy committed
	}
Michel Roy's avatar
Michel Roy committed
}
Michel Roy's avatar
Michel Roy committed

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

	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	fmt.Println("Create Model")
	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: moduleName, DbAddr: modelRedisAddr}
	m, err := NewModel(cfg)
Michel Roy's avatar
Michel Roy committed
	if err != nil {
		t.Fatalf("Unable to create model")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Get Node Names (empty)")
	l := m.GetNodeNames("")
	if len(l) != 0 {
		t.Fatalf("Node name list should be empty")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Set Model")
	err = m.SetScenario([]byte(testScenario))
	if err != nil {
		t.Fatalf("Error setting model")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Get Node Names")
	l = m.GetNodeNames("ANY")
		t.Fatalf("Node name list should not be empty")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println(l)
	fmt.Println(len(l))

	fmt.Println("Get UE Node Names")
	l = m.GetNodeNames("UE")
	if len(l) != 2 {
		t.Fatalf("UE node name list should be 2")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println(l)
	fmt.Println(len(l))

	fmt.Println("Get POA Node Names")
	l = m.GetNodeNames("POA")
	if len(l) != 3 {
		t.Fatalf("POA node name list should be 3")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println(l)
	fmt.Println(len(l))

	fmt.Println("Get Zone Node Names")
	l = m.GetNodeNames("ZONE")
	if len(l) != 2 {
		t.Fatalf("Zone node name list should be 2")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println(l)
	fmt.Println(len(l))

	fmt.Println("Get UE + POA Node Names")
	l = m.GetNodeNames("UE", "POA")
	if len(l) != 5 {
		t.Fatalf("UE + POA node name list should be 5")
	}
	fmt.Println(l)
	fmt.Println(len(l))

	fmt.Println("Get UE + POA + ZONE Node Names")
	l = m.GetNodeNames("UE", "POA", "ZONE")
	if len(l) != 7 {
		t.Fatalf("UE + POA + ZONE node name list should be 10")
	}
	fmt.Println(l)
	fmt.Println(len(l))

Michel Roy's avatar
Michel Roy committed
	fmt.Println("Get invalid node")
	n := m.GetNode("NOT-A-NODE")
	if n != nil {
		t.Fatalf("Node should not exist")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Get ue1 node")
	n = m.GetNode("ue1")
	if n == nil {
		t.Fatalf("Failed getting ue1 node")
Michel Roy's avatar
Michel Roy committed
	}
	pl, ok := n.(*dataModel.PhysicalLocation)
Michel Roy's avatar
Michel Roy committed
	if !ok {
		t.Fatalf("ue1 has wrong type %T -- expected *model.PhysicalLocation", n)
Michel Roy's avatar
Michel Roy committed
	}
	if pl.Name != "ue1" {
		t.Fatalf("Could not find ue1")
Michel Roy's avatar
Michel Roy committed
	}

	fmt.Println("Get edges")
	edges := m.GetEdges()
		t.Fatalf("Missing edges - expected 28")
Michel Roy's avatar
Michel Roy committed
	}
	if edges["ue1"] != "zone1-poa1" {
		t.Fatalf("UE1 edge - expected zone1-poa1 -- got %s", edges["ue1"])
Michel Roy's avatar
Michel Roy committed
	}
	if edges["zone1"] != "operator1" {
		t.Fatalf("Zone1 edge - expected operator1 -- got %s", edges["zone1"])
	// Node Type
	fmt.Println("Get node type for invalid node")
	nodeType := m.GetNodeType("NOT-A-NODE")
	if nodeType != "" {
		t.Fatalf("Node type should be empty")
	}
	fmt.Println("Get node type for OPERATOR")
	nodeType = m.GetNodeType("operator1")
	if nodeType != "OPERATOR" {
		t.Fatalf("Invalid node type")
	}
	fmt.Println("Get node type for ZONE")
	nodeType = m.GetNodeType("zone1")
	if nodeType != "ZONE" {
		t.Fatalf("Invalid node type")
	}
	fmt.Println("Get node type for POA")
	nodeType = m.GetNodeType("zone1-poa1")
	if nodeType != "POA" {
		t.Fatalf("Invalid node type")
	}
	fmt.Println("Get node type for FOG")
	nodeType = m.GetNodeType("zone1-fog1")
	if nodeType != "FOG" {
		t.Fatalf("Invalid node type")
	}
	fmt.Println("Get node type for UE")
	nodeType = m.GetNodeType("ue1")
	if nodeType != "UE" {
		t.Fatalf("Invalid node type")
	}
	fmt.Println("Get node type for UE-APP")
	nodeType = m.GetNodeType("ue1-iperf")
	if nodeType != "UE-APP" {
		t.Fatalf("Invalid node type")
	}
	fmt.Println("Get node type for EDGE-APP")
	nodeType = m.GetNodeType("zone1-edge1-svc")
	if nodeType != "EDGE-APP" {
		t.Fatalf("Invalid node type")
Michel Roy's avatar
Michel Roy committed
	fmt.Println("Get context for invalid node")
	ctx := m.GetNodeContext("NOT-A-NODE")
	if ctx != nil {
		t.Fatalf("Node context should not exist")
	fmt.Println("Get Deployment context")
	ctx = m.GetNodeContext("demo1")
	if ctx == nil {
		t.Fatalf("Node context should exist")
	nodeCtx, ok := ctx.(*NodeContext)
	if !ok || !validateNodeContext(nodeCtx, "demo1", "", "", "", "") {
		t.Fatalf("Invalid Deployment context")
Michel Roy's avatar
Michel Roy committed
	fmt.Println("Get Operator context")
	ctx = m.GetNodeContext("operator1")
	if ctx == nil {
		t.Fatalf("Node context should exist")
	nodeCtx, ok = ctx.(*NodeContext)
	if !ok || !validateNodeContext(nodeCtx, "demo1", "operator1", "", "", "") {
		t.Fatalf("Invalid Operator context")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Get Zone context")
	ctx = m.GetNodeContext("zone1")
	if ctx == nil {
		t.Fatalf("Node context should exist")
	nodeCtx, ok = ctx.(*NodeContext)
	if !ok || !validateNodeContext(nodeCtx, "demo1", "operator1", "zone1", "", "") {
		t.Fatalf("Invalid Operator context")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Get Net Location context")
	ctx = m.GetNodeContext("zone1-poa1")
	if ctx == nil {
		t.Fatalf("Node context should exist")
	nodeCtx, ok = ctx.(*NodeContext)
	if !ok || !validateNodeContext(nodeCtx, "demo1", "operator1", "zone1", "zone1-poa1", "") {
		t.Fatalf("Invalid Operator context")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Get Phy Location context")
	ctx = m.GetNodeContext("zone1-fog1")
	if ctx == nil {
		t.Fatalf("Node context should exist")
	nodeCtx, ok = ctx.(*NodeContext)
	if !ok || !validateNodeContext(nodeCtx, "demo1", "operator1", "zone1", "zone1-poa1", "zone1-fog1") {
		t.Fatalf("Invalid Operator context")
Michel Roy's avatar
Michel Roy committed
	}
	fmt.Println("Get App context")
	ctx = m.GetNodeContext("ue1-iperf")
	if ctx == nil {
		t.Fatalf("Node context should exist")
	nodeCtx, ok = ctx.(*NodeContext)
	if !ok || !validateNodeContext(nodeCtx, "demo1", "operator1", "zone1", "zone1-poa1", "ue1") {
		t.Fatalf("Invalid Operator context")

	// Network Graph
	graph := m.GetNetworkGraph()
	if len(graph.Verticies) != 29 {
		t.Fatalf("Invalid Network Graph")
func TestScenarioUpdate(t *testing.T) {
	fmt.Println("--- ", t.Name())
	log.MeepTextLogInit(t.Name())

	// Switch to a different table for testing
	redisTable = modelRedisTestTable

	cfg := ModelCfg{Name: modelName, Namespace: moduleNamespace, Module: "test-mod", DbAddr: modelRedisAddr}
	m, err := NewModel(cfg)
	if err != nil {
		t.Fatalf("Unable to create model")
	}
	fmt.Println("Set Model")
	err = m.SetScenario([]byte(testScenario))
	if err != nil {
		t.Fatalf("Error setting model")
	}
	if m.scenario.Name != "demo1" {
		t.Fatalf("SetScenario failed")
	}

	fmt.Println("Invalid Add Requests")
	ue_data := dataModel.NodeDataUnion{}
	ue_node := dataModel.ScenarioNode{NodeDataUnion: &ue_data, Parent: "zone1-poa1"}
	if err == nil {
		t.Fatalf("Action should have failed")
	}
	ue_data = dataModel.NodeDataUnion{}
	ue_node = dataModel.ScenarioNode{Type_: NodeTypeUE, Parent: "zone1-poa1"}
	if err == nil {
		t.Fatalf("Action should have failed")
	}