70 lines
2.2 KiB
Python
70 lines
2.2 KiB
Python
|
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)
|