import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session, sessionmaker

from directdnsonly.app.backends.coredns_mysql import CoreDNSMySQLBackend, CoreDNSRecord
from loguru import logger


@pytest.fixture
def mysql_backend(tmp_path):
    # Setup in-memory SQLite for testing (replace with test MySQL in CI)
    engine = create_engine("sqlite:///:memory:")
    CoreDNSRecord.metadata.create_all(engine)

    class TestBackend(CoreDNSMySQLBackend):
        def __init__(self):
            super().__init__()
            self.engine = engine
            self.Session = scoped_session(sessionmaker(bind=engine))

    yield TestBackend()
    engine.dispose()


def test_zone_operations(mysql_backend):
    zone_data = """
example.com. 300 IN SOA ns.example.com. admin.example.com. (2023 3600 1800 604800 86400)
example.com. 300 IN A 192.0.2.1
"""
    # Test zone creation
    assert mysql_backend.write_zone("example.com", zone_data)
    assert mysql_backend.zone_exists("example.com")

    # Test record update
    updated_zone = """
example.com. 3600 IN A 192.0.2.1
example.com. 300 IN AAAA 2001:db8::1
"""
    assert mysql_backend.write_zone("example.com", updated_zone)

    # Test record removal
    reduced_zone = "example.com. 300 IN A 192.0.2.1"
    assert mysql_backend.write_zone("example.com", reduced_zone)

    # Test zone deletion
    assert mysql_backend.delete_zone("example.com")
    assert not mysql_backend.zone_exists("example.com")