def test_validate_credentials():
session = mock.mock_auth()
httpretty.register_uri(httpretty.POST,
"http://test/themis/v1/validate/aws/arn",
status=200)
themis = alertlogic.services.Themis(session)
response = themis.validate_credentials("2", "aws", "arn", "v2", "test_arn")
assert (response.status_code == 200)
def test_redeploy():
session = mock.mock_auth()
httpretty.register_uri(httpretty.POST,
"http://test/saturn/v1/2/redeploy",
status=200)
saturn = alertlogic.services.Saturn(session)
response = saturn.redeploy("2")
assert (response.status_code == 200)
def test_create_credentials():
session = mock.mock_auth()
httpretty.register_uri(httpretty.POST,
"http://test/credentials/v2/2/credentials",
status=200)
credentials = alertlogic.services.Credentials(session)
response = credentials.create_credential("2", "name", "arn")
assert (response.status_code == 200)
def test_estimation():
session = mock.mock_auth()
httpretty.register_uri(httpretty.GET,
"http://test/skaletor/v1/2/scanners",
status=200)
skaletor = alertlogic.services.Skaletor(session)
response = skaletor.get_scanner_estimation("2")
assert (response.status_code == 200)
def test_deployed_installations():
session = mock.mock_auth()
httpretty.register_uri(httpretty.GET,
"http://test/saturn/v1/2/installations",
status=200)
saturn = alertlogic.services.Saturn(session)
response = saturn.deployed_installations("2")
assert (response.status_code == 200)
def test_get_role():
session = mock.mock_auth()
httpretty.register_uri(httpretty.GET,
"http://test/themis/v1/2/roles/aws/arn/v2",
status=200)
themis = alertlogic.services.Themis(session)
response = themis.get_role("2", "aws", "arn", "v2")
assert (response.status_code == 200)
def test_delete_source():
session = mock.mock_auth()
httpretty.register_uri(httpretty.DELETE,
"http://test/sources/v1/2/sources/3",
status=204)
sources = alertlogic.services.Sources(session)
response = sources.delete_source("2", "3")
assert (response.status_code == 204)
def test_list_deployed():
session = mock.mock_auth()
httpretty.register_uri(httpretty.GET,
"http://test/launcher/v1/2/3/resources",
status=200)
launcher = alertlogic.services.Launcher(session)
response = launcher.list_deployed("2", "3")
assert (response.status_code == 200)
def test_deployment_status():
session = mock.mock_auth()
httpretty.register_uri(httpretty.GET,
"http://test/launcher/v1/2/environments/3",
status=200)
launcher = alertlogic.services.Launcher(session)
response = launcher.deployment_status("2", "3")
assert (response.status_code == 200)
def test_delete_credentials():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.DELETE,
"http://test/credentials/v2/2/credentials/credential_Id",
status=200)
credentials = alertlogic.services.Credentials(session)
response = credentials.delete_credential("2", "credential_Id")
assert (response.status_code == 200)
def test_get_id():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.GET,
"http://test/scancollect/v1_remediation/2/3/appliance_vmserver_id/app_id",
status=200)
scancollect = alertlogic.services.ScanCollect(session)
response = scancollect.get_appliance_vmserver_id("2", "3", "app_id")
assert (response.status_code == 200)
def test():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.GET,
"http://test/assets_query/v1/2/deployments/3/assets",
status=200)
assets = alertlogic.services.AssetsQuery(session)
response = assets.get_assets_in_deployment("2", "3")
assert (response.status_code == 200)
def test_delete_deployment():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.DELETE,
"http://test/deployments/v1/2/deployments/deployment_id",
status=204
)
deployments = alertlogic.services.Deployments(session)
response = deployments.delete_deployment("2", "deployment_id")
assert (response.status_code == 204)
def test_get_options_fail():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.GET,
"http://test/otis/v2/2/options",
status=500
)
otis = alertlogic.services.Otis(session)
result = otis.get_options("2")
assert (result.status_code == 500)
def test_list_scan_assets():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.GET,
"http://test/scheduler/v1/2/3/list",
status=200
)
scheduler = alertlogic.services.ScanScheduler(session)
response = scheduler.list_scan_assets("2", "3")
assert (response.status_code == 200)
def test_scan_host():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.PUT,
"http://test/scheduler/v1/2/3/scan?assetkey=abc",
status=200
)
scheduler = alertlogic.services.ScanScheduler(session)
response = scheduler.scan_host("2", "3", "abc")
assert (response.status_code == 200)
def test_list_deployments():
session = mock.mock_auth()
httpretty.register_uri(
httpretty.GET,
"http://test/deployments/v1/2/deployments",
status=200
)
deployments = alertlogic.services.Deployments(session)
response = deployments.list_deployments("2")
assert (response.status_code == 200)
def test_get_source():
session = mock.mock_auth()
response_body = {"source": {"config": {"deployment_mode": "automatic"}}}
httpretty.register_uri(httpretty.GET,
"http://test/sources/v1/2/sources/3",
status=200,
content_type="text/json",
body=json.dumps(response_body))
sources = alertlogic.services.Sources(session)
response = sources.get_source("2", "3")
assert (response.json() == response_body)
def test_update_deployments():
session = mock.mock_auth()
update_json = "updated_deployment"
httpretty.register_uri(
httpretty.PUT,
"http://test/deployments/v1/2/deployments/deployment_id",
status=200
)
deployments = alertlogic.services.Deployments(session)
response = deployments.update_deployment("2", "deployment_id", update_json)
assert (response.status_code == 200)
def test_merge_sources():
session = mock.mock_auth()
response_body = "merged_source"
httpretty.register_uri(httpretty.POST,
"http://test/sources/v1/2/sources/3",
status=200,
content_type="text/json",
body=json.dumps(response_body))
sources = alertlogic.services.Sources(session)
response = sources.merge_sources("2", "3", response_body)
assert (response.json() == "merged_source")
def test_create_deployment():
session = mock.mock_auth()
body = "deployment_json"
httpretty.register_uri(
httpretty.POST,
"http://test/deployments/v1/2/deployments",
status=200
)
deployments = alertlogic.services.Deployments(session)
response = deployments.create_deployment("2", body)
assert (response.status_code == 200)
def test_get_options():
session = mock.mock_auth()
response_body = [
{
"name": "predefined_security_subnet",
"scope": {
"provider_type": "aws",
"provider_id": "1234",
"vpc_id": "vpc-1"
},
"value": "subnet-1"
}
]
httpretty.register_uri(
httpretty.GET,
"http://test/otis/v2/2/options",
status=200,
content_type="text/json",
body=json.dumps(response_body)
)
otis = alertlogic.services.Otis(session)
result = otis.get_options("2")
assert (result.json()==response_body)
