mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-30 18:27:53 +00:00
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import os
|
|
import importlib
|
|
|
|
|
|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
|
|
|
lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
|
|
LFCliBase = lfcli_base.LFCliBase
|
|
|
|
|
|
class TestGroupProfile(LFCliBase):
|
|
def __init__(self, lfclient_host, lfclient_port, local_realm, test_group_name=None, debug_=False):
|
|
super().__init__(lfclient_host, lfclient_port, debug_)
|
|
self.local_realm = local_realm
|
|
self.group_name = test_group_name
|
|
self.cx_list = []
|
|
|
|
def start_group(self):
|
|
if self.group_name is not None:
|
|
self.local_realm.json_post("/cli-json/start_group", {"name": self.group_name})
|
|
else:
|
|
raise ValueError("test_group name must be set.")
|
|
|
|
def quiesce_group(self):
|
|
if self.group_name is not None:
|
|
self.local_realm.json_post("/cli-json/quiesce_group", {"name": self.group_name})
|
|
else:
|
|
raise ValueError("test_group name must be set.")
|
|
|
|
def stop_group(self):
|
|
if self.group_name is not None:
|
|
self.local_realm.json_post("/cli-json/stop_group", {"name": self.group_name})
|
|
else:
|
|
raise ValueError("test_group name must be set.")
|
|
|
|
def create_group(self):
|
|
if self.group_name is not None:
|
|
self.local_realm.json_post("/cli-json/add_group", {"name": self.group_name})
|
|
else:
|
|
raise ValueError("test_group name must be set.")
|
|
|
|
def rm_group(self):
|
|
if self.group_name is not None:
|
|
self.local_realm.json_post("/cli-json/rm_group", {"name": self.group_name})
|
|
else:
|
|
raise ValueError("test_group name must be set.")
|
|
|
|
def add_cx(self, cx_name):
|
|
self.local_realm.json_post("/cli-json/add_tgcx", {"tgname": self.group_name, "cxname": cx_name})
|
|
|
|
def rm_cx(self, cx_name):
|
|
self.local_realm.json_post("/cli-json/rm_tgcx", {"tgname": self.group_name, "cxname": cx_name})
|
|
|
|
def check_group_exists(self):
|
|
test_groups = self.local_realm.json_get("/testgroups/all")
|
|
if test_groups is not None and "groups" in test_groups:
|
|
test_groups = test_groups["groups"]
|
|
for group in test_groups:
|
|
for k, v in group.items():
|
|
if v['name'] == self.group_name:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def list_groups(self):
|
|
test_groups = self.local_realm.json_get("/testgroups/all")
|
|
tg_list = []
|
|
if test_groups is not None:
|
|
test_groups = test_groups["groups"]
|
|
for group in test_groups:
|
|
for k, v in group.items():
|
|
tg_list.append(v['name'])
|
|
return tg_list
|
|
|
|
def list_cxs(self):
|
|
test_groups = self.local_realm.json_get("/testgroups/all")
|
|
if test_groups is not None:
|
|
test_groups = test_groups["groups"]
|
|
for group in test_groups:
|
|
for k, v in group.items():
|
|
if v['name'] == self.group_name:
|
|
return v['cross connects']
|
|
|
|
else:
|
|
return []
|