import asyncio

import pytest

from aiomql.lib.result import Result
from aiomql.core.models import OrderSendResult


class TestResult:
    @pytest.fixture(scope="class")
    def parameters(self):
        return {"name": "test_trades", "ema": 20, "rsi": 14}

    @pytest.fixture(scope="function")
    async def order_results(self, mt, sell_order, buy_order, parameters):
        res1 = await mt.order_send(sell_order)
        res2 = await mt.order_send(buy_order)
        res1 = Result(result=OrderSendResult(**res1._asdict()), parameters=parameters)
        res2 = Result(result=OrderSendResult(**res2._asdict()), parameters=parameters)
        return res1, res2

    async def test_get_data(self, order_results):
        res1, res2 = order_results
        data1 = res1.get_data()
        data2 = res2.get_data()
        assert data1["actual_profit"] == data2["actual_profit"] == 0
        assert data1["closed"] == data2["closed"] == False
        assert data1["win"] == data2["win"] == False

    async def test_csv(self, order_results):
        res1, res2 = order_results
        await asyncio.gather(res1.save(), res2.save())
        assert res1.config.records_dir.exists()
        record = res1.config.records_dir / f"{res1.name}.csv"
        assert record.exists()

    async def test_json(self, order_results):
        res1, res2 = order_results
        await asyncio.gather(res1.save(trade_record_mode="json"), res2.save(trade_record_mode="json"))
        assert res1.config.records_dir.exists()
        record = res1.config.records_dir / f"{res1.name}.json"
        assert record.exists()