Add test coverage for delegated hash bins

Signed-off-by: Ivana Atanasova <[email protected]>

Author: ivanayov

tests/test_updater_delegation_graphs.py

@@ -353,6 +353,154 @@ def test_fishy_rolenames(self) -> None:
         for fname in roles_to_filenames.values():
             self.assertTrue(fname in local_metadata)
 
+    hash_bins_metadata: utils.DataSet = {
+        "delegated hash bins": DelegationsTestCase(
+            delegations=[
+                TestDelegation("targets", "f8-ff"),
+                TestDelegation("targets", "20-27"),
+                TestDelegation("targets", "70-77"),
+                TestDelegation("targets", "58-5f"),
+            ],
+            target_files=[
+                TestTarget(
+                    "f8-ff",
+                    b"fff0838676c1dacb8827b02511e89f85b43814a74dd2c3ffb31d952a1db94311",
+                    "test_hashed_bins.ext",
+                ),
+                TestTarget(
+                    "20-27",
+                    b"22992ae806e6d652835f3c84e169f05127da69e599f902e4dbdc6ff27ce46d01",
+                    "delegate_1.ext",
+                ),
+                TestTarget(
+                    "70-77",
+                    b"77e5df355e2af6a842b47e8cb3fb0649ec29062917e6bcf0793adbf001faefc2",
+                    "delegate_2.ext",
+                ),
+                TestTarget(
+                    "58-5f",
+                    b"5ac2ad3230b5cd7f2f0cc99ef2d0d5024a3a34f6b7bf38a4db8694a1fd953271",
+                    "delegate_3.ext",
+                ),
+            ],
+            visited_order=["f8-ff", "20-27", "70-77", "58-5f"],
+        ),
+    }
+
+    @utils.run_sub_tests_with_dataset(hash_bins_metadata)
+    def test_delegated_hash_bins(self, test_data: DelegationsTestCase) -> None:
+        try:
+            self._init_repo(test_data)
+            self.setup_subtest()
+
+            updater = self._init_updater()
+            # Call explicitly refresh to simplify the expected_calls list
+            updater.refresh()
+            self.sim.fetch_tracker.metadata.clear()
+
+            roles_pref_map = [
+                (
+                    "f8-ff",
+                    [
+                        "f8",
+                        "f9",
+                        "fa",
+                        "fb",
+                        "fc",
+                        "fd",
+                        "fe",
+                        "ff",
+                    ],
+                ),
+                (
+                    "20-27",
+                    [
+                        "20",
+                        "21",
+                        "22",
+                        "23",
+                        "24",
+                        "25",
+                        "26",
+                        "27",
+                    ],
+                ),
+                (
+                    "70-77",
+                    [
+                        "70",
+                        "71",
+                        "72",
+                        "73",
+                        "74",
+                        "75",
+                        "76",
+                        "77",
+                    ],
+                ),
+                (
+                    "58-5f",
+                    [
+                        "58",
+                        "59",
+                        "5a",
+                        "5b",
+                        "5c",
+                        "5d",
+                        "5e",
+                        "5f",
+                    ],
+                ),
+            ]
+
+            delegated_role1 = DelegatedRole(
+                roles_pref_map[0][0], [], 1, False, roles_pref_map[0][1]
+            )
+            delegated_role1.path_hash_prefixes = roles_pref_map[0][1]
+
+            delegated_role2 = DelegatedRole(
+                roles_pref_map[1][0], [], 1, False, roles_pref_map[1][1]
+            )
+            delegated_role2.path_hash_prefixes = roles_pref_map[1][1]
+
+            delegated_role3 = DelegatedRole(
+                roles_pref_map[2][0], [], 1, False, roles_pref_map[2][1]
+            )
+            delegated_role3.path_hash_prefixes = roles_pref_map[2][1]
+
+            delegated_role4 = DelegatedRole(
+                roles_pref_map[3][0], [], 1, False, roles_pref_map[3][1]
+            )
+            delegated_role4.path_hash_prefixes = roles_pref_map[3][1]
+
+            spec_version = ".".join(SPECIFICATION_VERSION)
+            targets = Targets(1, spec_version, self.sim.safe_expiry, {}, None)
+            self.sim.add_delegation("targets", delegated_role1, targets)
+            self.sim.add_delegation("f8-ff", delegated_role2, targets)
+            self.sim.add_delegation("20-27", delegated_role3, targets)
+            self.sim.add_delegation("70-77", delegated_role4, targets)
+
+            # verify that the target info was successfully found and the correct delegated metadata file was downloaded
+            target_info = updater.get_targetinfo("test_hashed_bins.ext")
+            assert target_info is not None
+            self.assertEqual(target_info.path, "test_hashed_bins.ext")
+
+            target_info = updater.get_targetinfo("delegate_1.ext")
+            assert target_info is not None
+            self.assertEqual(target_info.path, "delegate_1.ext")
+
+            target_info = updater.get_targetinfo("delegate_2.ext")
+            assert target_info is not None
+            self.assertEqual(target_info.path, "delegate_2.ext")
+
+            target_info = updater.get_targetinfo("delegate_3.ext")
+            assert target_info is not None
+            self.assertEqual(target_info.path, "delegate_3.ext")
+
+            # add any other error cases that can come to mind
+        finally:
+            self.teardown_subtest()
+
 
 class TestTargetFileSearch(TestDelegations):
     r"""