def test_create_finding(self): # Setup Expected Response name = 'name3373707' parent_2 = 'parent21175163357' resource_name = 'resourceName979421212' category = 'category50511102' external_uri = 'externalUri-1385596168' expected_response = { 'name': name, 'parent': parent_2, 'resource_name': resource_name, 'category': category, 'external_uri': external_uri } expected_response = finding_pb2.Finding(**expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = securitycenter_v1beta1.SecurityCenterClient(channel=channel) # Setup Request parent = client.source_path('[ORGANIZATION]', '[SOURCE]') finding_id = 'findingId728776081' finding = {} response = client.create_finding(parent, finding_id, finding) assert expected_response == response assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.CreateFindingRequest( parent=parent, finding_id=finding_id, finding=finding) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_create_finding(self): # Setup Expected Response name = "name3373707" parent_2 = "parent21175163357" resource_name = "resourceName979421212" category = "category50511102" external_uri = "externalUri-1385596168" expected_response = { "name": name, "parent": parent_2, "resource_name": resource_name, "category": category, "external_uri": external_uri, } 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 = securitycenter_v1beta1.SecurityCenterClient() # Setup Request parent = client.source_path("[ORGANIZATION]", "[SOURCE]") finding_id = "findingId728776081" finding = {} response = client.create_finding(parent, finding_id, finding) assert expected_response == response assert len(channel.requests) == 1 expected_request = securitycenter_service_pb2.CreateFindingRequest( parent=parent, finding_id=finding_id, finding=finding ) actual_request = channel.requests[0][1] assert expected_request == actual_request