daily-counter/tests/database/crud_db_test.py

import queries.crud as crud
import queries.daily_stats as stats

from enums import CounterType


def test_create_counter():
    crud.create_counter("Test", CounterType.SIMPLE, "020122")

    counters = crud.get_counters()
    assert len(counters) == 1
    assert counters["id"][0] == 1
    assert counters["name"][0] == "Test"
    assert counters["type"][0] == CounterType.SIMPLE
    assert counters["color"][0] == '020122'


def test_remove_counter():
    crud.create_counter("Test", CounterType.SIMPLE, "020122")
    assert len(crud.get_counters()) == 1

    crud.remove_counter(1)
    assert len(crud.get_counters()) == 0


def test_increment_counter():
    crud.create_counter("Test", CounterType.SIMPLE, "020122")
    assert len(crud.get_counters()) == 1

    crud.increment_counter(1)

    daily_stats = stats.get_daily_analytics(1)
    assert daily_stats["count"][0] == 1

    crud.increment_counter(1)

    daily_stats = stats.get_daily_analytics(1)
    assert daily_stats["count"][0] == 2


def test_get_color_palettes():
    palettes = crud.get_color_palettes()
    assert len(palettes) == 5
    assert palettes['name'][0] == "Flames"


def test_get_user_colors():
    palettes = crud.get_color_palettes()
    palette_id = crud.get_color_palette()
    assert palette_id == 1
    assert palettes.loc[palettes['id'] == palette_id]["color1"][0] == 'F2F3AE'

    crud.set_color_palette(2)

    palettes = crud.get_color_palettes()
    palette_id = crud.get_color_palette()
    assert palette_id == 2
    assert palettes.loc[palettes['id'] == palette_id]["color1"][1] == '2B4141'