from hittekaart_py import (
    Track, HeatmapRenderer, MarktileRenderer, TilehuntRenderer, Settings, Storage, generate
)


def tracks():
    return [
        Track.from_file(b"../test-assets/Synthetic_BRouter_1.gpx.gz", "gzip"),
    ]


def storage():
    return Storage.MbTiles(b":memory:", "heat", "mappy")


def test_track_from_file():
    track = Track.from_file(b"../test-assets/Synthetic_BRouter_1.gpx.gz", "gzip")
    assert track is not None


def test_track_from_coordinates():
    track = Track.from_coordinates([(1.0, 2.0), (3.0, 4.0)])
    assert track is not None


def test_storage_folder():
    storage = Storage.Folder(b"/tmp")
    assert storage is not None


def test_storage_osmand():
    storage = Storage.OsmAnd(b":memory:")
    assert storage is not None


def test_storage_mbtiles():
    storage = Storage.MbTiles(b":memory:")
    assert storage is not None

    storage = Storage.MbTiles(b":memory:", "Heatmap")
    assert storage is not None

    storage = Storage.MbTiles(b":memory:", "Heatmap", "This is my heatmap")
    assert storage is not None


def test_generate_heatmap():
    generate(
        Settings(),
        tracks(),
        HeatmapRenderer(),
        storage(),
    )


def test_generate_marktile():
    generate(
        Settings(),
        tracks(),
        MarktileRenderer(),
        storage(),
    )


def test_generate_tilehunt():
    generate(
        Settings(),
        tracks(),
        TilehuntRenderer(10),
        storage(),
    )