def _maintenance_id_to_maintenance_view(self) -> Dict[str, MaintenanceView]:
if self._maintenance_id_to_maintenance_view_dict is None:
self._maintenance_id_to_maintenance_view_dict = {
mnt_id: MaintenanceView(
maintenance=self._maintenance_id_to_maintenance[mnt_id],
node_index_to_node_view={
ni: self._node_index_to_node_view[ni]
for ni in self._maintenance_id_to_node_indexes[mnt_id]
},
)
for mnt_id in self._maintenance_ids
}
return self._maintenance_id_to_maintenance_view_dict
def _maintenance_id_to_maintenance_view(self) -> Dict[str, MaintenanceView]:
if self._maintenance_id_to_maintenance_view_dict is None:
self._maintenance_id_to_maintenance_view_dict = {
mnt_id: MaintenanceView(
maintenance=self._maintenance_id_to_maintenance[mnt_id],
node_index_to_node_view={
ni: self._node_index_to_node_view[ni]
for ni in self._maintenance_id_to_node_indexes[mnt_id]
},
)
for mnt_id in self._maintenance_ids
}
# pyre-fixme[7]: Expected `Dict[str, MaintenanceView]` but got
# `Optional[Dict[str, MaintenanceView]]`.
return self._maintenance_id_to_maintenance_view_dict
async def test_no_timestamps(self):
async with MockAdminAPI() as client:
cv = await get_cluster_view(client)
await apply_maintenance(
client=client,
sequencer_nodes=[cv.get_node_view_by_node_index(0).node_id],
ttl=timedelta(days=2),
)
cv = await get_cluster_view(client)
mnt = list(cv.get_all_maintenances())[0]
mnt = mnt(created_on=None, expires_on=None)
mv = MaintenanceView(
maintenance=mnt,
node_index_to_node_view={0: cv.get_node_view(node_index=0)})
self.assertIsNone(mv.created_on)
self.assertIsNone(mv.expires_on)
self.assertIsNone(mv.expires_in)