feat: make command parameter optional in set_alarm mutation
This commit is contained in:
+13
-8
@@ -3,16 +3,19 @@ import tempfile
|
||||
import uuid
|
||||
from api.crontab_manager import CrontabManager
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def crontab_file():
|
||||
with tempfile.NamedTemporaryFile(mode='w+', delete=False) as f:
|
||||
with tempfile.NamedTemporaryFile(mode="w+", delete=False) as f:
|
||||
pass
|
||||
yield f.name
|
||||
|
||||
|
||||
def test_crontab_manager_empty(crontab_file):
|
||||
manager = CrontabManager(tabfile=crontab_file)
|
||||
assert manager.get_alarms() == []
|
||||
|
||||
|
||||
def test_add_and_list_alarm(crontab_file):
|
||||
manager = CrontabManager(tabfile=crontab_file)
|
||||
alarm_id = str(uuid.uuid4())
|
||||
@@ -20,9 +23,9 @@ def test_add_and_list_alarm(crontab_file):
|
||||
alarm_id=alarm_id,
|
||||
cron_expression="30 7 * * *",
|
||||
command="python wecker.py",
|
||||
is_enabled=True
|
||||
is_enabled=True,
|
||||
)
|
||||
|
||||
|
||||
alarms = manager.get_alarms()
|
||||
assert len(alarms) == 1
|
||||
assert alarms[0]["id"] == alarm_id
|
||||
@@ -30,6 +33,7 @@ def test_add_and_list_alarm(crontab_file):
|
||||
assert alarms[0]["command"] == "python wecker.py"
|
||||
assert alarms[0]["is_enabled"] is True
|
||||
|
||||
|
||||
def test_update_alarm(crontab_file):
|
||||
manager = CrontabManager(tabfile=crontab_file)
|
||||
alarm_id = "test-id"
|
||||
@@ -37,27 +41,28 @@ def test_update_alarm(crontab_file):
|
||||
alarm_id=alarm_id,
|
||||
cron_expression="30 7 * * *",
|
||||
command="python wecker.py",
|
||||
is_enabled=True
|
||||
is_enabled=True,
|
||||
)
|
||||
|
||||
|
||||
manager.set_alarm(
|
||||
alarm_id=alarm_id,
|
||||
cron_expression="0 8 * * *",
|
||||
command="python wecker.py --loud",
|
||||
is_enabled=False
|
||||
is_enabled=False,
|
||||
)
|
||||
|
||||
|
||||
alarms = manager.get_alarms()
|
||||
assert len(alarms) == 1
|
||||
assert alarms[0]["cron_expression"] == "0 8 * * *"
|
||||
assert alarms[0]["command"] == "python wecker.py --loud"
|
||||
assert alarms[0]["is_enabled"] is False
|
||||
|
||||
|
||||
def test_delete_alarm(crontab_file):
|
||||
manager = CrontabManager(tabfile=crontab_file)
|
||||
alarm_id = "test-id-2"
|
||||
manager.set_alarm(alarm_id, "0 0 * * *", "cmd", True)
|
||||
assert len(manager.get_alarms()) == 1
|
||||
|
||||
|
||||
manager.delete_alarm(alarm_id)
|
||||
assert len(manager.get_alarms()) == 0
|
||||
|
||||
Reference in New Issue
Block a user