60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
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'
|
|
|