import unittest
from datetime import datetime

from trade_queue import FIFOQueue

class TestFIFOQueue(unittest.TestCase):
    def setUp(self):
        """
        Set up a FIFOQueue instance and some test trades.
        """
        self.queue = FIFOQueue()
        self.queue.add(10.0, 100.0, "2025-04-14")
        self.queue.add(20.0, 200.0, "2025-04-15")
        self.queue.add(30.0, 300.0, "2025-04-16")

    def test_add(self):
        """
        Test adding trades to the queue.
        """
        self.assertEqual(len(self.queue.queue), 3)  # There should be 3 trades in the queue
        self.assertEqual(self.queue.queue[0].amount, 10.0)  # Check the first trade's amount
        self.assertEqual(self.queue.queue[1].date, "2025-04-15")  # Check the second trade's date

    def test_remove_exact_amount(self):
        """
        Test removing an exact amount from the queue.
        """
        trades = self.queue.remove(10.0)
        self.assertEqual(len(trades), 1)  # One trade should be returned
        self.assertEqual(trades[0].amount, 10.0)  # Amount should match the request
        self.assertEqual(len(self.queue.queue), 2)  # Two trades should remain in the queue

    def test_remove_partial_trade(self):
        """
        Test removing an amount that partially consumes a trade.
        """
        trades = self.queue.remove(5.0)
        self.assertEqual(len(trades), 1)  # One partial trade should be returned
        self.assertEqual(trades[0].amount, 5.0)  # Amount should match the request
        self.assertEqual(self.queue.queue[0].amount, 5.0)  # Remaining trade amount should update

    def test_remove_multiple_trades(self):
        """
        Test removing an amount that spans multiple trades.
        """
        trades = self.queue.remove(25.0)
        self.assertEqual(len(trades), 2)  # Two trades should be returned
        self.assertEqual(trades[0].amount, 10.0)  # The first trade should be fully consumed
        self.assertEqual(trades[1].amount, 15.0)  # The second trade should be partially consumed
        self.assertEqual(self.queue.queue[0].amount, 5.0)  # Remaining trade in queue should update

    def test_remove_insufficient_amount(self):
        """
        Test trying to remove more than is available in the queue.
        """
        with self.assertRaises(ValueError):
            self.queue.remove(100.0)  # This should raise an exception

    def test_remove_negative_amount(self):
        """
        Test trying to remove a negative amount.
        """
        with self.assertRaises(ValueError):
            self.queue.remove(-5.0)  # This should raise an exception

    def test_get_remaining_amount_initial(self):
        """
        Test the remaining amount in the queue after adding trades.
        """
        self.assertEqual(self.queue.get_remaining_amount(), 60.0)  # Total of all amounts: 10 + 20 + 30

    def test_get_remaining_amount_after_removal(self):
        """
        Test the remaining amount after removing some assets.
        """
        self.queue.remove(15.0)  # Remove 15 assets
        self.assertEqual(self.queue.get_remaining_amount(), 45.0)  # Remaining: 60 - 15

    def test_get_remaining_amount_empty_queue(self):
        """
        Test the remaining amount in an empty queue.
        """
        empty_queue = FIFOQueue()  # New empty queue
        self.assertEqual(empty_queue.get_remaining_amount(), 0.0)  # No trades in queue

    def test_get_remaining_amount_partial_removal(self):
        """
        Test the remaining amount after partially consuming a trade.
        """
        self.queue.remove(5.0)  # Remove 5 assets, leaving 5 in the first trade
        self.assertEqual(self.queue.get_remaining_amount(), 55.0)  # Remaining: 60 - 5

    def test_get_remaining_amount_full_removal(self):
        """
        Test the remaining amount after removing all trades.
        """
        self.queue.remove(60.0)  # Remove all assets
        self.assertEqual(self.queue.get_remaining_amount(), 0.0)  # Remaining: 0

    def test_remove_partial_trade_correct_cost(self):
        """
        Test removing a partial trade and ensure the correct cost is calculated.
        """
        trades = self.queue.remove(5.0)  # Remove 5 COIN from the first trade
        self.assertEqual(len(trades), 1)  # Only one trade should be returned

        self.assertAlmostEqual(trades[0].price_per_coin, 10, 1) # Coin-cost needs to stay constant
        self.assertEqual(trades[0].amount, 5.0)  # Check the removed amount
        self.assertEqual(trades[0].total_cost, 50.0)  # Total cost should be proportional: (100 * 5 / 10)

        self.assertAlmostEqual(self.queue.queue[0].price_per_coin, 10, 1)  # Original total cost remains unchanged
        self.assertEqual(self.queue.queue[0].amount, 5.0)  # Remaining amount in the first trade should be updated
        self.assertEqual(self.queue.queue[0].total_cost, 50.0)  # Original total cost remains unchanged

if __name__ == "__main__":
    unittest.main()