import pytest
import asyncio
from playwright.async_api import expect, Page

@pytest.mark.asyncio
class TestBillingUserFlowRefactored:

    async def test_navigate_to_billing_dashboard(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)
        await expect(page.locator('billing-dashboard')).to_be_visible()

    async def test_view_current_usage(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.usage-card')).to_be_visible()
        await expect(page.locator('text=Current Usage')).to_be_visible()
        await expect(page.locator('.usage-label:has-text("Storage")')).to_be_visible()
        await expect(page.locator('.usage-label:has-text("Bandwidth")')).to_be_visible()

    async def test_view_estimated_cost(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.cost-card')).to_be_visible()
        await expect(page.locator('text=Estimated Monthly Cost')).to_be_visible()
        await expect(page.locator('.estimated-cost')).to_be_visible()

        cost_text = await page.locator('.estimated-cost').text_content()
        assert '$' in cost_text

    async def test_view_pricing_information(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.pricing-card')).to_be_visible()
        await expect(page.locator('text=Current Pricing')).to_be_visible()

        await expect(page.locator('.pricing-item:has-text("Storage")')).to_be_visible()
        await expect(page.locator('.pricing-item:has-text("Bandwidth")')).to_be_visible()
        await expect(page.locator('.pricing-item:has-text("Free Tier")')).to_be_visible()

    async def test_view_invoice_history_empty(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.invoices-section')).to_be_visible()
        await expect(page.locator('text=Recent Invoices')).to_be_visible()

        no_invoices = page.locator('.no-invoices')
        if await no_invoices.is_visible():
            await expect(no_invoices).to_contain_text('No invoices yet')

    async def test_upload_file_to_track_usage(self, page: Page):
        # Navigate back to files view
        await page.click('a.nav-link[data-view="files"]')
        await expect(page.locator('h2:has-text("Files")')).to_be_visible(timeout=5000)

        await page.set_input_files('input[type="file"]', {
            'name': 'test-file.txt',
            'mimeType': 'text/plain',
            'buffer': b'This is a test file for billing usage tracking.'
        })

        await page.click('button:has-text("Upload")')
        
        await expect(page.locator('text=test-file.txt')).to_be_visible(timeout=10000)

    async def test_verify_usage_updated_after_upload(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        storage_value_locator = page.locator('.usage-item:has(.usage-label:has-text("Storage")) .usage-value')
        await expect(storage_value_locator).not_to_contain_text("0 B", timeout=10000)


    async def test_add_payment_method_button(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.payment-methods-section')).to_be_visible()
        await expect(page.locator('text=Payment Methods')).to_be_visible()
        await expect(page.locator('#addPaymentMethod')).to_be_visible()

    async def test_click_add_payment_method(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        page.on('dialog', lambda dialog: dialog.accept())

        await page.click('#addPaymentMethod')
        # We can't assert much here as it likely navigates to Stripe
        await page.wait_for_timeout(1000) # Allow time for navigation

    async def test_view_subscription_status(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.subscription-badge')).to_be_visible()

        badge_text = await page.locator('.subscription-badge').text_content()
        assert badge_text in ['Pay As You Go', 'Free', 'Active']

    async def test_verify_free_tier_display(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.usage-info:has-text("GB included free")')).to_be_visible()

        free_tier_info = await page.locator('.usage-info').text_content()
        assert '15' in free_tier_info or 'GB' in free_tier_info

    async def test_verify_progress_bar(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.usage-progress')).to_be_visible()
        await expect(page.locator('.usage-progress-bar')).to_be_visible()

    async def test_verify_cost_breakdown(self, page: Page):
        await page.click('a.nav-link[data-view="billing"]')
        await expect(page.locator('h1:has-text("Billing & Usage")')).to_be_visible(timeout=5000)

        await expect(page.locator('.cost-breakdown')).to_be_visible()
        await expect(page.locator('.cost-item:has-text("Storage")')).to_be_visible()
        await expect(page.locator('.cost-item:has-text("Bandwidth")')).to_be_visible()