def test_run_model():
# Actuyally disable
return
if wait_elasticsearch(host="elasticsearch"):
os.environ["LOKTAR_RUN_DB_DB"] = "loktar_ci_{}".format(str(uuid4()))
os.environ["LOKTAR_RUN_DB_TABLE"] = "run"
run = Run()
payloads = [{
"run_archive_path": "run_archive{}".format(i),
"depency_graph": [],
"artifacts_modified": [],
"commit_id": "commit_id_{}".format(i),
"committer": "committer_{}".format(i),
"git_branch": "git_branch{}".format(i),
"result": "result_{}".format(i),
"jobs_state": {
"success": None,
"in_queue": True,
"faillure": None
},
"start_time": time.time() + i,
"end_time": None
}
for i in xrange(20)]
responses = [run.set_run(payload) for payload in payloads]
assert len(filter(None, responses)) == len(payloads)
for id_doc, payload in zip(responses, payloads):
assert run.get_run(id_doc) == payload
payload_lenght = len(payloads)
first_ten_payloads = payloads[:(payload_lenght/2)]
last_ten_payloads = payloads[payload_lenght/2:payload_lenght]
from pprint import pprint
pprint(first_ten_payloads)
print "="*10
pprint(run.get_runs())
print "="*10
pprint(run.get_runs(page=1))
print "="*10
pprint(run.get_runs(size=20))
assert run.get_runs() == first_ten_payloads
assert run.get_runs(page=1) == last_ten_payloads
assert run.get_runs(size=20) == payloads
del os.environ["LOKTAR_RUN_DB_DB"]
del os.environ["LOKTAR_RUN_DB_TABLE"]
def test_define_job_status_on_github_commit(mocker, mock_env):
mocker.patch("loktar.db.Elasticsearch", return_value=FakeElasticSearch())
results = list()
run = Run()
results.append(run.set_run({"foo": "bar0", "awesome": "payload0"}))
results.append(run.set_run({"foo": "bar1", "awesome": "payload1"}))
results.append(run.set_run({"foo": "bar2", "awesome": "payload2"}))
for i in xrange(len(results)):
assert run.get_run(results[i]) == {"foo": "bar{}".format(i), "awesome": "payload{}".format(i)}
assert run.get_runs() == [{'awesome': 'payload0', 'foo': 'bar0'},
{'awesome': 'payload1', 'foo': 'bar1'},
{'awesome': 'payload2', 'foo': 'bar2'}]