This commit is contained in:
195
tests/database/stats_db_test.py
Normal file
195
tests/database/stats_db_test.py
Normal file
@@ -0,0 +1,195 @@
|
||||
import json
|
||||
import queries.crud as crud
|
||||
import queries.daily_stats as daily_stats
|
||||
import queries.monthly_stats as monthly_stats
|
||||
import queries.yearly_stats as yearly_stats
|
||||
import queries.weekly_stats as weekly_stats
|
||||
|
||||
from enums import CounterType
|
||||
from queries.connection import connection
|
||||
from sqlalchemy.sql.expression import text
|
||||
|
||||
|
||||
def test_all_daily_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
crud.create_counter("Test2", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-1 days'), 2),
|
||||
(1, date(date(), '-3 days'), 3),
|
||||
(2, date(), 2),
|
||||
(2, date(date(), '-1 days'), 4),
|
||||
(2, date(date(), '-3 days'), 6)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = daily_stats.get_all_daily_analytics()
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test"] == 1
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test2"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[1])["Test"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[1])["Test2"] == 4
|
||||
assert len(json.loads(stats[::-1]["counters"].iloc[2]).keys()) == 0
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test"] == 3
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test2"] == 6
|
||||
|
||||
|
||||
def test_daily_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-1 days'), 2),
|
||||
(1, date(date(), '-3 days'), 3)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = daily_stats.get_daily_analytics(1)
|
||||
assert stats["count"][0] == 1
|
||||
assert stats["count"][1] == 2
|
||||
assert stats["count"][2] == 0
|
||||
assert stats["count"][3] == 3
|
||||
|
||||
|
||||
def test_all_monthly_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
crud.create_counter("Test2", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-1 month'), 2),
|
||||
(1, date(date(), '-3 months'), 3),
|
||||
(2, date(), 2),
|
||||
(2, date(date(), '-2 months'), 4),
|
||||
(2, date(date(), '-3 months'), 6)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = monthly_stats.get_all_monthly_analytics()
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test"] == 1
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test2"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[1])["Test"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[2])["Test2"] == 4
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test"] == 3
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test2"] == 6
|
||||
|
||||
|
||||
def test_monthly_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-1 months'), 2),
|
||||
(1, date(date(), '-3 months'), 3)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = monthly_stats.get_monthly_analytics(1)
|
||||
assert stats[::-1]["count"].iloc[0] == 1
|
||||
assert stats[::-1]["count"].iloc[1] == 2
|
||||
assert stats[::-1]["count"].iloc[2] == 0
|
||||
assert stats[::-1]["count"].iloc[3] == 3
|
||||
|
||||
|
||||
def test_all_yearly_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
crud.create_counter("Test2", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-1 year'), 2),
|
||||
(1, date(date(), '-3 years'), 3),
|
||||
(2, date(), 2),
|
||||
(2, date(date(), '-2 years'), 4),
|
||||
(2, date(date(), '-3 years'), 6)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = yearly_stats.get_all_yearly_analytics()
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test"] == 1
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test2"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[1])["Test"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[2])["Test2"] == 4
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test"] == 3
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test2"] == 6
|
||||
|
||||
|
||||
def test_yearly_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-1 years'), 2),
|
||||
(1, date(date(), '-3 years'), 3)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = yearly_stats.get_yearly_analytics(1)
|
||||
assert stats[::-1]["count"].iloc[0] == 1
|
||||
assert stats[::-1]["count"].iloc[1] == 2
|
||||
assert stats[::-1]["count"].iloc[2] == 0
|
||||
assert stats[::-1]["count"].iloc[3] == 3
|
||||
|
||||
|
||||
def test_all_weekly_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
crud.create_counter("Test2", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-7 days'), 2),
|
||||
(1, date(date(), '-21 days'), 3),
|
||||
(2, date(), 2),
|
||||
(2, date(date(), '-14 days'), 4),
|
||||
(2, date(date(), '-21 days'), 6)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = weekly_stats.get_all_weekly_analytics()
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test"] == 1
|
||||
assert json.loads(stats[::-1]["counters"].iloc[0])["Test2"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[1])["Test"] == 2
|
||||
assert json.loads(stats[::-1]["counters"].iloc[2])["Test2"] == 4
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test"] == 3
|
||||
assert json.loads(stats[::-1]["counters"].iloc[3])["Test2"] == 6
|
||||
|
||||
|
||||
def test_weekly_stats():
|
||||
crud.create_counter("Test", CounterType.SIMPLE, "020122")
|
||||
with connection().session as session:
|
||||
query = text("""
|
||||
INSERT INTO entries (counter_id, "timestamp", increment)
|
||||
VALUES
|
||||
(1, date(), 1),
|
||||
(1, date(date(), '-7 days'), 2),
|
||||
(1, date(date(), '-21 days'), 3)
|
||||
""")
|
||||
session.execute(query)
|
||||
session.commit()
|
||||
|
||||
stats = weekly_stats.get_weekly_analytics(1)
|
||||
assert stats["count"][0] == 1
|
||||
assert stats["count"][1] == 2
|
||||
assert stats["count"][2] == 0
|
||||
assert stats["count"][3] == 3
|
||||
Reference in New Issue
Block a user