from decimal import Decimal
from unittest.mock import patch

from django.contrib.auth import get_user_model
from django.test import TestCase
from rest_framework.authtoken.models import Token
from rest_framework.test import APIClient

from accounts.models import TransporterDriverLink, TransporterProfile, UserRole
from core.models import Bid, Notification, Shipment, Trip

User = get_user_model()


class TripProgressNotificationTests(TestCase):
    def setUp(self):
        self.shipper = User.objects.create_user(
            'trip-notif-ship@test.com',
            email='trip-notif-ship@test.com',
            password='secret',
        )
        UserRole.objects.create(user=self.shipper, role=UserRole.Role.SHIPPER, phone='', language='en')

        self.individual_driver = User.objects.create_user(
            'trip-notif-ind@test.com',
            email='trip-notif-ind@test.com',
            password='secret',
        )
        UserRole.objects.create(
            user=self.individual_driver,
            role=UserRole.Role.TRANSPORTER,
            phone='',
            language='en',
        )
        TransporterProfile.objects.create(
            user=self.individual_driver,
            account_type=TransporterProfile.AccountType.DRIVER,
            local=True,
            country_to_country=False,
            documents_verified=True,
        )

        self.fleet_owner = User.objects.create_user(
            'trip-notif-owner@test.com',
            email='trip-notif-owner@test.com',
            password='secret',
        )
        UserRole.objects.create(
            user=self.fleet_owner,
            role=UserRole.Role.TRANSPORTER,
            phone='',
            language='en',
        )
        TransporterProfile.objects.create(
            user=self.fleet_owner,
            account_type=TransporterProfile.AccountType.FLEET_OWNER,
            local=True,
            country_to_country=False,
            documents_verified=True,
        )

        self.fleet_driver = User.objects.create_user(
            'trip-notif-fleetdr@test.com',
            email='trip-notif-fleetdr@test.com',
            password='secret',
        )
        UserRole.objects.create(
            user=self.fleet_driver,
            role=UserRole.Role.TRANSPORTER,
            phone='',
            language='en',
        )
        TransporterProfile.objects.create(
            user=self.fleet_driver,
            account_type=TransporterProfile.AccountType.TRANSPORTER_DRIVER,
            local=True,
            country_to_country=False,
            documents_verified=True,
        )
        TransporterDriverLink.objects.create(
            transporter=self.fleet_owner,
            driver=self.fleet_driver,
            is_active=True,
        )

    def _shipment(self):
        return Shipment.objects.create(
            shipper=self.shipper,
            pickup_address='A',
            delivery_address='B',
            cargo_type='General',
            weight='1 ton',
            vehicle_type_required='Flatbed',
            local=True,
            country_to_country=False,
            status=Shipment.Status.ASSIGNED,
        )

    def _client_for(self, user):
        token, _ = Token.objects.get_or_create(user=user)
        client = APIClient()
        client.credentials(HTTP_AUTHORIZATION=f'Bearer {token.key}')
        return client

    def _individual_trip(self):
        shipment = self._shipment()
        bid = Bid.objects.create(
            shipment=shipment,
            transporter=self.individual_driver,
            amount=Decimal('100.00'),
            status=Bid.Status.ACCEPTED,
        )
        return Trip.objects.create(
            shipment=shipment,
            accepted_bid=bid,
            transporter=self.individual_driver,
            status=Trip.Status.ASSIGNED,
        )

    def _fleet_trip(self):
        shipment = self._shipment()
        bid = Bid.objects.create(
            shipment=shipment,
            transporter=self.fleet_owner,
            amount=Decimal('200.00'),
            status=Bid.Status.ACCEPTED,
        )
        return Trip.objects.create(
            shipment=shipment,
            accepted_bid=bid,
            transporter=self.fleet_owner,
            assigned_driver=self.fleet_driver,
            status=Trip.Status.ASSIGNED,
        )

    @patch('api.views.notify_users')
    def test_individual_driver_status_update_notifies_shipper_only(self, mock_notify):
        trip = self._individual_trip()
        client = self._client_for(self.individual_driver)

        response = client.patch(
            f'/api/transporter/trips/{trip.id}/status/',
            {'status': Trip.Status.EN_ROUTE},
            format='json',
        )
        self.assertEqual(response.status_code, 200)
        mock_notify.assert_called_once()
        self.assertEqual(mock_notify.call_args[0][0], [self.shipper.id])
        self.assertEqual(mock_notify.call_args[0][1], 'TRIP_STATUS')

    @patch('api.views.notify_users')
    def test_fleet_driver_status_update_notifies_shipper_and_owner(self, mock_notify):
        trip = self._fleet_trip()
        client = self._client_for(self.fleet_driver)

        response = client.patch(
            f'/api/transporter/trips/{trip.id}/status/',
            {'status': Trip.Status.EN_ROUTE},
            format='json',
        )
        self.assertEqual(response.status_code, 200)
        mock_notify.assert_called_once()
        self.assertEqual(
            sorted(mock_notify.call_args[0][0]),
            sorted([self.shipper.id, self.fleet_owner.id]),
        )

    @patch('api.views.notify_users')
    def test_fleet_driver_complete_trip_notifies_shipper_and_owner(self, mock_notify):
        trip = self._fleet_trip()
        trip.status = Trip.Status.DELIVERED
        trip.save(update_fields=['status', 'updated_at'])
        client = self._client_for(self.fleet_driver)

        response = client.patch(
            f'/api/transporter/trips/{trip.id}/status/',
            {'status': Trip.Status.COMPLETED},
            format='json',
        )
        self.assertEqual(response.status_code, 200)
        mock_notify.assert_called_once()
        self.assertEqual(
            sorted(mock_notify.call_args[0][0]),
            sorted([self.shipper.id, self.fleet_owner.id]),
        )

    @patch('api.notify.send_fcm_data_to_user', return_value=False)
    def test_fleet_driver_status_update_persists_notifications_for_shipper_and_owner(self, _mock_fcm):
        trip = self._fleet_trip()
        client = self._client_for(self.fleet_driver)

        response = client.patch(
            f'/api/transporter/trips/{trip.id}/status/',
            {'status': Trip.Status.EN_ROUTE},
            format='json',
        )

        self.assertEqual(response.status_code, 200)
        shipper_notifs = Notification.objects.filter(user=self.shipper, type='trip_status')
        owner_notifs = Notification.objects.filter(user=self.fleet_owner, type='trip_status')
        self.assertEqual(shipper_notifs.count(), 1)
        self.assertEqual(owner_notifs.count(), 1)
        self.assertEqual(shipper_notifs.first().data.get('status'), Trip.Status.EN_ROUTE)
        self.assertEqual(owner_notifs.first().data.get('status'), Trip.Status.EN_ROUTE)