#!/usr/bin/env python3

import sys
import asyncio
import tempfile
import aiosqlite
import string
import random
from pypaste.server import Paste
from pypaste.server.sqlite import Sqlite
from datetime import datetime
from pathlib import Path


def truncate(path: Path) -> None:
    with open(path, "w") as f:
        f.truncate(0)


def generate_key() -> str:
    chars = string.ascii_letters

    return "".join(random.choice(chars) for _ in range(10))


async def test_exists_but_not_in_our_table(storage: Sqlite) -> None:
    key = generate_key()

    await storage.connection.execute(
        "insert into pastes values(?, ?, ?, ?)",
        (key, datetime.now().isoformat(), None, bytes()),
    )

    assert not await storage.exists(key)


async def test_exists(storage: Sqlite) -> None:
    dt = datetime.now()
    key = generate_key()

    await storage.insert(Paste(key, dt, "test", "hello world"))

    assert await storage.exists(key)


async def test_delete(storage: Sqlite) -> None:
    dt = datetime.now()
    key = generate_key()

    await storage.insert(Paste(key, dt, "test", "hello world"))

    assert await storage.exists(key)

    await storage.delete(key)

    assert not await storage.exists(key)


async def test_insert_retrieve(storage: Sqlite) -> None:
    dt = datetime.now()
    key = generate_key()

    await storage.insert(Paste(key, dt, "test", "hello world"))

    paste = await storage.retrieve(key)

    assert paste is not None
    assert paste.key == key
    assert paste.dt == dt
    assert paste.syntax == "test"
    assert paste.text == "hello world"


async def main() -> int:
    with tempfile.TemporaryDirectory() as tmpdir:
        f = Path(tmpdir) / "database"
        truncate(f)
        async with aiosqlite.connect(f) as connection:
            await connection.execute(
                (
                    "create table pastes("
                    "key text,"
                    "datetime text,"
                    "size int,"
                    "syntax text"
                    ")"
                )
            )

            storage = Sqlite(connection)
            await storage.setup()

            await asyncio.gather(
                test_insert_retrieve(storage),
                test_delete(storage),
                test_exists_but_not_in_our_table(storage),
            )

    return 0


if __name__ == "__main__":
    sys.exit(asyncio.run(main()))