issues #9 part 2, test extracting specific students.

.gitignore

@@ -131,3 +131,6 @@ dmypy.json
 # Editor config
 .idea
 .vscode
+
+# Docx lock file
+*.~lock*

tests/integration/test_main.py

@@ -21,7 +21,7 @@
 from src.file_management.compile import compile_c
 
 
-class TestMainSafeMode(unittest.TestCase):
+class Base:
     # needed as class variables to run teardown after all tests, instead of each test.
     cwd = os.path.join("tests", "fixtures")
     safe_cwd = os.path.join(cwd, "example")
@@ -61,6 +61,8 @@ def test_01_safe_mode(self):
         assert os.path.exists(self.safe_cwd)
         assert os.path.exists(self.zip_path)
 
+
+class TestMainSafeModeSharedState(Base, unittest.TestCase):
     def test_02_unzip(self):
         """
         unzip_outer should unzip the main .zip archive
@@ -121,5 +123,21 @@ def test_07_feedback_content(self):
             for expected_cell, actual_cell in zip(expected_row.cells, actual_row.cells):
                 assert expected_cell.text == actual_cell.text
 
+
+class TestMainSafeModeSpecificStudents(Base, unittest.TestCase):
+    def test_02_unzip(self):
+        """
+        when students are passed, unzip_outer should only extract files containing that students name
+        """
+        assert not os.path.exists(self.example_student_dir)
+        unzip_outer(self.safe_zip_path, ["fake_student", "other_fake_student"])
+
+        # final_fake_student is the only one not extracted, all others are extracted
+        assert not os.path.exists(self.example_student_dir)
+        for dirname in ["fake_student_2012345", "other_fake_student_2012346"]:
+            student_dir = os.path.join(self.safe_cwd, dirname)
+            assert os.path.exists(student_dir)
+
+
 if __name__ == "__main__":
     unittest.main()