From ba2b32cb9ab447b402aa1f25b1b3a03c1dd0e05a Mon Sep 17 00:00:00 2001
From: JasonGrace2282 <aarush.deshpande@gmail.com>
Date: Wed, 5 Jun 2024 23:10:47 -0400
Subject: [PATCH] Add tests for folders

---
 tin/apps/assignments/tests/test_folders.py | 47 ++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 tin/apps/assignments/tests/test_folders.py

diff --git a/tin/apps/assignments/tests/test_folders.py b/tin/apps/assignments/tests/test_folders.py
new file mode 100644
index 00000000..65275b6a
--- /dev/null
+++ b/tin/apps/assignments/tests/test_folders.py
@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+import io
+
+from django.conf import settings
+from django.urls import reverse
+
+from tin.tests import is_redirect, login, str_to_html
+
+
+@login("teacher")
+def test_create_folder(client, course) -> None:
+    response = client.post(
+        reverse("assignments:add_folder", args=[course.id]), {"name": "Fragment Shader"}
+    )
+    assert is_redirect(response)
+    assert course.folders.exists()
+
+
+@login("teacher")
+def test_delete_folder(client, course) -> None:
+    folder = course.folders.create(name="My Folder")
+    response = client.post(reverse("assignments:delete_folder", args=[course.id, folder.id]))
+
+    assert is_redirect(response)
+    assert not course.folders.exists()
+
+
+@login("teacher")
+def test_incorrect_files(client, assignment):
+    binary_data = io.BytesIO(b"b" * (settings.SUBMISSION_SIZE_LIMIT + 1))
+    response = client.post(
+        reverse("assignments:manage_files", args=[assignment.id]), {"upload_file": binary_data}
+    )
+
+    error = str_to_html("That file's too large.")
+    assert error in response.content.decode("utf-8")
+
+
+@login("teacher")
+def test_files_management(client, assignment):
+    sample_grader = io.BytesIO(b"some text")
+    response = client.post(
+        reverse("assignments:manage_files", args=[assignment.id]), {"upload_file": sample_grader}
+    )
+
+    assert is_redirect(response)