Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
Show First 20 Lines • Show All 1,379 Lines • ▼ Show 20 Lines | def test_origin_visit_get_random(self, swh_storage, sample_data): | ||||
) | ) | ||||
swh_storage.refresh_stat_counters() | swh_storage.refresh_stat_counters() | ||||
stats = swh_storage.stat_counters() | stats = swh_storage.stat_counters() | ||||
assert stats["origin"] == len(origins) | assert stats["origin"] == len(origins) | ||||
assert stats["origin_visit"] == len(origins) * len(visits) | assert stats["origin_visit"] == len(origins) * len(visits) | ||||
random_origin_visit = swh_storage.origin_visit_get_random(visit_type) | random_ov, random_ovs = swh_storage.origin_visit_get_random(visit_type) | ||||
assert random_origin_visit | assert random_ov and random_ovs | ||||
assert random_origin_visit["origin"] is not None | assert random_ov.origin is not None | ||||
assert random_origin_visit["origin"] in [o.url for o in origins] | assert random_ov.origin == random_ovs.origin | ||||
assert random_ov.origin in [o.url for o in origins] | |||||
def test_origin_visit_get_random_nothing_found(self, swh_storage, sample_data): | def test_origin_visit_get_random_nothing_found(self, swh_storage, sample_data): | ||||
origins = sample_data.origins | origins = sample_data.origins | ||||
swh_storage.origin_add(origins) | swh_storage.origin_add(origins) | ||||
visit_type = "hg" | visit_type = "hg" | ||||
# Add some visits outside of the random generation selection so nothing | # Add some visits outside of the random generation selection so nothing | ||||
# will be found by the random selection | # will be found by the random selection | ||||
visits = self._generate_random_visits(nb_visits=3, start=13, end=24) | visits = self._generate_random_visits(nb_visits=3, start=13, end=24) | ||||
▲ Show 20 Lines • Show All 2,777 Lines • Show Last 20 Lines |