ngclient TrustedMetadataSet: improve unit testing

The current situation with the TrustedMetadataSet testing is that
we don't have a mnimimal amount of unit tests testing the different
branches in the various API functionality in the class.

This commit proposes simple unit tests covering almost all of the
branches in the API functions and increasing the unit test coverage
(as reported from the "coverage" tool) from 74 % to 97 %.

The code could be complicated at places, because the different
branches in the update_* functions depend on other metadata classes
as well.
Still, I hope we can find a way and simplify the code.

Signed-off-by: Martin Vrachev <mvrachev@vmware.com>

Author: MVrachev

tests/test_trusted_metadata_set.py

@@ -1,14 +1,21 @@
+import copy
 import json
 import logging
 import os
-import shutil
 import sys
-import tempfile
 import unittest
+from typing import Dict, Any
+from datetime import datetime
 
 from tuf import exceptions
-from tuf.api.metadata import Metadata
-from tuf.ngclient._internal.trusted_metadata_set import TrustedMetadataSet
+from tuf.api.metadata import Metadata, MetaFile
+from tuf.ngclient._internal.trusted_metadata_set import(
+    TrustedMetadataSet,
+    verify_with_threshold
+)
+from securesystemslib import hash as sslib_hash
+from securesystemslib.signer import SSlibSigner
+from securesystemslib.interface import import_ed25519_privatekey_from_file
 
 from tests import utils
 
@@ -26,6 +33,19 @@ def setUpClass(cls):
             with open(os.path.join(cls.repo_dir, f"{md}.json"), "rb") as f:
                 cls.metadata[md] = f.read()
 
+        cls.root = Metadata.from_bytes(cls.metadata["root"])
+        cls.timestamp = Metadata.from_bytes(cls.metadata["timestamp"])
+        cls.snapshot = Metadata.from_bytes(cls.metadata["snapshot"])
+        cls.targets = Metadata.from_bytes(cls.metadata["targets"])
+        cls.delegated_role = Metadata.from_bytes(cls.metadata["role1"])
+
+        keystore_dir = os.path.join(os.getcwd(), 'repository_data', 'keystore')
+        cls.keystore = {}
+        for role in ['delegation', 'snapshot', 'targets', 'timestamp']:
+            cls.keystore[role] = import_ed25519_privatekey_from_file(
+                os.path.join(keystore_dir, role + '_key'),
+                password="password"
+            )
 
     def test_update(self):
         trusted_set = TrustedMetadataSet(self.metadata["root"])
@@ -41,6 +61,9 @@ def test_update(self):
             self.metadata["role2"], "role2", "role1"
         )
 
+        # the 4 top level metadata objects + 2 additional delegated targets
+        self.assertTrue(len(trusted_set), 6)
+
     def test_out_of_order_ops(self):
         trusted_set = TrustedMetadataSet(self.metadata["root"])
 
@@ -118,9 +141,228 @@ def test_update_with_invalid_json(self):
 
             update_func(metadata)
 
+    def test_verify_with_threshold(self):
+        # Call verify_with_threshold with non root or targets delegator.
+        with self.assertRaises(ValueError):
+            verify_with_threshold(self.timestamp, "role1", self.delegated_role)
+
+        # Call verify_with_threshold with non existent role_name.
+        with self.assertRaises(ValueError):
+            verify_with_threshold(self.targets, "foo", self.delegated_role)
+
+    def test_invalid_update_root(self):
+        trusted_set = TrustedMetadataSet(self.metadata["root"])
+        # new_root data with invalid _type
+        invalid_type_data = json.loads(self.metadata["root"])
+        invalid_type_data["signed"]["_type"] = "foo"
+        invalid_type_data = json.dumps(invalid_type_data)
+        # RepositoryError is throwned during new_root deserialization.
+        # It's not throwned when checking new_root.signed.type != "root"
+        with self.assertRaises(exceptions.RepositoryError):
+            trusted_set.update_root(invalid_type_data.encode())
+
+        # new_root data with threshold which cannot be verified.
+        modified_threshold_data = copy.deepcopy(
+            json.loads(self.metadata["root"])
+        )
+        modified_threshold_data["signed"]["roles"]["root"]["threshold"] = 2
+        modified_threshold_data = json.dumps(modified_threshold_data).encode()
+        with self.assertRaises(exceptions.UnsignedMetadataError):
+            trusted_set.update_root(modified_threshold_data)
+
+        # new_root.signed.version has the same version as old root
+        with self.assertRaises(exceptions.ReplayedMetadataError):
+            trusted_set.update_root(self.metadata["root"])
+
+        # if _root_update_finished, then fail when calling update_root
+        trusted_set.root_update_finished()
+        with self.assertRaises(RuntimeError):
+            trusted_set.update_root(self.metadata["root"])
+        trusted_set._root_update_finished = False
+
+    def test_root_update_finished_expired(self):
+        trusted_set = TrustedMetadataSet(self.metadata["root"])
+        # call root_update_finished when trusted root has expired
+        expired_datetime = datetime.strptime(
+            "1970-01-01T00:00:00Z", "%Y-%m-%dT%H:%M:%SZ"
+        )
+        trusted_set.root.signed.expires = expired_datetime
+        with self.assertRaises(exceptions.ExpiredMetadataError):
+            trusted_set.root_update_finished()
+
+    def _sign_modified_obj(
+        self,
+        role:str,
+        metadata_obj: Metadata
+    ) -> Dict[str, Any]:
+        key_dict = self.keystore[role]
+        sslib_signer = SSlibSigner(key_dict)
+        signature = metadata_obj.sign(sslib_signer)
+        return signature.to_dict()
+
+    def test_update_timestamp(self):
+        trusted_set = TrustedMetadataSet(self.metadata["root"])
+        trusted_set.root_update_finished()
+        trusted_set.update_timestamp(self.metadata["timestamp"])
+        # new_timestamp.version < trusted_timestamp.version
+        trusted_set.timestamp.signed.version = 2
+        with self.assertRaises(exceptions.ReplayedMetadataError):
+            trusted_set.update_timestamp(self.metadata["timestamp"])
+        trusted_set.timestamp.signed.version = 1
+
+        # new_timestamp.snapshot.version < trusted_timestamp.snapshot.version
+        trusted_set.timestamp.signed.meta["snapshot.json"].version = 2
+        with self.assertRaises(exceptions.ReplayedMetadataError):
+            trusted_set.update_timestamp(self.metadata["timestamp"])
+        trusted_set.timestamp.signed.meta["snapshot.json"].version = 1
+
+        # new_timestamp has expired
+        timestamp = Metadata.from_bytes(self.metadata["timestamp"])
+        timestamp.signed.expires = datetime.strptime(
+            "1970-01-01T00:00:00Z", "%Y-%m-%dT%H:%M:%SZ"
+        )
+        self._sign_modified_obj("timestamp", timestamp)
+        new_timestamp_byte_data = json.dumps(timestamp.to_dict()).encode()
+        with self.assertRaises(exceptions.ExpiredMetadataError):
+            trusted_set.update_timestamp(new_timestamp_byte_data)
+
+    def _calculate_modified_hashes(
+        self, true_hashes,
+        data: bytes
+    ) -> Dict[str, str]:
+        modified_hashes = {}
+        # Calculate correct hashes to pass the first check
+        for algo in true_hashes.keys():
+            digest_object = sslib_hash.digest(algo)
+            digest_object.update(data)
+            observed_hash = digest_object.hexdigest()
+            modified_hashes[algo] = observed_hash
+        return modified_hashes
+
+    def test_update_snapshot(self):
+        trusted_set = TrustedMetadataSet(self.metadata["root"])
+        trusted_set.root_update_finished()
+        trusted_set.update_timestamp(self.metadata["timestamp"])
+        # cannot update snapshot after targets update completes or targets != None
+        targets_obj = Metadata.from_bytes(self.metadata["targets"])
+        trusted_set._trusted_set["targets"] = targets_obj
+        with self.assertRaises(RuntimeError):
+            trusted_set.update_snapshot(self.metadata["snapshot"])
+        del trusted_set._trusted_set["targets"]
+
+        #  Deserialization error - failed to load new_snapshot.
+        timestamp_meta = trusted_set.timestamp.signed.meta["snapshot.json"]
+        true_hashes = timestamp_meta.hashes or {}
+
+        modified_hashes = self._calculate_modified_hashes(
+            true_hashes, b'{""sig": }'
+        )
+        trusted_set.timestamp.signed.meta["snapshot.json"].hashes = modified_hashes
+        with self.assertRaises(exceptions.RepositoryError):
+            trusted_set.update_snapshot(b'{""sig": }')
+        trusted_set.timestamp.signed.meta["snapshot.json"].hashes = true_hashes
+
+        # root data with threshold which cannot be verified for new_snapshot
+        trusted_set.root.signed.roles["snapshot"].threshold = 2
+        with self.assertRaises(exceptions.UnsignedMetadataError):
+            trusted_set.update_snapshot(self.metadata["snapshot"])
+        trusted_set.root.signed.roles["snapshot"].threshold = 1
+
+        # new_snapshot.version != trusted timestamp.meta["snapshot"].version
+        trusted_set.timestamp.signed.meta["snapshot.json"].version = 2
+        with self.assertRaises(exceptions.BadVersionNumberError):
+            trusted_set.update_snapshot(self.metadata["snapshot"])
+        trusted_set.timestamp.signed.meta["snapshot.json"].version = 1
+
+
+    def test_update_snapshot_after_succesfull_update(self):
+        trusted_set = TrustedMetadataSet(self.metadata["root"])
+        trusted_set.root_update_finished()
+        trusted_set.update_timestamp(self.metadata["timestamp"])
+        trusted_set.update_snapshot(self.metadata["snapshot"])
+
+        # Test removing a meta_file in new_snapshot compared to the old snapshot
+        snapshot_obj = Metadata.from_bytes(self.metadata["snapshot"])
+        snapshot_obj.signed.meta = {}
+        # prepare timestamp.meta["snapshot"].hashes
+        self._sign_modified_obj("snapshot", snapshot_obj)
+        timestamp_meta = trusted_set.timestamp.signed.meta["snapshot.json"]
+        true_hashes = timestamp_meta.hashes or {}
+        modified_snapshot_data = json.dumps(snapshot_obj.to_dict()).encode()
+        modified_hashes = self._calculate_modified_hashes(
+            true_hashes, json.dumps(snapshot_obj.to_dict()).encode()
+        )
+        trusted_set.timestamp.signed.meta["snapshot.json"].hashes = modified_hashes
+
+        with self.assertRaises(exceptions.RepositoryError):
+            trusted_set.update_snapshot(modified_snapshot_data)
+        trusted_set.timestamp.signed.meta["snapshot.json"].hashes = true_hashes
+
+        # snapshot.meta["project1"].version != new_snapshot.meta["project1"].version
+        for meta_file_path in trusted_set.snapshot.signed.meta.keys():
+            trusted_set.snapshot.signed.meta[meta_file_path].version = 2
+        with self.assertRaises(exceptions.BadVersionNumberError):
+            trusted_set.update_snapshot(self.metadata["snapshot"])
+        for meta_file_path in trusted_set.snapshot.signed.meta.keys():
+            trusted_set.snapshot.signed.meta[meta_file_path].version = 1
+
+        # new_snapshot has expired
+        snapshot_obj = Metadata.from_bytes(self.metadata["snapshot"])
+        snapshot_obj.signed.expires = datetime.strptime(
+            "1970-01-01T00:00:00Z", "%Y-%m-%dT%H:%M:%SZ"
+        )
+        self._sign_modified_obj("snapshot", snapshot_obj)
+        modified_snapshot_data = json.dumps(snapshot_obj.to_dict()).encode()
+        modified_hashes = self._calculate_modified_hashes(
+            true_hashes, modified_snapshot_data
+        )
+        trusted_set.timestamp.signed.meta["snapshot.json"].hashes = modified_hashes
+        with self.assertRaises(exceptions.ExpiredMetadataError):
+            trusted_set.update_snapshot(modified_snapshot_data)
+        trusted_set.timestamp.signed.meta["snapshot.json"].hashes = true_hashes
+
+    def test_update_targets(self):
+        trusted_set = TrustedMetadataSet(self.metadata["root"])
+        trusted_set.root_update_finished()
+        trusted_set.update_timestamp(self.metadata["timestamp"])
+        trusted_set.update_snapshot(self.metadata["snapshot"])
+        
+        # remove meta information with information about targets from snapshot
+        trusted_set.snapshot.signed.meta = {}
+        with self.assertRaises(exceptions.RepositoryError):
+            trusted_set.update_targets(self.metadata["targets"])
+        trusted_set.snapshot.signed.meta = self.snapshot.signed.meta
+
+        # observed_hash != stored hash in snapshot for targets
+        true_hashes = {}
+        for target_path, meta_file in trusted_set.snapshot.signed.meta.items():
+            true_hashes[target_path] = meta_file.hashes
+            trusted_set.snapshot.signed.meta[target_path].hashes = {"sha256": "b"}
+        with self.assertRaises(exceptions.BadHashError):
+            trusted_set.update_targets(self.metadata["targets"])
+        # Return to the original hash values
+        for target_path in true_hashes.keys():
+            trusted_set.snapshot.signed.meta[target_path].hashes = \
+                true_hashes[target_path]
+
+        # new_delegate.signed.version != meta.version stored in snapshot
+        for target_path in trusted_set.snapshot.signed.meta.keys():
+            trusted_set.snapshot.signed.meta[target_path].version = 2
+        with self.assertRaises(exceptions.BadVersionNumberError):
+            trusted_set.update_targets(self.metadata["targets"])
+        trusted_set.snapshot.signed.meta[target_path].version = 1
+
+        # new_delegate has expired
+        targets_obj = Metadata.from_bytes(self.metadata["targets"])
+        targets_obj.signed.expires = datetime.strptime(
+            "1970-01-01T00:00:00Z", "%Y-%m-%dT%H:%M:%SZ"
+        )
+        self._sign_modified_obj("targets", targets_obj)
+        modified_targets_data = json.dumps(targets_obj.to_dict()).encode()
+        with self.assertRaises(exceptions.ExpiredMetadataError):
+            trusted_set.update_targets(modified_targets_data)
 
     # TODO test updating over initial metadata (new keys, newer timestamp, etc)
-    # TODO test the actual specification checks
 
 
 if __name__ == '__main__':