def test_it_can_associate_config_to_context(webapp):
test_env = Environment(name="test_env")
db.session.add(test_env)
job = Job(name="retrieve cash register data", type=JobType.SCHEDULED)
job.set_config(API_URL="http://a.place.somewhere", OPERATION_MODE="normal")
db.session.add(job)
context = Context(environment=test_env, job=job)
db.session.add(context)
config = ConfigKey(name="API_URL", value="http://another.place.somewhere")
config.context = context
db.session.commit()
assert config.scope == ConfigScope.CONTEXT
assert context.variables() == {
"API_URL": "http://another.place.somewhere",
"OPERATION_MODE": "normal",
}
def test_it_creates_a_job(webapp):
job = Job(
name="retrieve cash register data",
description="connects to the cash register API and fetches latest sale",
type=JobType.SCHEDULED,
)
test_env = Environment(name="test")
context = Context(environment=test_env)
context.set_config(url="http://*****:*****@local.local")
job.add_context(context)
job.add_notification(group, on_retry=True, on_failure=True)
db.session.add(job)
db.session.commit()
assert (
job.notifications[0].groups[0].emails[0].email == "*****@*****.**")
def test_it_creates_a_script(webapp):
test_env = Environment(name="test_env")
db.session.add(test_env)
job = Job(
name="retrieve cash register data",
description="connects to the cash register API and fetches latest sale",
type=JobType.SCHEDULED,
)
db.session.add(job)
context = Context(environment=test_env, job=job)
schedule = Schedule(name="every two minutes", hour="*", minute="*/2")
db.session.add(schedule)
context.add_schedule(schedule)
db.session.add(context)
script = Script(
context=context,
package_name="test_package",
package_version="0.0.1",
first_seen=datetime.now(),
)
source = Topic(name="source")
db.session.add(source)
destination = Topic(name="destination")
db.session.add(destination)
script.add_source(source)
script.add_destination(destination)
db.session.add(script)
db.session.commit()
assert script.sources == [source]
assert script.destinations == [destination]
assert script.context.environment.name == "test_env"
assert script.context.job.name == "retrieve cash register data"
assert script.context.schedules[0].name == "every two minutes"
def db_contexts(db_env, db_jobs):
[orders_job, cashregister_job, prepare_job, abort_job] = db_jobs
orders_context = Context(
environment=db_env,
job=orders_job,
schedules=[Schedule(name="Every minute")],
)
cashregister_context = Context(environment=db_env, job=cashregister_job)
prepare_context = Context(
environment=db_env,
job=prepare_job,
schedules=[Schedule(name="Every day at 00:00", minute="0", hour="0")],
)
abort_context = Context(environment=db_env, job=abort_job)
contexts = [
orders_context,
cashregister_context,
prepare_context,
abort_context,
]
db.session.add_all(contexts)
db.session.commit()
return contexts