From dc888b879c34dbd3ad10b970243168aaa833a8d6 Mon Sep 17 00:00:00 2001
From: Oksamies <asko.soisalo@gmail.com>
Date: Mon, 11 Dec 2023 15:10:46 +0200
Subject: [PATCH] Add tests for TeamCreateAPIView

---
 .../api/cyberstorm/tests/test_team.py         | 42 +++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/django/thunderstore/api/cyberstorm/tests/test_team.py b/django/thunderstore/api/cyberstorm/tests/test_team.py
index 4c566b941..12cbf6532 100644
--- a/django/thunderstore/api/cyberstorm/tests/test_team.py
+++ b/django/thunderstore/api/cyberstorm/tests/test_team.py
@@ -296,3 +296,45 @@ def test_team_add_member__when_adding_a_member__fails_because_user_is_already_in
         "Team Member with this User and Team already exists."
         in response2.json()["__all__"]
     )
+
+
+@pytest.mark.django_db
+def test_team_create__when_creating_a_team__succeeds(
+    api_client: APIClient,
+    user: UserType,
+):
+    api_client.force_authenticate(user)
+
+    response = api_client.post(
+        "/api/cyberstorm/teams/create/",
+        json.dumps({"name": "CoolestTeamNameEver"}),
+        content_type="application/json",
+    )
+
+    assert response.status_code == 200
+    response_json = response.json()
+    assert response_json["name"] == "CoolestTeamNameEver"
+    assert len(
+        Team.objects.get(name="CoolestTeamNameEver").members.filter(
+            user__username=user.username
+        )
+    )
+
+
+@pytest.mark.django_db
+def test_team_create__when_creating_a_team__fails_because_team_with_provided_name_exists(
+    api_client: APIClient,
+    user: UserType,
+    team: Team,
+):
+    api_client.force_authenticate(user)
+
+    response = api_client.post(
+        "/api/cyberstorm/teams/create/",
+        json.dumps({"name": team.name}),
+        content_type="application/json",
+    )
+
+    assert response.status_code == 400
+    response_json = response.json()
+    assert "A team with the provided name already exists" in response_json["name"]