mirror of
https://github.com/clearml/clearml-server
synced 2025-01-31 10:56:48 +00:00
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from apiserver.config_repo import config
|
|
from apiserver.tests.automated import TestService
|
|
|
|
log = config.logger(__file__)
|
|
|
|
|
|
class TestTasksDiff(TestService):
|
|
def setUp(self, version="2.0"):
|
|
super(TestTasksDiff, self).setUp(version=version)
|
|
|
|
def new_task(self, **kwargs):
|
|
return self.create_temp(
|
|
"tasks", name="test", type="testing", **kwargs
|
|
)
|
|
|
|
def _compare_script(self, task_id, script):
|
|
task = self.api.tasks.get_by_id(task=task_id).task
|
|
if not script:
|
|
self.assertTrue(
|
|
task.get(
|
|
"script",
|
|
dict(
|
|
binary="python", repository="", entry_point="", requirements={}
|
|
),
|
|
)
|
|
)
|
|
else:
|
|
for key, value in script.items():
|
|
self.assertEqual(task.script[key], value)
|
|
|
|
def test_not_deleted(self):
|
|
task_id = self.new_task()
|
|
script = dict(
|
|
requirements=dict(pip=["six"]),
|
|
repository="https://example.come/foo/bar",
|
|
entry_point="test.py",
|
|
diff="foo",
|
|
)
|
|
self.api.tasks.edit(task=task_id, script=script)
|
|
self.api.tasks.started(task=task_id)
|
|
|
|
self.api.tasks.reset(task=task_id)
|
|
self._compare_script(task_id, script)
|
|
|
|
new_reqs = dict()
|
|
self.api.tasks.set_requirements(task=task_id, requirements=new_reqs)
|
|
script["requirements"] = new_reqs
|
|
self._compare_script(task_id, script)
|
|
|
|
self.api.tasks.reset(task=task_id, clear_all=True)
|
|
self._compare_script(task_id, {})
|