from django.test import TestCase
from django.contrib.auth import get_user_model
import datetime

from modules.task_management.models import Task


class TaskModelTests(TestCase):
    def test_str_returns_title_and_status(self):
        user = get_user_model().objects.create_user(username="u1", email="u1@example.test", password="pass")
        task = Task.objects.create(title="My Task", description="Desc", assigned_to=user)
        assert str(task).startswith("My Task")
        assert "aktivan" in str(task) or "(" in str(task)

    def test_scheduled_date_defaults_to_null(self):
        user = get_user_model().objects.create_user(username="u2", email="u2@example.test", password="pass")
        task = Task.objects.create(title="Task bez datuma", assigned_to=user)
        assert task.scheduled_date is None

    def test_scheduled_date_is_saved_and_retrieved(self):
        user = get_user_model().objects.create_user(username="u3", email="u3@example.test", password="pass")
        target = datetime.date(2026, 8, 15)
        task = Task.objects.create(title="Planirani task", assigned_to=user, scheduled_date=target)
        task.refresh_from_db()
        assert task.scheduled_date == target

    def test_scheduled_date_can_be_updated_to_null(self):
        user = get_user_model().objects.create_user(username="u4", email="u4@example.test", password="pass")
        target = datetime.date(2026, 8, 15)
        task = Task.objects.create(title="Task s datumom", assigned_to=user, scheduled_date=target)
        task.scheduled_date = None
        task.save()
        task.refresh_from_db()
        assert task.scheduled_date is None