def test_api_version_routing(self):
"""Test api_builder.VersionRouter inits class based on class correctly"""
api_client = MSClient(configuration=Configuration(
host="0.0.0.0",
username="******",
password="******",
verify_ssl=False,
))
api_client.ms_version = "2.1.1"
expected_api_methods = [
"put_activate_api_key",
"post_create_token",
"put_expire_token",
"put_invalidate_api_key_tokens",
"get_api_keys",
"post_create_api_key",
"put_update_api_key",
"delete_api_key",
]
access_api = api_client.access
for method_name in expected_api_methods:
assert hasattr(access_api,
method_name), ("Missing method %s on access_api" %
method_name)
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 ms_client():
return MSClient(configuration=Configuration(
host=MockConstants.Host,
username=MockConstants.User,
api_key=MockConstants.ApiKey,
api_token=MockConstants.ApiToken,
verify_ssl=MockConstants.VerifySSL,
))
def test_host_uri_property(self):
api_client = MSClient(configuration=Configuration(
host="0.0.0.0:443",
username="******",
api_key="abcdefge",
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_latest_available_version(self):
api_client = MSClient(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
api_key="abcdefge",
verify_ssl=False,
))
latest_version = api_client.latest_version()
assert LATEST_VNS3_MS_VERSION == latest_version
def test_get_client_state(self):
api_client = MSClient(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
api_key="abcdefge",
verify_ssl=False,
))
state = api_client.state
assert type(state) is dict
assert len(state) == 0
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_client_state(self):
api_client = MSClient(configuration=Configuration(
host="0.0.0.0:8000",
username="******",
api_key="abcdefge",
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_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 = MSClient(configuration=Configuration(
host="0.0.0.0",
username="******",
api_key="abcdefge",
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_client_state(self):
api_client = MSClient(configuration=Configuration(
host="0.0.0.0",
username="******",
api_key="abcdefge",
verify_ssl=False,
))
updates = {
"topology_container_network": "192.169.16.0/24",
"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_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
def test_api_properties(self):
"""Test all API groups are accessible as properties"""
api_client = MSClient(configuration=Configuration(
host="0.0.0.0",
username="******",
api_key="abcdefge",
verify_ssl=False,
))
assert hasattr(api_client, "access")
assert type(api_client.access) is vns3ms_api.AccessApi
assert hasattr(api_client, "admin")
assert type(api_client.admin) is vns3ms_api.AdministrationApi
assert hasattr(api_client, "backups")
assert type(api_client.backups) is vns3ms_api.BackupsApi
assert hasattr(api_client, "cloud_monitoring")
assert type(
api_client.cloud_monitoring) is vns3ms_api.CloudMonitoringApi
assert hasattr(api_client, "system")
assert type(api_client.system) is vns3ms_api.SystemApi
assert hasattr(api_client, "user")
assert type(api_client.user) is vns3ms_api.UserApi
assert hasattr(api_client, "vns3_management")
assert type(api_client.vns3_management) is vns3ms_api.VNS3ManagementApi
