Merge pull request #1831 from MVrachev/small-tests

Add missing small tests
theupdateframework · Feb 8, 2022 · c7da199 · c7da199
2 parents efc6877 + 6db3f69
commit c7da199
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -639,6 +639,13 @@ def test_targetfile_from_file(self) -> None:
  with self.assertRaises(ValueError):
  TargetFile.from_file(file_path, file_path, ["123"])
 
+ def test_targetfile_custom(self) -> None:
+ # Test creating TargetFile and accessing custom.
+ targetfile = TargetFile(
+ 100, {"sha256": "abc"}, "file.txt", {"custom": "foo"}
+ )
+ self.assertEqual(targetfile.custom, "foo")
+
  def test_targetfile_from_data(self) -> None:
  data = b"Inline test content"
  target_file_path = os.path.join(

diff --git a/tests/test_metadata_serialization.py b/tests/test_metadata_serialization.py
@@ -214,6 +214,7 @@ def test_invalid_root_serialization(self, test_case_data: str) -> None:
 
  invalid_metafiles: utils.DataSet = {
  "wrong length type": '{"version": 1, "length": "a", "hashes": {"sha256" : "abc"}}',
+ "version 0": '{"version": 0, "length": 1, "hashes": {"sha256" : "abc"}}',
  "length 0": '{"version": 1, "length": 0, "hashes": {"sha256" : "abc"}}',
  "length below 0": '{"version": 1, "length": -1, "hashes": {"sha256" : "abc"}}',
  "empty hashes dict": '{"version": 1, "length": 1, "hashes": {}}',