def vns3_client():
return VNS3Client(configuration=Configuration(
host="%s:%s" % (MockConstants.Host, MockConstants.Port),
username=MockConstants.User,
password=MockConstants.Password,
verify_ssl=MockConstants.VerifySSL,
))
def get_vns3_client(host,
username=None,
password=None,
api_token=None,
verify=False):
"""Get VNS3 API Client for host
Arguments:
host {str}
username {str}
password {str}
Keyword Arguments:
verify {bool} -- Verify SSL certificate (default: {False})
Returns:
[VNS3Client]
"""
basic_auth = all([username, password])
token_auth = api_token is not None
assert (basic_auth or
token_auth), "Must provide either username/password or api_token"
return VNS3Client(
Configuration(
host=host,
username=username,
password=password,
api_token=api_token,
verify_ssl=verify,
))
def test_host_uri_property(self):
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
assert api_client.host_uri == "0.0.0.0"
def test_api_version_routing(self):
"""Test api_builder.VersionRouter inits class based on class correctly"""
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
api_client.add_to_state("vns3_version", "4.8.4")
def test_get_controller_state(self):
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
state = api_client.state
assert type(state) is dict
assert len(state) == 0
def test_add_controller_state(self):
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
api_client.add_to_state("private_ip", "10.0.24.30")
state = api_client.state
assert "private_ip" in state and state["private_ip"] == "10.0.24.30"
assert len(state) == 1
def test_query_state(self):
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
test_value = 12312
updates = {
"container_network": "192.169.16.0/28",
"subnet": "10.0.2.0/24",
"test_key": test_value,
}
api_client.update_state(updates)
assert api_client.query_state("test_key") == test_value
def test_update_controller_state(self):
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
updates = {
"container_network": "192.169.16.0/28",
"subnet": "10.0.2.0/24"
}
api_client.update_state(updates)
state = api_client.state
assert set(updates.keys()).issubset(set(state.keys()))
assert state["subnet"] == updates["subnet"]
assert len(state) == 2
def test_api_properties(self):
"""Test all API groups are accessible as properties"""
api_client = VNS3Client(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
password="******",
verify_ssl=False,
))
assert hasattr(api_client, "bgp")
assert type(api_client.bgp) is cohesivenet.BGPApi
assert hasattr(api_client, "config")
assert type(api_client.config) is cohesivenet.ConfigurationApi
assert hasattr(api_client, "firewall")
assert type(api_client.firewall) is cohesivenet.FirewallApi
assert hasattr(api_client, "access")
assert type(api_client.access) is cohesivenet.AccessApi
assert hasattr(api_client, "ipsec")
assert type(api_client.ipsec) is cohesivenet.IPsecApi
assert hasattr(api_client, "interfaces")
assert type(api_client.interfaces) is cohesivenet.InterfacesApi
assert hasattr(api_client, "licensing")
assert type(api_client.licensing) is cohesivenet.LicensingApi
assert hasattr(api_client, "monitoring")
assert type(api_client.monitoring) is cohesivenet.MonitoringAlertingApi
assert hasattr(api_client, "network_edge_plugins")
assert (type(api_client.network_edge_plugins) is
cohesivenet.NetworkEdgePluginsApi)
assert hasattr(api_client, "overlay_network")
assert type(
api_client.overlay_network) is cohesivenet.OverlayNetworkApi
assert hasattr(api_client, "peering")
assert type(api_client.peering) is cohesivenet.PeeringApi
assert hasattr(api_client, "routing")
assert type(api_client.routing) is cohesivenet.RoutingApi
assert hasattr(api_client, "snapshots")
assert type(api_client.snapshots) is cohesivenet.SnapshotsApi
assert hasattr(api_client, "sys_admin")
assert type(
api_client.sys_admin) is cohesivenet.SystemAdministrationApi