def test_summaries_adds_uri(self) -> None:
col = self.naip_collection
col.stac_extensions = []
self.assertRaisesRegex(
pystac.ExtensionNotImplemented,
r"Could not find extension schema URI.*",
StorageExtension.summaries,
col,
False,
)
_ = StorageExtension.summaries(col, add_if_missing=True)
self.assertIn(StorageExtension.get_schema_uri(), col.stac_extensions)
StorageExtension.remove_from(col)
self.assertNotIn(StorageExtension.get_schema_uri(),
col.stac_extensions)
def test_tier(self) -> None:
col = self.naip_collection
col_dict = col.to_dict()
storage_summaries = StorageExtension.summaries(col)
# Get
self.assertEqual(storage_summaries.tier,
col_dict["summaries"]["storage:tier"])
# Set
new_tier_summary = [random.choice(ascii_letters)]
self.assertNotEqual(storage_summaries.tier, new_tier_summary)
storage_summaries.tier = new_tier_summary
self.assertEqual(storage_summaries.tier, new_tier_summary)
col_dict = col.to_dict()
self.assertEqual(col_dict["summaries"]["storage:tier"],
new_tier_summary)
def test_platform(self) -> None:
col = self.naip_collection
col_dict = col.to_dict()
storage_summaries = StorageExtension.summaries(col)
# Get
self.assertEqual(storage_summaries.platform,
col_dict["summaries"]["storage:platform"])
# Set
new_platform_summary = [random.choice([v for v in CloudPlatform])]
self.assertNotEqual(storage_summaries.platform, new_platform_summary)
storage_summaries.platform = new_platform_summary
self.assertEqual(storage_summaries.platform, new_platform_summary)
col_dict = col.to_dict()
self.assertEqual(col_dict["summaries"]["storage:platform"],
new_platform_summary)
def test_requester_pays(self) -> None:
col = self.naip_collection
col_dict = col.to_dict()
storage_summaries = StorageExtension.summaries(col)
# Get
self.assertEqual(
storage_summaries.requester_pays,
col_dict["summaries"]["storage:requester_pays"],
)
# Set
new_requester_pays_summary = [True]
self.assertNotEqual(storage_summaries.requester_pays,
new_requester_pays_summary)
storage_summaries.requester_pays = new_requester_pays_summary
self.assertEqual(storage_summaries.requester_pays,
new_requester_pays_summary)
col_dict = col.to_dict()
self.assertEqual(col_dict["summaries"]["storage:requester_pays"],
new_requester_pays_summary)