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)