afonso/core-extra
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

grpc added mobility tests

Browse source
This commit is contained in:
bharnden 2019-03-25 12:59:07 -07:00
parent 03c221efa9
commit b15e525cc1
3 changed files with 65 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
daemon/core/grpc/server.py
View file

@ -731,7 +731,7 @@ class CoreGrpcServer(core_pb2_grpc.CoreApiServicer):
def GetMobilityConfigs(self, request, context):
logging.debug("get mobility configs: %s", request)
session = self.get_session(request.session, context)
mobility_configs = {}
response = core_pb2.GetMobilityConfigsResponse()
for node_id, model_config in session.mobility.node_configurations.iteritems():
if node_id == -1:
continue
@ -740,8 +740,8 @@ class CoreGrpcServer(core_pb2_grpc.CoreApiServicer):
continue
config = session.mobility.get_model_config(node_id, model_name)
groups = get_config_groups(config, Ns2ScriptedMobility)
mobility_configs[node_id] = groups
return core_pb2.GetMobilityConfigsResponse(configs=mobility_configs)
response.configs[node_id].groups.extend(groups)
return response
def GetMobilityConfig(self, request, context):
logging.debug("get mobility config: %s", request)