271 lines
7.8 KiB
Python
271 lines
7.8 KiB
Python
# (C) Copyright Broadcom Corporation 2016
|
|
#
|
|
# 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.
|
|
|
|
from agentapi import AgentAPI
|
|
from broadview_lib.config.broadviewconfig import BroadViewBSTSwitches
|
|
import unittest
|
|
|
|
class GetSwitchProperties(AgentAPI):
|
|
def __init__(self, host, port):
|
|
super(GetSwitchProperties, self).__init__()
|
|
self.setFeature("")
|
|
self.setHttpMethod("POST")
|
|
self.setHost(host)
|
|
self.setPort(port)
|
|
self.__asic_id = "1"
|
|
self.__json = None
|
|
|
|
def setASIC(self, val):
|
|
self.__asic_id = val
|
|
|
|
def toDict(self):
|
|
ret = {}
|
|
params = {}
|
|
ret["asic-id"] = self.__asic_id
|
|
ret["params"] = params
|
|
ret["method"] = "get-switch-properties"
|
|
return ret
|
|
|
|
def getJSON(self):
|
|
return self.__json
|
|
|
|
def send(self, timeout=30):
|
|
status, json = self._send(self.toDict(), timeout)
|
|
if status == 200:
|
|
self.__version = json["version"]
|
|
res = json["result"]
|
|
self.__json = res
|
|
return status
|
|
|
|
class GetSystemFeature(AgentAPI):
|
|
def __init__(self, host, port):
|
|
super(GetSystemFeature, self).__init__()
|
|
self.setFeature("")
|
|
self.setHttpMethod("POST")
|
|
self.setHost(host)
|
|
self.setPort(port)
|
|
self.__asic_id = "1"
|
|
self.__json = None
|
|
|
|
def setASIC(self, val):
|
|
self.__asic_id = val
|
|
|
|
def toDict(self):
|
|
ret = {}
|
|
params = {}
|
|
ret["asic-id"] = self.__asic_id
|
|
ret["params"] = params
|
|
ret["method"] = "get-system-feature"
|
|
return ret
|
|
|
|
def getJSON(self):
|
|
return self.__json
|
|
|
|
def send(self, timeout=30):
|
|
status, json = self._send(self.toDict(), timeout)
|
|
if status == 200:
|
|
self.__version = json["version"]
|
|
res = json["result"]
|
|
self.__json = res
|
|
return status
|
|
|
|
class ConfigureSystemFeature(AgentAPI):
|
|
def __init__(self, host, port):
|
|
super(ConfigureSystemFeature, self).__init__()
|
|
self.setFeature("")
|
|
self.setHttpMethod("POST")
|
|
self.setHost(host)
|
|
self.setPort(port)
|
|
self.__asic_id = "1"
|
|
self.__json = None
|
|
self.__heartbeatEnable = False
|
|
self.__msgInterval = 30
|
|
|
|
def setASIC(self, val):
|
|
self.__asic_id = val
|
|
|
|
def setHeartbeatEnable(self, val):
|
|
self.__heartbeatEnable = val
|
|
|
|
def setMsgInterval(self, val):
|
|
self.__msgInterval = val
|
|
|
|
def toDict(self):
|
|
ret = {}
|
|
params = {}
|
|
params["heartbeat-enable"] = 1 if self.__heartbeatEnable else 0
|
|
params["msg-interval"] = self.__msgInterval
|
|
ret["asic-id"] = self.__asic_id
|
|
ret["params"] = params
|
|
ret["method"] = "configure-system-feature"
|
|
return ret
|
|
|
|
def getJSON(self):
|
|
return self.__json
|
|
|
|
def send(self, timeout=30):
|
|
status, json = self._send(self.toDict(), timeout)
|
|
if status == 200:
|
|
self.__version = json["version"]
|
|
res = json["result"]
|
|
self.__json = res
|
|
return status
|
|
|
|
class CancelRequest(AgentAPI):
|
|
def __init__(self, host, port):
|
|
super(CancelRequest, self).__init__()
|
|
self.setFeature("")
|
|
self.setHttpMethod("POST")
|
|
self.setHost(host)
|
|
self.setPort(port)
|
|
self.__asic_id = "1"
|
|
self.__json = None
|
|
self.__requestId = None
|
|
|
|
def setASIC(self, val):
|
|
self.__asic_id = val
|
|
|
|
def setRequestId(self, val):
|
|
self.__requestId = val
|
|
|
|
def toDict(self):
|
|
ret = {}
|
|
params = {}
|
|
params["request-id"] = self.__requestId
|
|
ret["asic-id"] = self.__asic_id
|
|
ret["params"] = params
|
|
ret["method"] = "cancel-request"
|
|
return ret
|
|
|
|
def getJSON(self):
|
|
return self.__json
|
|
|
|
def send(self, timeout=30):
|
|
status, json = self._send(self.toDict(), timeout)
|
|
if status == 200:
|
|
self.__version = json["version"]
|
|
res = json["result"]
|
|
self.__json = res
|
|
return status
|
|
|
|
class TestTAPIParams(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def test_GetSwitchProperties(self):
|
|
|
|
sw = BroadViewBSTSwitches()
|
|
if len(sw):
|
|
for x in sw:
|
|
host = x["ip"]
|
|
port = x["port"]
|
|
break
|
|
else:
|
|
host = "192.168.3.1"
|
|
port = 8080
|
|
|
|
x = GetSwitchProperties(host, port)
|
|
d = x.toDict()
|
|
self.assertTrue("asic-id" in d)
|
|
self.assertTrue("params" in d)
|
|
self.assertTrue("method" in d)
|
|
self.assertTrue(x.getFeature() == "")
|
|
self.assertTrue(x.getHttpMethod() == "POST")
|
|
self.assertTrue(x.getHost() == host)
|
|
self.assertTrue(x.getPort() == port)
|
|
self.assertTrue(d["asic-id"] == "1")
|
|
self.assertTrue(d["method"] == "get-switch-properties")
|
|
|
|
def test_GetSystemFeature(self):
|
|
|
|
sw = BroadViewBSTSwitches()
|
|
if len(sw):
|
|
for x in sw:
|
|
host = x["ip"]
|
|
port = x["port"]
|
|
break
|
|
else:
|
|
host = "192.168.3.1"
|
|
port = 8080
|
|
|
|
x = GetSystemFeature(host, port)
|
|
d = x.toDict()
|
|
self.assertTrue("asic-id" in d)
|
|
self.assertTrue("params" in d)
|
|
self.assertTrue("method" in d)
|
|
self.assertTrue(x.getFeature() == "")
|
|
self.assertTrue(x.getHttpMethod() == "POST")
|
|
self.assertTrue(x.getHost() == host)
|
|
self.assertTrue(x.getPort() == port)
|
|
self.assertTrue(d["asic-id"] == "1")
|
|
self.assertTrue(d["method"] == "get-system-feature")
|
|
|
|
def test_ConfigureSystemFeature(self):
|
|
|
|
sw = BroadViewBSTSwitches()
|
|
if len(sw):
|
|
for x in sw:
|
|
host = x["ip"]
|
|
port = x["port"]
|
|
break
|
|
else:
|
|
host = "192.168.3.1"
|
|
port = 8080
|
|
|
|
x = ConfigureSystemFeature(host, port)
|
|
x.setHeartbeatEnable(False)
|
|
x.setMsgInterval(10)
|
|
d = x.toDict()
|
|
self.assertTrue("asic-id" in d)
|
|
self.assertTrue("params" in d)
|
|
self.assertTrue(d["params"]["msg-interval"] == 10)
|
|
self.assertTrue(d["params"]["heartbeat-enable"] == 0)
|
|
self.assertTrue("method" in d)
|
|
self.assertTrue(x.getFeature() == "")
|
|
self.assertTrue(x.getHttpMethod() == "POST")
|
|
self.assertTrue(x.getHost() == host)
|
|
self.assertTrue(x.getPort() == port)
|
|
self.assertTrue(d["asic-id"] == "1")
|
|
self.assertTrue(d["method"] == "configure-system-feature")
|
|
|
|
def test_CancelRequest(self):
|
|
|
|
sw = BroadViewBSTSwitches()
|
|
if len(sw):
|
|
for x in sw:
|
|
host = x["ip"]
|
|
port = x["port"]
|
|
break
|
|
else:
|
|
host = "192.168.3.1"
|
|
port = 8080
|
|
|
|
x = CancelRequest(host, port)
|
|
x.setRequestId(2)
|
|
d = x.toDict()
|
|
self.assertTrue("asic-id" in d)
|
|
self.assertTrue("params" in d)
|
|
self.assertTrue(d["params"]["request-id"] == 2)
|
|
self.assertTrue("method" in d)
|
|
self.assertTrue(x.getFeature() == "")
|
|
self.assertTrue(x.getHttpMethod() == "POST")
|
|
self.assertTrue(x.getHost() == host)
|
|
self.assertTrue(x.getPort() == port)
|
|
self.assertTrue(d["asic-id"] == "1")
|
|
self.assertTrue(d["method"] == "cancel-request")
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|