import pytest
from playwright.sync_api import Page, expect
from vega_sim.service import VegaService
from conftest import init_vega
from fixtures.market import setup_continuous_market

TOOLTIP_LABEL = "margin-health-tooltip-label"
TOOLTIP_VALUE = "margin-health-tooltip-value"
COL_ID_USED = ".ag-center-cols-container [col-id='used'] .ag-cell-value"

@pytest.fixture(scope="module")
def vega(request):
    with init_vega(request) as vega:
        yield vega

@pytest.fixture(scope="module")
def continuous_market(vega: VegaService):
    return setup_continuous_market(vega)

@pytest.mark.usefixtures("auth", "risk_accepted")
def test_usage_breakdown(continuous_market, page: Page):
    page.goto(f"/#/markets/{continuous_market}")
    page.get_by_test_id("Collateral").click()
    page.locator(".ag-floating-top-container .ag-row [col-id='used']").click()
    usage_breakdown = page.get_by_test_id('usage-breakdown')

    # Verify headers
    headers = ['Market', 'Account type', 'Balance', 'Margin health']
    ag_headers = usage_breakdown.locator('.ag-header-cell-text').element_handles()
    for i, header_element in enumerate(ag_headers):
        header_text = header_element.text_content()
        assert header_text == headers[i]

    # Other expectations
    expect(usage_breakdown.locator('[class="mb-2 text-sm"]')).to_have_text("You have 1,000,000.00 tDAI in total.")
    expect(usage_breakdown.locator(COL_ID_USED).first).to_have_text("8.50269 (0%)")
    expect(usage_breakdown.locator(COL_ID_USED).nth(1)).to_have_text("999,991.49731 (99%)")

    # Maintenance Level
    expect(usage_breakdown.locator(".ag-center-cols-container [col-id='market.id'] .ag-cell-value").first).to_have_text("2.85556 above maintenance level")

    # Margin health tooltip
    usage_breakdown.get_by_test_id("margin-health-chart-track").hover()
    tooltip_data = [("maintenance level", "5.64713"), ("search level", "6.21184"), ("initial level", "8.47069"), ("balance", "8.50269"), ("release level", "9.60012")]

    for index, (label, value) in enumerate(tooltip_data):
        expect(page.get_by_test_id(TOOLTIP_LABEL).nth(index)).to_have_text(label)
        expect(page.get_by_test_id(TOOLTIP_VALUE).nth(index)).to_have_text(value)


    page.get_by_test_id('dialog-close').click()