from unittest.mock import Mock

from django.conf import settings
from django.contrib.auth.models import User
from django.test import TestCase
from django.utils import timezone
from finance.models import Ledger
from finance.models import Statement
from finance.rules import is_creator
from finance.rules import leads_excursion
from finance.rules import not_submitted
from members.models import FEMALE
from members.models import Freizeit
from members.models import GEMEINSCHAFTS_TOUR
from members.models import Group
from members.models import MALE
from members.models import Member
from members.models import MUSKELKRAFT_ANREISE


class FinanceRulesTestCase(TestCase):
    def setUp(self):
        self.group = Group.objects.create(name="Test Group")
        self.ledger = Ledger.objects.create(name="Test Ledger")

        self.user1 = User.objects.create_user(username="alice", password="test123")
        self.member1 = Member.objects.create(
            prename="Alice",
            lastname="Smith",
            birth_date=timezone.now().date(),
            email=settings.TEST_MAIL,
            gender=FEMALE,
            user=self.user1,
        )
        self.member1.group.add(self.group)

        self.user2 = User.objects.create_user(username="bob", password="test123")
        self.member2 = Member.objects.create(
            prename="Bob",
            lastname="Jones",
            birth_date=timezone.now().date(),
            email=settings.TEST_MAIL,
            gender=MALE,
            user=self.user2,
        )
        self.member2.group.add(self.group)

        self.freizeit = Freizeit.objects.create(
            name="Test Excursion",
            kilometers_traveled=100,
            tour_type=GEMEINSCHAFTS_TOUR,
            tour_approach=MUSKELKRAFT_ANREISE,
            difficulty=2,
        )
        self.freizeit.jugendleiter.add(self.member1)

        self.statement = Statement.objects.create(
            short_description="Test Statement",
            explanation="Test explanation",
            night_cost=27,
            created_by=self.member1,
            excursion=self.freizeit,
        )
        self.statement.allowance_to.add(self.member1)

    def test_is_creator_true(self):
        """Test is_creator predicate returns True when user created the statement"""
        self.assertTrue(is_creator(self.user1, self.statement))
        self.assertFalse(is_creator(self.user2, self.statement))

    def test_not_submitted_statement(self):
        """Test not_submitted predicate returns True when statement is not submitted"""
        self.statement.status = Statement.UNSUBMITTED
        self.assertTrue(not_submitted(self.user1, self.statement))
        self.statement.status = Statement.SUBMITTED
        self.assertFalse(not_submitted(self.user1, self.statement))

    def test_not_submitted_freizeit_with_statement(self):
        """Test not_submitted predicate with Freizeit having unsubmitted statement"""
        self.freizeit.statement = self.statement
        self.statement.status = Statement.UNSUBMITTED
        self.assertTrue(not_submitted(self.user1, self.freizeit))

    def test_not_submitted_freizeit_without_statement(self):
        """Test not_submitted predicate with Freizeit having no statement attribute"""
        # Create a mock Freizeit that truly doesn't have the statement attribute
        mock_freizeit = Mock(spec=Freizeit)
        # Remove the statement attribute entirely
        if hasattr(mock_freizeit, "statement"):
            delattr(mock_freizeit, "statement")
        self.assertTrue(not_submitted(self.user1, mock_freizeit))

    def test_leads_excursion_freizeit_user_is_leader(self):
        """Test leads_excursion predicate returns True when user leads the Freizeit"""
        self.assertTrue(leads_excursion(self.user1, self.freizeit))
        self.assertFalse(leads_excursion(self.user2, self.freizeit))

    def test_leads_excursion_statement_with_excursion(self):
        """Test leads_excursion predicate with statement having excursion led by user"""
        result = leads_excursion(self.user1, self.statement)
        self.assertTrue(result)

    def test_leads_excursion_statement_no_excursion_attribute(self):
        """Test leads_excursion predicate with statement having no excursion attribute"""
        mock_statement = Mock()
        del mock_statement.excursion
        result = leads_excursion(self.user1, mock_statement)
        self.assertFalse(result)

    def test_leads_excursion_statement_excursion_is_none(self):
        """Test leads_excursion predicate with statement having None excursion"""
        statement_no_excursion = Statement.objects.create(
            short_description="Test Statement No Excursion",
            explanation="Test explanation",
            night_cost=27,
            created_by=self.member1,
            excursion=None,
        )
        result = leads_excursion(self.user1, statement_no_excursion)
        self.assertFalse(result)