Add unit test for chunk_metadata

tests/test_publish2cloud.py

@@ -0,0 +1,26 @@
+import hashlib
+import time
+
+from mock import mock_open, patch
+
+from publish2cloud import chunk_metadata
+
+
+def test_chunk_metadata():
+    """Test getting metadata from the chunk header of a list file."""
+    chunknum = int(time.time())
+    domain = "base-fingerprinting-track-digest256.dummytracker.org/"
+    domain_hash = hashlib.sha256(domain.encode("utf-8")).digest()
+    hash_size = hashlib.sha256().digest_size
+    header = b"a:%d:%d:%d\n" % (chunknum, hash_size, hash_size)
+    data = header + domain_hash
+
+    with patch("test_publish2cloud.open", mock_open(read_data=data)):
+        with open("base-fingerprinting-track-digest256", "rb") as fp:
+            metadata = chunk_metadata(fp)
+
+    assert metadata["type"] == "a"
+    assert metadata["num"] == str(chunknum)
+    assert metadata["hash_size"] == str(hash_size)
+    assert metadata["len"] == str(hash_size)
+    assert metadata["checksum"] == hashlib.sha256(domain_hash).hexdigest()