  def testGetDiskTypes(self, mock_gce_api):
    """Test getting the Disk Type API object."""
    disk_types_mock = mock_gce_api.return_value.diskTypes.return_value
    disk_types_mock.get.return_value.execute.return_value = gcp_mocks.MOCK_DISK_TYPES
    disk_type_name = 'pd-standard'
    disk_type = gcp_mocks.FAKE_ANALYSIS_PROJECT.compute.GetDiskTypes(
        disk_type_name, 'us-central1-a')
    self.assertEqual(disk_type_name, disk_type.get('name'))
    disk_types_mock.get.assert_called()
Test getting the Disk Type API object.
tests/providers/gcp/internal/test_compute.py