[test][export] Add assertions for which PDFs should be in the ZIPs

Author: mpsijm

test/test_problems.py

@@ -160,9 +160,10 @@ def test_samplezip(self):
         zip_path.unlink()
 
     def test_zip(self):
-        tools.test(["zip", "--force"])
         zip_path = Path("identity.zip")
 
+        tools.test(["zip", "--force"])
+
         # The full zip should contain the samples with the original file extensions.
         assert sorted(
             (info.filename, info.file_size)
@@ -178,6 +179,25 @@ def test_zip(self):
             *((f"identity/data/sample/6.{ext}.statement", 2) for ext in ["ans", "in"]),
         ], "Zip contents for data/sample/ are not correct"
 
+        # The full zip should contain all PDFs in their corresponding directories.
+        assert sorted(
+            info.filename for info in ZipFile(zip_path).infolist() if info.filename.endswith(".pdf")
+        ) == [
+            f"identity/{path}.{lang}.pdf"
+            for path in ["problem_slide/problem-slide", "solution/solution", "statement/problem"]
+            for lang in ["de", "en"]
+        ], "Zip contents for PDFs with both languages are not correct"
+
+        tools.test(["zip", "--force", "--lang", "en"])
+
+        # The full zip should contain all PDFs in their corresponding directories.
+        assert sorted(
+            info.filename for info in ZipFile(zip_path).infolist() if info.filename.endswith(".pdf")
+        ) == [
+            f"identity/{path}.en.pdf"
+            for path in ["problem_slide/problem-slide", "solution/solution", "statement/problem"]
+        ], "Zip contents for PDFs with `--lang en` are not correct"
+
         zip_path.unlink()
 
     # Misc
@@ -234,6 +254,32 @@ def test_problem_slides(self):
     def test_gitlabci(self):
         tools.test(["gitlabci"])
 
+    def test_zip(self):
+        zip_path = Path("problems.zip")
+
+        for languages in [["en", "de"], ["en"]]:
+            tools.test(["zip", "--force", "--lang", *languages])
+
+            # The full zip should contain all PDFs in their corresponding directories.
+            assert sorted(info.filename for info in ZipFile(zip_path).infolist()) == sorted(
+                [
+                    "contest.yaml",
+                    "identity.zip",
+                    "problems.yaml",
+                    "samples.zip",
+                    *(
+                        f"{name}{suffix}.{lang}.pdf"
+                        for name in ["contest", "solutions", "problem-slides"]
+                        for lang in languages
+                        for suffix in ["", "-web"]
+                        # The problem slides do not have a -web version.
+                        if (name, suffix) != ("problem-slides", "-web")
+                    ),
+                ]
+            ), f"Zip contents for contest zip are not correct for languages {languages}"
+
+        zip_path.unlink()
+
 
 @pytest.fixture(scope="function")
 def tmp_contest_dir(tmp_path):