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 | |||||