earthengine-api 1.5.13rc0__py3-none-any.whl → 1.7.4__py3-none-any.whl

ee/tests/data_test.py

@@ -2,7 +2,7 @@
 """Test for the ee.data module."""
 
 import json
-from typing import Any, Optional
+from typing import Any
 from unittest import mock
 
 import googleapiclient
@@ -12,6 +12,7 @@ import requests
 import unittest
 import ee
 from ee import _cloud_api_utils
+from ee import _state
 from ee import apitestcase
 from ee import featurecollection
 from ee import image
@@ -27,7 +28,7 @@ def NotFoundError() -> googleapiclient.errors.HttpError:
 
 
 def NewFolderAsset(
-    name: str, quota: Optional[dict[str, int]] = None
+    name: str, quota: dict[str, int] | None = None
 ) -> dict[str, Any]:
   return {
       'type': 'FOLDER',
@@ -49,22 +50,71 @@ class DataTest(unittest.TestCase):
 
   def tearDown(self):
     super().tearDown()
+    ee.data.reset()
     mock.patch.stopall()
 
-  def testIsInitialized(self):
+  def test_is_initialized(self):
     self.assertFalse(ee.data.is_initialized())
     with apitestcase.UsingCloudApi():
       self.assertTrue(ee.data.is_initialized())
 
-  def testSetMaxRetries_badValues(self):
+  @mock.patch.object(ee.data, '_install_cloud_api_resource', return_value=None)
+  def test_initialize(self, mock_install_cloud_api_resource):
+    ee.data.initialize()
+
+    self.assertTrue(ee.data.is_initialized())
+    mock_install_cloud_api_resource.assert_called_once()
+
+  @mock.patch.object(ee.data, '_install_cloud_api_resource', return_value=None)
+  def test_initialize_with_project(
+      self, unused_mock_install_cloud_api_resource
+  ):
+    ee.data.initialize(project='my-project')
+
+    self.assertTrue(ee.data.is_initialized())
+    self.assertEqual(
+        _state.get_state().cloud_api_user_project, 'my-project'
+    )
+
+  @mock.patch.object(ee.data, '_install_cloud_api_resource', return_value=None)
+  def test_initialize_with_no_project(
+      self, unused_mock_install_cloud_api_resource
+  ):
+    ee.data.initialize()
+
+    self.assertTrue(ee.data.is_initialized())
+    self.assertEqual(
+        _state.get_state().cloud_api_user_project, 'earthengine-legacy'
+    )
+
+  @mock.patch.object(ee.data, '_install_cloud_api_resource', return_value=None)
+  def test_initialize_with_credentials(
+      self, unused_mock_install_cloud_api_resource
+  ):
+    creds = mock.MagicMock()
+    ee.data.initialize(credentials=creds)
+
+    self.assertTrue(ee.data.is_initialized())
+    self.assertEqual(creds, _state.get_state().credentials)
+
+  @mock.patch.object(ee.data, '_install_cloud_api_resource', return_value=None)
+  def test_initialize_with_cloud_api_key(
+      self, unused_mock_install_cloud_api_resource
+  ):
+    cloud_api_key = 'a cloud api key'
+    ee.data.initialize(cloud_api_key=cloud_api_key)
+
+    self.assertTrue(ee.data.is_initialized())
+    self.assertEqual(cloud_api_key, _state.get_state().cloud_api_key)
+
+  def test_set_max_retries_bad_values(self):
     with self.assertRaises(ValueError):
       ee.data.setMaxRetries(-1)
     with self.assertRaises(ValueError):
       ee.data.setMaxRetries(100)
 
-  def testSetMaxRetries(self):
+  def test_set_max_retries(self):
     mock_result = {'result': 5}
-    original_max_retries = ee.data._max_retries
     ee.data.setMaxRetries(3)
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
@@ -79,9 +129,48 @@ class DataTest(unittest.TestCase):
           .compute()
           .execute.call_args.kwargs['num_retries'],
       )
-    ee.data._max_retries = original_max_retries
 
-  def testListOperations(self):
+  def test_set_cloud_api_key(self):
+    cloud_api_key = 'a cloud api key'
+    with mock.patch.object(
+        ee.data, '_install_cloud_api_resource', return_value=None
+    ) as mock_install_cloud_api_resource:
+      ee.data.setCloudApiKey(cloud_api_key)
+      self.assertEqual(cloud_api_key, _state.get_state().cloud_api_key)
+      mock_install_cloud_api_resource.assert_called_once()
+
+  def test_set_deadline(self):
+    deadline_ms = 12345
+    with mock.patch.object(
+        ee.data, '_install_cloud_api_resource', return_value=None
+    ) as mock_install_cloud_api_resource:
+      ee.data.setDeadline(deadline_ms)
+      self.assertEqual(deadline_ms, _state.get_state().deadline_ms)
+      mock_install_cloud_api_resource.assert_called_once()
+
+  def test_get_set_user_agent(self):
+    self.assertIsNone(ee.data.getUserAgent())
+    user_agent = 'user-agent'
+    ee.data.setUserAgent(user_agent)
+    self.assertEqual(user_agent, ee.data.getUserAgent())
+
+  def test_authorize_http_no_credentials(self):
+    self.assertIsNone(ee.data._get_state().credentials)
+    http = mock.MagicMock()
+    self.assertEqual(http, ee.data.authorizeHttp(http))
+
+  def test_authorize_http_with_credentials(self):
+    creds = mock.MagicMock()
+    ee.data._get_state().credentials = creds
+    http = mock.MagicMock()
+    with mock.patch.object(
+        ee.data.google_auth_httplib2, 'AuthorizedHttp'
+    ) as mock_authorized_http:
+      result = ee.data.authorizeHttp(http)
+      self.assertEqual(mock_authorized_http.return_value, result)
+      mock_authorized_http.assert_called_once_with(creds)
+
+  def test_list_operations(self):
     mock_http = mock.MagicMock(httplib2.Http)
     # Return in three groups.
     mock_http.request.side_effect = [
@@ -101,14 +190,120 @@ class DataTest(unittest.TestCase):
           'name': 'name3'
       }], ee.data.listOperations())
 
-  def testListOperationsEmptyList(self):
+  def test_list_operations_empty_list(self):
     # Empty lists don't appear at all in the result.
     mock_http = mock.MagicMock(httplib2.Http)
     mock_http.request.return_value = (httplib2.Response({'status': 200}), b'{}')
     with apitestcase.UsingCloudApi(mock_http=mock_http):
       self.assertEqual([], ee.data.listOperations())
 
-  def testCreateAsset(self):
+  def test_get_operation(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      name = 'projects/test-project/operations/foo'
+      cloud_api_resource.projects().operations().get.execute.return_value = {
+          'name': name,
+          'done': False,
+      }
+      ee.data.getOperation(name)
+      cloud_api_resource.projects().operations().get.assert_called_once_with(
+          name=name
+      )
+
+  def test_get_task_status(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      cloud_api_resource.projects().operations().get.return_value.execute.return_value = {
+          'name': 'projects/earthengine-legacy/operations/foo',
+          'done': False,
+          'metadata': {'state': 'RUNNING'},
+      }
+      result = ee.data.getTaskStatus('foo')
+      cloud_api_resource.projects().operations().get.assert_called_once_with(
+          name='projects/earthengine-legacy/operations/foo'
+      )
+      self.assertEqual(
+          result,
+          [{
+              'id': 'foo',
+              'state': 'RUNNING',
+              'name': 'projects/earthengine-legacy/operations/foo',
+          }],
+      )
+
+  def test_get_task_status_with_not_found(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      cloud_api_resource.projects().operations().get.return_value.execute.side_effect = [
+          {
+              'name': 'projects/earthengine-legacy/operations/foo',
+              'done': False,
+              'metadata': {'state': 'RUNNING'},
+          },
+          NotFoundError(),
+          {
+              'name': 'projects/earthengine-legacy/operations/bar',
+              'done': True,
+              'metadata': {'state': 'SUCCEEDED'},
+          },
+      ]
+      result = ee.data.getTaskStatus(['foo', 'missing', 'bar'])
+      cloud_api_resource.projects().operations().get.assert_has_calls([
+          mock.call(name='projects/earthengine-legacy/operations/foo'),
+          mock.call().execute(num_retries=5),
+          mock.call(name='projects/earthengine-legacy/operations/missing'),
+          mock.call().execute(num_retries=5),
+          mock.call(name='projects/earthengine-legacy/operations/bar'),
+          mock.call().execute(num_retries=5),
+      ])
+      self.assertEqual(
+          3,
+          cloud_api_resource.projects()
+          .operations()
+          .get.return_value.execute.call_count,
+      )
+      self.assertEqual(
+          result,
+          [
+              {
+                  'id': 'foo',
+                  'state': 'RUNNING',
+                  'name': 'projects/earthengine-legacy/operations/foo',
+              },
+              {
+                  'id': 'missing',
+                  'state': 'UNKNOWN',
+                  'name': 'projects/earthengine-legacy/operations/missing',
+              },
+              {
+                  'id': 'bar',
+                  'state': 'COMPLETED',
+                  'name': 'projects/earthengine-legacy/operations/bar',
+              },
+          ],
+      )
+
+  def test_cancel_operation(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      cancel_mock = cloud_api_resource.projects().operations().cancel
+      cancel_mock.execute.return_value = {}
+      ee.data.cancelOperation('projects/test-project/operations/foo')
+      cancel_mock.assert_called_once_with(
+          name='projects/test-project/operations/foo', body={}
+      )
+
+  def test_cancel_task(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      cancel_mock = cloud_api_resource.projects().operations().cancel
+      cancel_mock.execute.return_value = {}
+      ee.data.cancelTask('foo')
+      cancel_mock.assert_called_once_with(
+          name='projects/earthengine-legacy/operations/foo', body={}
+      )
+
+  def test_create_asset(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {
@@ -129,7 +324,7 @@ class DataTest(unittest.TestCase):
       asset = mock_create_asset.call_args.kwargs['body']
       self.assertEqual(asset, {'type': 'FOLDER'})
 
-  def testCreateAssetWithV1AlphaParams(self):
+  def test_create_asset_with_v1alpha_params(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {
@@ -186,7 +381,7 @@ class DataTest(unittest.TestCase):
           {'uris': ['gs://my-bucket/path']},
       )
 
-  def testCreateFolder(self):
+  def test_create_folder(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {
@@ -207,7 +402,15 @@ class DataTest(unittest.TestCase):
       asset = mock_create_asset.call_args.kwargs['body']
       self.assertEqual(asset, {'type': 'FOLDER'})
 
-  def testCreateAssets(self):
+  @mock.patch.object(ee.data, 'createAsset')
+  def test_create_asset_home(self, mock_create_asset):
+    ee.data.createAssetHome('users/test')
+    mock_create_asset.assert_called_once_with({
+        'name': 'projects/earthengine-legacy/assets/users/test',
+        'type': 'FOLDER',
+    })
+
+  def test_create_assets(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       asset_name = 'projects/some-project/assets/some-asset'
@@ -223,14 +426,14 @@ class DataTest(unittest.TestCase):
           prettyPrint=False,
       )
 
-  def testCreateAssets_empty(self):
+  def test_create_assets_empty(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       ee.data.create_assets([], 'FOLDER', False)
       mock_create_asset = cloud_api_resource.projects().assets().create
       mock_create_asset.assert_not_called()
 
-  def testCreateAssets_noOpIfAssetExists(self):
+  def test_create_assets_no_op_if_asset_exists(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       asset_name = 'projects/some-project/assets/some-asset'
@@ -241,7 +444,7 @@ class DataTest(unittest.TestCase):
       mock_create_asset = cloud_api_resource.projects().assets().create
       mock_create_asset.assert_not_called()
 
-  def testCreateAssets_withParents(self):
+  def test_create_assets_with_parents(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       asset_name = 'projects/some-project/assets/foo/bar'
@@ -267,7 +470,7 @@ class DataTest(unittest.TestCase):
           mock.call().execute(num_retries=5),
       ])
 
-  def testStartIngestion(self):
+  def test_start_ingestion(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'name': 'operations/ingestion', 'done': False}
@@ -297,7 +500,69 @@ class DataTest(unittest.TestCase):
       )
       self.assertTrue(import_args['overwrite'])
 
-  def testSetAssetProperties(self):
+  def test_start_table_ingestion(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      mock_result = {'name': 'operations/ingestion', 'done': False}
+      cloud_api_resource.projects().table().import_.return_value.execute.return_value = (
+          mock_result
+      )
+      params = {
+          'id': 'users/test/table',
+          'sources': [{'uris': ['gs://bucket/file.shp'], 'charset': 'UTF-8'}],
+      }
+      result = ee.data.startTableIngestion('request_id', params, True)
+      self.assertEqual(result['id'], 'ingestion')
+      self.assertEqual(result['name'], 'operations/ingestion')
+
+      mock_import = cloud_api_resource.projects().table().import_
+      mock_import.assert_called_once()
+      call_kwargs = mock_import.call_args.kwargs
+      self.assertEqual(call_kwargs['project'], 'projects/earthengine-legacy')
+      body = call_kwargs['body']
+      self.assertEqual(
+          body['tableManifest'],
+          {
+              'name': 'projects/earthengine-legacy/assets/users/test/table',
+              'sources': [
+                  {'uris': ['gs://bucket/file.shp'], 'charset': 'UTF-8'}
+              ],
+          },
+      )
+      self.assertEqual(body['requestId'], 'request_id')
+      self.assertTrue(body['overwrite'])
+
+  def test_start_external_image_ingestion(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      cloud_api_resource.projects().image().importExternal.return_value.execute.return_value = (
+          {}
+      )
+      manifest = {
+          'id': 'users/test/image',
+          'tilesets': [{'sources': [{'uris': ['gs://bucket/file.tif']}]}],
+      }
+      result = ee.data.startExternalImageIngestion(manifest, True)
+      expected_name = 'projects/earthengine-legacy/assets/users/test/image'
+      self.assertEqual(result['name'], expected_name)
+
+      mock_import_external = (
+          cloud_api_resource.projects().image().importExternal
+      )
+      mock_import_external.assert_called_once()
+      call_kwargs = mock_import_external.call_args.kwargs
+      self.assertEqual(call_kwargs['project'], 'projects/earthengine-legacy')
+      body = call_kwargs['body']
+      self.assertEqual(
+          body['imageManifest'],
+          {
+              'name': expected_name,
+              'tilesets': [{'sources': [{'uris': ['gs://bucket/file.tif']}]}],
+          },
+      )
+      self.assertTrue(body['overwrite'])
+
+  def test_set_asset_properties(self):
     mock_http = mock.MagicMock(httplib2.Http)
     with apitestcase.UsingCloudApi(mock_http=mock_http), mock.patch.object(
         ee.data, 'updateAsset', autospec=True) as mock_update_asset:
@@ -315,10 +580,22 @@ class DataTest(unittest.TestCase):
       update_mask = mock_update_asset.call_args[0][2]
       self.assertSetEqual(
           set(update_mask),
-          set(['properties.\"mYPropErTy\"',
-               'properties.\"system:time_start\"']))
+          {'properties.\"mYPropErTy\"',
+               'properties.\"system:time_start\"'})
+
+  def test_update_asset(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_id = 'users/test/asset'
+      asset = {'properties': {'foo': 'bar'}}
+      update_mask = ['properties.foo']
+      ee.data.updateAsset(asset_id, asset, update_mask)
+      cloud_api_resource.projects().assets().patch.assert_called_once_with(
+          name='projects/earthengine-legacy/assets/users/test/asset',
+          body={'updateMask': {'paths': update_mask}, 'asset': asset},
+      )
 
-  def testListAssets(self):
+  def test_list_assets(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'assets': [{'path': 'id1', 'type': 'type1'}]}
@@ -330,7 +607,7 @@ class DataTest(unittest.TestCase):
       ).execute.assert_called_once()
       self.assertEqual(mock_result, actual_result)
 
-  def testListAssetsWithPageSize(self):
+  def test_list_assets_with_page_size(self):
     mock_http = mock.MagicMock(httplib2.Http)
     ok_resp = httplib2.Response({'status': 200})
     page = (
@@ -347,7 +624,7 @@ class DataTest(unittest.TestCase):
       }
       self.assertEqual(expected_result, actual_result)
 
-  def testListAssetsMultiplePages(self):
+  def test_list_assets_multiple_pages(self):
     mock_http = mock.MagicMock(httplib2.Http)
     ok_resp = httplib2.Response({'status': 200})
     page1 = (
@@ -373,7 +650,7 @@ class DataTest(unittest.TestCase):
       }
       self.assertEqual(expected_result, actual_result)
 
-  def testListImages(self):
+  def test_list_images(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'assets': [{'path': 'id1', 'type': 'type1'}]}
@@ -388,7 +665,7 @@ class DataTest(unittest.TestCase):
           'type': 'type1'
       }]}, actual_result)
 
-  def testListImagesWithPageSize(self):
+  def test_list_images_with_page_size(self):
     mock_http = mock.MagicMock(httplib2.Http)
     ok_resp = httplib2.Response({'status': 200})
     page = (
@@ -405,7 +682,7 @@ class DataTest(unittest.TestCase):
       }
       self.assertEqual(expected_result, actual_result)
 
-  def testListImagesMultiplePages(self):
+  def test_list_images_multiple_pages(self):
     mock_http = mock.MagicMock(httplib2.Http)
     ok_resp = httplib2.Response({'status': 200})
     page1 = (
@@ -431,7 +708,7 @@ class DataTest(unittest.TestCase):
       }
       self.assertEqual(expected_result, actual_result)
 
-  def testListBuckets(self):
+  def test_list_buckets(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'assets': [{'name': 'id1', 'type': 'FOLDER'}]}
@@ -441,7 +718,17 @@ class DataTest(unittest.TestCase):
     cloud_api_resource.projects().listAssets().execute.assert_called_once()
     self.assertEqual(mock_result, actual_result)
 
-  def testSimpleGetListViaCloudApi(self):
+  def test_get_asset_roots(self):
+    with mock.patch.object(
+        ee.data,
+        'listBuckets',
+        return_value={'assets': [{'name': 'id1', 'type': 'FOLDER'}]},
+    ) as mock_list_buckets:
+      result = ee.data.getAssetRoots()
+      mock_list_buckets.assert_called_once()
+      self.assertEqual([{'id': 'id1', 'type': 'Folder'}], result)
+
+  def test_simple_get_list_via_cloud_api(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'assets': [{'name': 'id1', 'type': 'IMAGE_COLLECTION'}]}
@@ -458,7 +745,7 @@ class DataTest(unittest.TestCase):
           **expected_params)
       self.assertEqual(expected_result, actual_result)
 
-  def testGetListAssetRootViaCloudApi(self):
+  def test_get_list_asset_root_via_cloud_api(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'assets': [{'name': 'id1', 'type': 'IMAGE_COLLECTION'}]}
@@ -478,7 +765,7 @@ class DataTest(unittest.TestCase):
           **expected_params)
       self.assertEqual(expected_result, actual_result)
 
-  def testGetListAssetRootViaCloudApiNoSlash(self):
+  def test_get_list_asset_root_via_cloud_api_no_slash(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'assets': [{'name': 'id1', 'type': 'IMAGE_COLLECTION'}]}
@@ -498,7 +785,7 @@ class DataTest(unittest.TestCase):
           **expected_params)
       self.assertEqual(expected_result, actual_result)
 
-  def testComplexGetListViaCloudApi(self):
+  def test_complex_get_list_via_cloud_api(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {
@@ -527,7 +814,7 @@ class DataTest(unittest.TestCase):
           **expected_params)
       self.assertEqual(expected_result, actual_result)
 
-  def testGetMapId(self):
+  def test_get_map_id(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {
@@ -544,7 +831,7 @@ class DataTest(unittest.TestCase):
       self.assertEqual('', actual_result['token'])
       self.assertIsInstance(actual_result['tile_fetcher'], ee.data.TileFetcher)
 
-  def testGetMapId_withWorkloadTag(self):
+  def test_get_map_id_with_workload_tag(self):
     with ee.data.workloadTagContext('mapid-tag'):
       cloud_api_resource = mock.MagicMock()
       with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
@@ -561,7 +848,7 @@ class DataTest(unittest.TestCase):
             cloud_api_resource.projects().maps().create.call_args_list[1]
             .kwargs['workloadTag'])
 
-  def testGetDownloadId(self):
+  def test_get_download_id(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'name': 'projects/earthengine-legacy/thumbnails/DOCID'}
@@ -579,7 +866,7 @@ class DataTest(unittest.TestCase):
               'token': ''
           }, actual_result)
 
-  def testGetDownloadId_withWorkloadTag(self):
+  def test_get_download_id_with_workload_tag(self):
     with ee.data.workloadTagContext('downloadid-tag'):
       cloud_api_resource = mock.MagicMock()
       with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
@@ -595,7 +882,7 @@ class DataTest(unittest.TestCase):
             cloud_api_resource.projects().thumbnails().create.call_args
             .kwargs['workloadTag'])
 
-  def testGetDownloadId_withBandList(self):
+  def test_get_download_id_with_band_list(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'name': 'projects/earthengine-legacy/thumbnails/DOCID'}
@@ -614,14 +901,14 @@ class DataTest(unittest.TestCase):
               'token': ''
           }, actual_result)
 
-  def testGetDownloadId_withImageID(self):
+  def test_get_download_id_with_image_id(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       with self.assertRaisesRegex(ee.ee_exception.EEException,
                                   '^Image ID string is not supported.'):
         ee.data.getDownloadId({'id': 'my-image', 'name': 'dummy'})
 
-  def testGetDownloadId_withSerializedImage(self):
+  def test_get_download_id_with_serialized_image(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       with self.assertRaisesRegex(ee.ee_exception.EEException,
@@ -631,7 +918,7 @@ class DataTest(unittest.TestCase):
             'name': 'dummy'
         })
 
-  def testGetThumbId(self):
+  def test_get_thumb_id(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'name': 'projects/earthengine-legacy/thumbnails/DOCID'}
@@ -649,7 +936,7 @@ class DataTest(unittest.TestCase):
               'token': ''
           }, actual_result)
 
-  def testGetThumbId_withWorkloadTag(self):
+  def test_get_thumb_id_with_workload_tag(self):
     with ee.data.workloadTagContext('thumbid-tag'):
       cloud_api_resource = mock.MagicMock()
       with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
@@ -662,7 +949,7 @@ class DataTest(unittest.TestCase):
             cloud_api_resource.projects().thumbnails().create.call_args
             .kwargs['workloadTag'])
 
-  def testGetTableDownloadId(self):
+  def test_get_table_download_id(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'name': 'projects/earthengine-legacy/table/DOCID'}
@@ -680,7 +967,7 @@ class DataTest(unittest.TestCase):
               'token': ''
           }, actual_result)
 
-  def testGetTableDownloadId_withWorkloadTag(self):
+  def test_get_table_download_id_with_workload_tag(self):
     with ee.data.workloadTagContext('tableid-tag'):
       cloud_api_resource = mock.MagicMock()
       with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
@@ -696,7 +983,7 @@ class DataTest(unittest.TestCase):
             cloud_api_resource.projects().tables().create.call_args
             .kwargs['workloadTag'])
 
-  def testCloudProfilingEnabled(self):
+  def test_cloud_profiling_enabled(self):
     seen = []
 
     def ProfileHook(profile_id):
@@ -707,11 +994,11 @@ class DataTest(unittest.TestCase):
         ee.data.listImages({'parent': 'projects/earthengine-public/assets/q'})
     self.assertEqual(['someProfileId'], seen)
 
-  def testCloudProfilingDisabled(self):
+  def test_cloud_profiling_disabled(self):
     with apitestcase.UsingCloudApi(), DoCloudProfileStubHttp(self, False):
       ee.data.listImages({'parent': 'projects/earthengine-public/assets/q'})
 
-  def testCloudErrorTranslation(self):
+  def test_cloud_error_translation(self):
     mock_http = mock.MagicMock(httplib2.Http)
     mock_http.request.return_value = (httplib2.Response({'status': 400}),
                                       b'{"error": {"message": "errorly"} }')
@@ -719,7 +1006,7 @@ class DataTest(unittest.TestCase):
       with self.assertRaisesRegex(ee.ee_exception.EEException, '^errorly$'):
         ee.data.listImages({'parent': 'projects/earthengine-public/assets/q'})
 
-  def testListFeatures(self):
+  def test_list_features(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {
@@ -746,9 +1033,53 @@ class DataTest(unittest.TestCase):
       ).execute.assert_called_once()
       self.assertEqual(mock_result, actual_result)
 
-  @mock.patch.object(ee.data, '_tile_base_url', new='base_url')
-  def testGetFeatureViewTilesKey(self):
+  def test_get_pixels(self):
+    cloud_api_resource_raw = mock.MagicMock()
+    with apitestcase.UsingCloudApi(
+        cloud_api_resource_raw=cloud_api_resource_raw
+    ):
+      assets = cloud_api_resource_raw.projects().assets()
+      mock_result = b'pixel data'
+      assets.getPixels.return_value.execute.return_value = mock_result
+      asset_id = 'users/foo/bar'
+      params = {'assetId': asset_id}
+      result = ee.data.getPixels(params)
+      self.assertEqual(mock_result, result)
+      assets.getPixels.assert_called_once_with(
+          name='projects/earthengine-legacy/assets/users/foo/bar',
+          body={'fileFormat': 'AUTO_JPEG_PNG'},
+      )
+
+  def test_compute_pixels(self):
+    cloud_api_resource_raw = mock.MagicMock()
+    with apitestcase.UsingCloudApi(
+        cloud_api_resource_raw=cloud_api_resource_raw
+    ):
+      mock_result = b'pixel data'
+      (
+          cloud_api_resource_raw.projects()
+          .image()
+          .computePixels.return_value.execute.return_value
+      ) = mock_result
+      expression = ee.Image(1)
+      params = {'expression': expression}
+      result = ee.data.computePixels(params)
+      self.assertEqual(mock_result, result)
+      (
+          cloud_api_resource_raw.projects()
+          .image()
+          .computePixels.assert_called_once_with(
+              project='projects/earthengine-legacy',
+              body={
+                  'expression': ee.serializer.encode(expression),
+                  'fileFormat': 'AUTO_JPEG_PNG',
+              },
+          )
+      )
+
+  def test_get_feature_view_tiles_key(self):
     cloud_api_resource = mock.MagicMock()
+    _state.get_state().tile_base_url = 'base_url'
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_name = 'projects/projectfoo/featureView/tiles-key-foo'
       mock_result = {'name': mock_name}
@@ -769,7 +1100,7 @@ class DataTest(unittest.TestCase):
           f'base_url/{_cloud_api_utils.VERSION}/{mock_name}/tiles/7/5/6',
           actual_result['formatTileUrl'](5, 6, 7))
 
-  def testGetProjectConfig(self) -> None:
+  def test_get_project_config(self) -> None:
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'fake-project-config-value': 1}
@@ -780,20 +1111,43 @@ class DataTest(unittest.TestCase):
       cloud_api_resource.projects().getConfig().execute.assert_called_once()
       self.assertEqual(mock_result, actual_result)
 
-  def testUpdateProjectConfig(self) -> None:
+  def test_update_project_config(self) -> None:
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       mock_result = {'fake-project-config-value': 1}
       cloud_api_resource.projects().updateConfig().execute.return_value = (
           mock_result
       )
+      project_config = {'maxConcurrentExports': 2}
       actual_result = ee.data.updateProjectConfig(
-          {'maxConcurrentExports': 2}, ['max_concurrent_exports']
+          project_config, ['max_concurrent_exports']
+      )
+      cloud_api_resource.projects().updateConfig.assert_called_with(
+          name='projects/earthengine-legacy/config',
+          body=project_config,
+          updateMask='max_concurrent_exports',
       )
       cloud_api_resource.projects().updateConfig().execute.assert_called_once()
       self.assertEqual(mock_result, actual_result)
 
-  def testWorkloadTag(self):
+  def test_update_project_config_no_mask(self) -> None:
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      mock_result = {'fake-project-config-value': 1}
+      cloud_api_resource.projects().updateConfig().execute.return_value = (
+          mock_result
+      )
+      project_config = {'maxConcurrentExports': 2}
+      actual_result = ee.data.updateProjectConfig(project_config)
+      cloud_api_resource.projects().updateConfig.assert_called_with(
+          name='projects/earthengine-legacy/config',
+          body=project_config,
+          updateMask='max_concurrent_exports',
+      )
+      cloud_api_resource.projects().updateConfig().execute.assert_called_once()
+      self.assertEqual(mock_result, actual_result)
+
+  def test_workload_tag(self):
     self.assertEqual('', ee.data.getWorkloadTag())
     ee.data.setDefaultWorkloadTag(None)
     self.assertEqual('', ee.data.getWorkloadTag())
@@ -851,13 +1205,13 @@ class DataTest(unittest.TestCase):
     ee.data.resetWorkloadTag(True)
     self.assertEqual('', ee.data.getWorkloadTag())
 
-  def testResetWorkloadTagOptParams(self):
+  def test_reset_workload_tag_opt_params(self):
     ee.data.setDefaultWorkloadTag('reset-me')
     self.assertEqual('reset-me', ee.data.getWorkloadTag())
     ee.data.resetWorkloadTag(opt_resetDefault=True)
     self.assertEqual('', ee.data.getWorkloadTag())
 
-  def testGetAssetRootQuota_V1Alpha(self):
+  def test_get_asset_root_quota_v1alpha(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       fake_asset = {
@@ -881,7 +1235,7 @@ class DataTest(unittest.TestCase):
       }
       self.assertEqual(expected, quota)
 
-  def testGetAssetRootQuota(self):
+  def test_get_asset_root_quota(self):
     cloud_api_resource = mock.MagicMock()
     with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
       fake_asset = {
@@ -905,6 +1259,81 @@ class DataTest(unittest.TestCase):
       }
       self.assertEqual(expected, quota)
 
+  def test_get_asset_root_quota_not_root(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_id = 'users/test/asset'
+      fake_asset = {
+          'type': 'IMAGE',
+          'name': 'projects/earthengine-legacy/assets/users/test/asset',
+      }
+      cloud_api_resource.projects().assets().get().execute.return_value = (
+          fake_asset
+      )
+      with self.assertRaisesRegex(
+          ee.ee_exception.EEException, f'{asset_id} is not a root folder.'
+      ):
+        ee.data.getAssetRootQuota(asset_id)
+
+  def test_get_iam_policy(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_id = 'users/test/asset'
+      ee.data.getIamPolicy(asset_id)
+      cloud_api_resource.projects().assets().getIamPolicy.assert_called_once_with(
+          resource='projects/earthengine-legacy/assets/users/test/asset',
+          body={},
+          prettyPrint=False,
+      )
+
+  def test_get_asset_acl(self):
+    asset_id = 'users/test/asset'
+    policy = {'bindings': [{'role': 'roles/viewer', 'members': ['allUsers']}]}
+    acl = {'readers': ['allUsers']}
+    with mock.patch.object(
+        ee.data, 'getIamPolicy', return_value=policy
+    ) as mock_get_iam_policy, mock.patch.object(
+        ee.data._cloud_api_utils, 'convert_iam_policy_to_acl', return_value=acl
+    ) as mock_convert:
+      result = ee.data.getAssetAcl(asset_id)
+      mock_get_iam_policy.assert_called_once_with(asset_id)
+      mock_convert.assert_called_once_with(policy)
+      self.assertEqual(acl, result)
+
+  def test_set_asset_acl(self):
+    asset_id = 'users/test/asset'
+    acl_update_dict = {'readers': ['allUsers']}
+    acl_update_str = '{"readers": ["allUsers"]}'
+    policy = {'bindings': [{'role': 'roles/viewer', 'members': ['allUsers']}]}
+    with mock.patch.object(
+        ee.data._cloud_api_utils,
+        'convert_acl_to_iam_policy',
+        return_value=policy,
+    ) as mock_convert, mock.patch.object(
+        ee.data, 'setIamPolicy'
+    ) as mock_set_iam_policy:
+      ee.data.setAssetAcl(asset_id, acl_update_dict)
+      mock_convert.assert_called_once_with(acl_update_dict)
+      mock_set_iam_policy.assert_called_once_with(asset_id, policy)
+
+      mock_convert.reset_mock()
+      mock_set_iam_policy.reset_mock()
+      ee.data.setAssetAcl(asset_id, acl_update_str)
+      mock_convert.assert_called_once_with(acl_update_dict)
+      mock_set_iam_policy.assert_called_once_with(asset_id, policy)
+
+  def test_set_iam_policy(self):
+    cloud_api_resource = mock.MagicMock()
+    with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource):
+      asset_id = 'users/test/asset'
+      policy = {'bindings': [{'role': 'roles/viewer', 'members': ['allUsers']}]}
+      ee.data.setIamPolicy(asset_id, policy)
+      cloud_api_resource.projects().assets().setIamPolicy.assert_called_once_with(
+          resource='projects/earthengine-legacy/assets/users/test/asset',
+          body={'policy': policy},
+          prettyPrint=False,
+      )
+
 
 def DoCloudProfileStubHttp(test, expect_profiling):