| 1438 | self.assertEqual(expected, quota) |
| 1439 | |
| 1440 | def test_get_asset_root_quota(self): |
| 1441 | cloud_api_resource = mock.MagicMock() |
| 1442 | with apitestcase.UsingCloudApi(cloud_api_resource=cloud_api_resource): |
| 1443 | fake_asset = { |
| 1444 | 'type': 'FOLDER', |
| 1445 | 'name': 'projects/test-proj/assets', |
| 1446 | 'quota': { |
| 1447 | 'assetCount': 123, |
| 1448 | 'maxAssetCount': 456, |
| 1449 | 'sizeBytes': 789, |
| 1450 | 'maxSizeBytes': 1001, |
| 1451 | }, |
| 1452 | } |
| 1453 | cloud_api_resource.projects().assets().get().execute.return_value = ( |
| 1454 | fake_asset |
| 1455 | ) |
| 1456 | |
| 1457 | quota = ee.data.getAssetRootQuota('projects/test-proj/assets') |
| 1458 | expected = { |
| 1459 | 'asset_count': {'usage': 123, 'limit': 456}, |
| 1460 | 'asset_size': {'usage': 789, 'limit': 1001}, |
| 1461 | } |
| 1462 | self.assertEqual(expected, quota) |
| 1463 | |
| 1464 | def test_get_asset_root_quota_not_root(self): |
| 1465 | cloud_api_resource = mock.MagicMock() |