def test_list_finding_type_stats(self):
# Setup Expected Response
expected_response = {}
expected_response = web_security_scanner_pb2.ListFindingTypeStatsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
parent = client.scan_run_path("[PROJECT]", "[SCAN_CONFIG]",
"[SCAN_RUN]")
response = client.list_finding_type_stats(parent)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.ListFindingTypeStatsRequest(
parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_create_scan_config(self):
# Setup Expected Response
name = "name3373707"
display_name = "displayName1615086568"
max_qps = 844445913
expected_response = {
"name": name,
"display_name": display_name,
"max_qps": max_qps,
}
expected_response = scan_config_pb2.ScanConfig(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
parent = client.project_path("[PROJECT]")
scan_config = {}
response = client.create_scan_config(parent, scan_config)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.CreateScanConfigRequest(
parent=parent, scan_config=scan_config)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_list_scan_configs(self):
# Setup Expected Response
next_page_token = ""
scan_configs_element = {}
scan_configs = [scan_configs_element]
expected_response = {
"next_page_token": next_page_token,
"scan_configs": scan_configs,
}
expected_response = web_security_scanner_pb2.ListScanConfigsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
parent = client.project_path("[PROJECT]")
paged_list_response = client.list_scan_configs(parent)
resources = list(paged_list_response)
assert len(resources) == 1
assert expected_response.scan_configs[0] == resources[0]
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.ListScanConfigsRequest(
parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_list_findings(self):
# Setup Expected Response
next_page_token = ""
findings_element = {}
findings = [findings_element]
expected_response = {
"next_page_token": next_page_token,
"findings": findings
}
expected_response = web_security_scanner_pb2.ListFindingsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
parent = client.scan_run_path("[PROJECT]", "[SCAN_CONFIG]",
"[SCAN_RUN]")
filter_ = "filter-1274492040"
paged_list_response = client.list_findings(parent, filter_)
resources = list(paged_list_response)
assert len(resources) == 1
assert expected_response.findings[0] == resources[0]
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.ListFindingsRequest(
parent=parent, filter=filter_)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_list_crawled_urls(self):
# Setup Expected Response
next_page_token = ""
crawled_urls_element = {}
crawled_urls = [crawled_urls_element]
expected_response = {
"next_page_token": next_page_token,
"crawled_urls": crawled_urls,
}
expected_response = web_security_scanner_pb2.ListCrawledUrlsResponse(
**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
parent = client.scan_run_path("[PROJECT]", "[SCAN_CONFIG]",
"[SCAN_RUN]")
paged_list_response = client.list_crawled_urls(parent)
resources = list(paged_list_response)
assert len(resources) == 1
assert expected_response.crawled_urls[0] == resources[0]
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.ListCrawledUrlsRequest(
parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_stop_scan_run(self):
# Setup Expected Response
name_2 = "name2-1052831874"
urls_crawled_count = 1749797253
urls_tested_count = 1498664068
has_vulnerabilities = False
progress_percent = 2137894861
expected_response = {
"name": name_2,
"urls_crawled_count": urls_crawled_count,
"urls_tested_count": urls_tested_count,
"has_vulnerabilities": has_vulnerabilities,
"progress_percent": progress_percent,
}
expected_response = scan_run_pb2.ScanRun(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
name = client.scan_run_path("[PROJECT]", "[SCAN_CONFIG]", "[SCAN_RUN]")
response = client.stop_scan_run(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.StopScanRunRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_scan_config(self):
# Setup Expected Response
name_2 = "name2-1052831874"
display_name = "displayName1615086568"
max_qps = 844445913
expected_response = {
"name": name_2,
"display_name": display_name,
"max_qps": max_qps,
}
expected_response = scan_config_pb2.ScanConfig(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
name = client.scan_config_path("[PROJECT]", "[SCAN_CONFIG]")
response = client.get_scan_config(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.GetScanConfigRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_stop_scan_run_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup request
name = client.scan_run_path("[PROJECT]", "[SCAN_CONFIG]", "[SCAN_RUN]")
with pytest.raises(CustomException):
client.stop_scan_run(name)
def test_list_scan_runs_exception(self):
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup request
parent = client.scan_config_path("[PROJECT]", "[SCAN_CONFIG]")
paged_list_response = client.list_scan_runs(parent)
with pytest.raises(CustomException):
list(paged_list_response)
def test_update_scan_config_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup request
scan_config = {}
update_mask = {}
with pytest.raises(CustomException):
client.update_scan_config(scan_config, update_mask)
def test_create_scan_config_exception(self):
# Mock the API response
channel = ChannelStub(responses=[CustomException()])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup request
parent = client.project_path("[PROJECT]")
scan_config = {}
with pytest.raises(CustomException):
client.create_scan_config(parent, scan_config)
def test_delete_scan_config(self):
channel = ChannelStub()
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
name = client.scan_config_path("[PROJECT]", "[SCAN_CONFIG]")
client.delete_scan_config(name)
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.DeleteScanConfigRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_finding(self):
# Setup Expected Response
name_2 = "name2-1052831874"
finding_type = "findingType274496048"
http_method = "httpMethod820747384"
fuzzed_url = "fuzzedUrl-2120677666"
body = "body3029410"
description = "description-1724546052"
reproduction_url = "reproductionUrl-244934180"
frame_url = "frameUrl545464221"
final_url = "finalUrl355601190"
tracking_id = "trackingId1878901667"
expected_response = {
"name": name_2,
"finding_type": finding_type,
"http_method": http_method,
"fuzzed_url": fuzzed_url,
"body": body,
"description": description,
"reproduction_url": reproduction_url,
"frame_url": frame_url,
"final_url": final_url,
"tracking_id": tracking_id,
}
expected_response = finding_pb2.Finding(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = websecurityscanner_v1beta.WebSecurityScannerClient()
# Setup Request
name = client.finding_path("[PROJECT]", "[SCAN_CONFIG]", "[SCAN_RUN]",
"[FINDING]")
response = client.get_finding(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = web_security_scanner_pb2.GetFindingRequest(
name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
