Changeset View
Changeset View
Standalone View
Standalone View
swh/scheduler/tests/test_scheduler.py
Show First 20 Lines • Show All 665 Lines • ▼ Show 20 Lines | def test_get_listers(self, swh_scheduler): | ||||
assert swh_scheduler.get_listers() == [] | assert swh_scheduler.get_listers() == [] | ||||
db_listers = [] | db_listers = [] | ||||
for lister_args in LISTERS: | for lister_args in LISTERS: | ||||
db_listers.append(swh_scheduler.get_or_create_lister(**lister_args)) | db_listers.append(swh_scheduler.get_or_create_lister(**lister_args)) | ||||
assert swh_scheduler.get_listers() == db_listers | assert swh_scheduler.get_listers() == db_listers | ||||
def test_get_listers_by_id(self, swh_scheduler): | |||||
assert swh_scheduler.get_listers_by_id([str(uuid.uuid4())]) == [] | |||||
db_listers = [] | |||||
for lister_args in LISTERS: | |||||
db_listers.append(swh_scheduler.get_or_create_lister(**lister_args)) | |||||
id0 = db_listers[0].id | |||||
id1 = db_listers[1].id | |||||
assert swh_scheduler.get_listers_by_id([id0]) == [db_listers[0]] | |||||
assert swh_scheduler.get_listers_by_id([id1]) == [db_listers[1]] | |||||
assert swh_scheduler.get_listers_by_id([id0, id1]) == [ | |||||
db_listers[0], | |||||
db_listers[1], | |||||
] | |||||
assert swh_scheduler.get_listers_by_id([id0, str(uuid.uuid4())]) == [ | |||||
db_listers[0] | |||||
] | |||||
def test_update_lister(self, swh_scheduler, stored_lister): | def test_update_lister(self, swh_scheduler, stored_lister): | ||||
lister = attr.evolve(stored_lister, current_state={"updated": "now"}) | lister = attr.evolve(stored_lister, current_state={"updated": "now"}) | ||||
updated_lister = swh_scheduler.update_lister(lister) | updated_lister = swh_scheduler.update_lister(lister) | ||||
assert updated_lister.updated > lister.updated | assert updated_lister.updated > lister.updated | ||||
assert updated_lister == attr.evolve(lister, updated=updated_lister.updated) | assert updated_lister == attr.evolve(lister, updated=updated_lister.updated) | ||||
▲ Show 20 Lines • Show All 848 Lines • Show Last 20 Lines |