async def test_freshness(coord: Coordinator, source: HelperTestSource,
dest: HelperTestSource, snapshot: Snapshot,
time: FakeTime):
source.setMax(2)
dest.setMax(2)
await coord.sync()
assert snapshot.getPurges() == {source.name(): False, dest.name(): False}
source.setMax(1)
dest.setMax(1)
await coord.sync()
assert snapshot.getPurges() == {source.name(): True, dest.name(): True}
dest.setMax(0)
await coord.sync()
assert snapshot.getPurges() == {source.name(): True, dest.name(): False}
source.setMax(0)
await coord.sync()
assert snapshot.getPurges() == {source.name(): False, dest.name(): False}
source.setMax(2)
dest.setMax(2)
time.advance(days=7)
await coord.sync()
assert len(coord.snapshots()) == 2
assert snapshot.getPurges() == {source.name(): True, dest.name(): True}
assert coord.snapshots()[1].getPurges() == {
source.name(): False,
dest.name(): False
}
# should refresh on delete
source.setMax(1)
dest.setMax(1)
await coord.delete([source.name()], snapshot.slug())
assert coord.snapshots()[0].getPurges() == {dest.name(): True}
assert coord.snapshots()[1].getPurges() == {
source.name(): True,
dest.name(): False
}
# should update on retain
await coord.retain({dest.name(): True}, snapshot.slug())
assert coord.snapshots()[0].getPurges() == {dest.name(): False}
assert coord.snapshots()[1].getPurges() == {
source.name(): True,
dest.name(): True
}
# should update on upload
await coord.uploadSnapshot(coord.snapshots()[0].slug())
assert coord.snapshots()[0].getPurges() == {
dest.name(): False,
source.name(): True
}
assert coord.snapshots()[1].getPurges() == {
source.name(): False,
dest.name(): True
}
def getSnapshotDetails(self, snapshot: Snapshot):
drive = snapshot.getSource(SOURCE_GOOGLE_DRIVE)
ha = snapshot.getSource(SOURCE_HA)
return {
'name': snapshot.name(),
'slug': snapshot.slug(),
'size': snapshot.sizeString(),
'status': snapshot.status(),
'date': self._time.toLocal(snapshot.date()).strftime("%c"),
'inDrive': drive is not None,
'inHA': ha is not None,
'isPending': ha is not None and type(ha) is PendingSnapshot,
'protected': snapshot.protected(),
'type': snapshot.snapshotType(),
'details': snapshot.details(),
'deleteNextDrive': snapshot.getPurges().get(SOURCE_GOOGLE_DRIVE) or False,
'deleteNextHa': snapshot.getPurges().get(SOURCE_HA) or False,
'driveRetain': drive.retained() if drive else False,
'haRetain': ha.retained() if ha else False
}
def getSnapshotDetails(self, snapshot: Snapshot):
ha = snapshot.getSource(SOURCE_HA)
sources = []
for source_key in snapshot.sources:
source: AbstractSnapshot = snapshot.sources[source_key]
sources.append({
'name':
source.name(),
'key':
source_key,
'size':
source.size(),
'retained':
source.retained(),
'delete_next':
snapshot.getPurges().get(source_key) or False
})
return {
'name':
snapshot.name(),
'slug':
snapshot.slug(),
'size':
snapshot.sizeString(),
'status':
snapshot.status(),
'date':
self._time.toLocal(snapshot.date()).strftime("%c"),
'isPending':
ha is not None and type(ha) is PendingSnapshot,
'protected':
snapshot.protected(),
'type':
snapshot.snapshotType(),
'details':
snapshot.details(),
'sources':
sources,
'uploadable':
snapshot.getSource(SOURCE_HA) is None
and len(snapshot.sources) > 0,
'restorable':
snapshot.getSource(SOURCE_HA) is not None,
}
def getSnapshotDetails(self, snapshot: Snapshot):
ha = snapshot.getSource(SOURCE_HA)
sources = []
for source_key in snapshot.sources:
source: AbstractSnapshot = snapshot.sources[source_key]
sources.append({
'name':
source.name(),
'key':
source_key,
'size':
source.size(),
'retained':
source.retained(),
'delete_next':
snapshot.getPurges().get(source_key) or False,
'slug':
snapshot.slug(),
'ignored':
source.ignore(),
})
return {
'name':
snapshot.name(),
'slug':
snapshot.slug(),
'size':
snapshot.sizeString(),
'status':
snapshot.status(),
'date':
self._time.toLocal(snapshot.date()).strftime("%c"),
'createdAt':
self._time.formatDelta(snapshot.date()),
'isPending':
ha is not None and type(ha) is PendingSnapshot,
'protected':
snapshot.protected(),
'type':
snapshot.snapshotType(),
'folders':
snapshot.details().get("folders", []),
'addons':
self.formatAddons(snapshot.details()),
'sources':
sources,
'haVersion':
snapshot.version(),
'uploadable':
snapshot.getSource(SOURCE_HA) is None
and len(snapshot.sources) > 0,
'restorable':
snapshot.getSource(SOURCE_HA) is not None,
'status_detail':
snapshot.getStatusDetail(),
'upload_info':
snapshot.getUploadInfo(self._time),
'ignored':
snapshot.ignore(),
'timestamp':
snapshot.date().timestamp(),
}