37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
import json
|
|
import pytest
|
|
from vigilar.storage.queries import (
|
|
count_pet_rules, delete_pet_rule, get_all_enabled_rules,
|
|
get_pet_rules, insert_pet_rule, update_pet_rule,
|
|
)
|
|
|
|
def test_insert_and_get(test_db):
|
|
rid = insert_pet_rule(test_db, "pet1", "Outdoor timer",
|
|
json.dumps([{"type": "in_zone_longer_than", "zone": "EXTERIOR", "minutes": 45}]),
|
|
"push_notify", "{pet_name} outside", 30, 0)
|
|
assert rid > 0
|
|
rules = get_pet_rules(test_db, "pet1")
|
|
assert len(rules) == 1
|
|
assert rules[0]["name"] == "Outdoor timer"
|
|
|
|
def test_get_all_enabled(test_db):
|
|
insert_pet_rule(test_db, "pet1", "A", json.dumps([]), "push_notify", "", 30, 0)
|
|
insert_pet_rule(test_db, "pet2", "B", json.dumps([]), "log_event", "", 60, 0)
|
|
assert len(get_all_enabled_rules(test_db)) == 2
|
|
|
|
def test_update(test_db):
|
|
rid = insert_pet_rule(test_db, "pet1", "Test", json.dumps([]), "push_notify", "", 30, 0)
|
|
update_pet_rule(test_db, rid, name="Updated", cooldown_minutes=60)
|
|
rules = get_pet_rules(test_db, "pet1")
|
|
assert rules[0]["name"] == "Updated"
|
|
|
|
def test_delete(test_db):
|
|
rid = insert_pet_rule(test_db, "pet1", "Test", json.dumps([]), "push_notify", "", 30, 0)
|
|
delete_pet_rule(test_db, rid)
|
|
assert len(get_pet_rules(test_db, "pet1")) == 0
|
|
|
|
def test_count(test_db):
|
|
for i in range(5):
|
|
insert_pet_rule(test_db, "pet1", f"Rule {i}", json.dumps([]), "push_notify", "", 30, i)
|
|
assert count_pet_rules(test_db, "pet1") == 5
|