From f444a59b0ae0cc227a2f4767a91841cd36d3fd57 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         | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/django/thunderstore/api/cyberstorm/tests/test_team.py b/django/thunderstore/api/cyberstorm/tests/test_team.py
index 0c8e73e63..7831c6719 100644
--- a/django/thunderstore/api/cyberstorm/tests/test_team.py
+++ b/django/thunderstore/api/cyberstorm/tests/test_team.py
@@ -318,3 +318,63 @@ def test_team_member_add_api_view__when_adding_a_member__fails_because_user_is_n
         .count()
         == 0
     )
+
+
+@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 (
+        Team.objects.get(name="CoolestTeamNameEver")
+        .members.filter(user__username=user.username)
+        .count()
+        == 1
+    )
+
+
+@pytest.mark.django_db
+def test_team_create__when_creating_a_team__fails_because_user_is_not_authenticated(
+    api_client: APIClient,
+    user: UserType,
+):
+    response = api_client.post(
+        "/api/cyberstorm/teams/create/",
+        json.dumps({"name": "CoolestTeamNameEver"}),
+        content_type="application/json",
+    )
+
+    assert response.status_code == 401
+    response_json = response.json()
+    assert response_json["detail"] == "Authentication credentials were not provided."
+    assert Team.objects.filter(name="CoolestTeamNameEver").count() == 0
+
+
+@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"]