mirror of
https://github.com/bspeice/metrik
synced 2024-11-05 06:58:12 -05:00
19 lines
575 B
Python
19 lines
575 B
Python
from unittest import TestCase
|
|
from datetime import datetime
|
|
|
|
from metrik.batch import flows, build_cron_file
|
|
|
|
|
|
class BatchTest(TestCase):
|
|
def test_flows_return_schedule(self):
|
|
present = datetime.now()
|
|
live = False
|
|
for flow_name, flow_class in flows.items():
|
|
assert flow_class(present=present,
|
|
live=live).get_schedule() is not None
|
|
|
|
def test_cron_string(self):
|
|
cron_string = build_cron_file()
|
|
assert len(cron_string.split('\n')) == len(flows) + 1
|
|
assert cron_string[-1] == '\n'
|