Initial commit

dota-random-builds-back/api/tests.py

@@ -0,0 +1,69 @@
+from django.core.management import call_command
+from django.test import TestCase
+from rest_framework import status
+from rest_framework.test import APIClient
+
+from api.models import Hero, Item
+
+
+class RandomizeBuildAPITest(TestCase):
+    def setUp(self):
+        self.client = APIClient()
+        self.hero = Hero.objects.create(name="Axe", primary="Strength")
+        Item.objects.bulk_create(
+            [
+                Item(name="Blink Dagger"),
+                Item(name="Black King Bar"),
+                Item(name="Boots of Travel"),
+                Item(name="Force Staff"),
+            ]
+        )
+
+    def test_randomize_with_skills(self):
+        response = self.client.post(
+            "/api/randomize",
+            {"includeSkills": True, "includeAspect": False, "itemsCount": 3},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+
+        self.assertIn("hero", data)
+        self.assertEqual(len(data.get("items", [])), 3)
+        self.assertIn("skillBuild", data)
+        self.assertTrue(data.get("skillBuilds"))
+        self.assertNotIn("aspect", data)
+
+    def test_randomize_with_aspect_only(self):
+        response = self.client.post(
+            "/api/randomize",
+            {"includeSkills": False, "includeAspect": True, "itemsCount": 2},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+
+        self.assertIn("aspect", data)
+        self.assertTrue(data.get("aspects"))
+        self.assertNotIn("skillBuild", data)
+        self.assertEqual(len(data.get("items", [])), 2)
+
+    def test_rejects_invalid_payload(self):
+        response = self.client.post(
+            "/api/randomize",
+            {"includeSkills": "yes", "includeAspect": True, "itemsCount": 0},
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertIn("message", response.json())
+
+
+class LoadStaticDataCommandTest(TestCase):
+    def test_loads_heroes_and_items(self):
+        call_command("load_static_data")
+
+        self.assertGreater(Hero.objects.count(), 0)
+        self.assertGreater(Item.objects.count(), 0)