def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.run = RunFactory(
user=self.user,
project=self.project,
content="test",
raw_content="test",
is_managed=True,
)
self.url = "/{}/{}/{}/runs/{}/artifacts_lineage/names/".format(
API_V1, "polyaxon", self.project.name, self.run.uuid.hex)
self.objects = []
for i in range(self.num_objects):
obj = self.factory_class(
name=f"foo{i}",
state=uuid.uuid4(),
)
self.objects.append(obj)
ArtifactLineage.objects.create(run=self.run, artifact=obj)
self.query = self.queryset.filter(run=self.run)
# Other user objects
other_project = ProjectFactory()
other_run = RunFactory(
user=self.user,
project=other_project,
content="test",
raw_content="test",
is_managed=True,
)
other_obj = self.factory_class(state=uuid.uuid4(), )
ArtifactLineage.objects.create(run=other_run, artifact=other_obj)
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.url = "/{}/polyaxon/{}/runs/".format(API_V1, self.project.name)
self.objects = [
self.factory_class(project=self.project, user=self.user)
for _ in range(self.num_objects)
]
self.queryset = self.model_class.objects.filter(
project=self.project).order_by("-updated_at")
# one object that does not belong to the filter
self.other_project = ProjectFactory()
self.other_url = "/{}/polyaxon/{}/runs/".format(
API_V1, self.other_project.name)
self.other_object = self.factory_class(project=self.other_project, )
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.objects = []
self.objects.append(
self.factory_class(
project=self.project,
user=self.user,
tags=["tag11"],
)
)
self.objects.append(
self.factory_class(
project=self.project,
user=self.user,
tags=["new", "tag21", "tag22"],
)
)
self.objects.append(self.factory_class(project=self.project, user=self.user))
self.objects.append(self.factory_class(project=self.project, user=self.user))
self.url = "/{}/{}/{}/runs/tag/".format(
API_V1, self.user.username, self.project.name
)
self.queryset = self.model_class.objects.filter(project=self.project).order_by(
"created_at"
)
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.run = RunFactory(
project=self.project, content="test", raw_content="test", is_managed=True,
)
self.state = self.project.owner.uuid
def setUp(self):
super().setUp()
# Force tasks autodiscover
from scheduler import tasks # noqa
self.user = UserFactory()
self.project = ProjectFactory()
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.objects = [
self.factory_class(project=self.project, user=self.user)
for _ in range(4)
]
self.url = "/{}/{}/{}/runs/stop/".format(API_V1, self.user.username,
self.project.name)
def setUp(self):
super().setUp()
self.user = UserFactory()
self.user2 = UserFactory()
self.project = ProjectFactory()
op_spec = OperationSpecification.read(values=get_fxt_job_with_inputs())
self.run = compile_operation_run(project_id=self.project.id,
op_spec=op_spec,
user_id=self.user.id)
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.objects = [
self.factory_class(
project=self.project, user=self.user, pending=V1RunPending.APPROVAL
)
for _ in range(4)
]
self.url = "/{}/{}/{}/runs/approve/".format(
API_V1, self.user.username, self.project.name
)
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.object = self.factory_class(
user=self.user,
project=self.project,
content="test",
raw_content="test",
is_managed=True,
)
self.url = "/{}/{}/{}/runs/{}/".format(API_V1, self.user.username,
self.project.name,
self.object.uuid.hex)
self.queryset = self.model_class.objects.all()
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.objects = [
self.factory_class(project=self.project, user=self.user)
for _ in range(4)
]
for obj in self.objects:
new_run_status(
run=obj,
condition=V1StatusCondition.get_condition(
type=V1Statuses.RUNNING, status=True),
)
self.url = "/{}/{}/{}/runs/stop/".format(API_V1, self.user.username,
self.project.name)
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.run = RunFactory(
user=self.user,
project=self.project,
content="test",
raw_content="test",
is_managed=True,
)
self.artifact = self.factory_class(name="foo",
state=self.project.owner.uuid)
self.artifact_lineage = ArtifactLineage.objects.create(
artifact=self.artifact, run=self.run)
self.url = "/{}/{}/{}/runs/{}/artifacts_lineage/{}/".format(
API_V1,
"polyaxon",
self.project.name,
self.run.uuid.hex,
self.artifact.name,
)
self.queryset = Artifact.objects.all()
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.objects = [
RunFactory(project=self.project, user=self.user) for _ in range(3)
]
self.url = "/{}/polyaxon/{}/runs/artifacts_lineage/".format(
API_V1,
self.project.name,
)
obj = self.factory_class(name="in1", state=self.project.uuid)
ArtifactLineage.objects.create(run=self.objects[0], artifact=obj)
ArtifactLineage.objects.create(run=self.objects[1], artifact=obj)
obj = self.factory_class(name="in2", state=self.project.uuid)
ArtifactLineage.objects.create(run=self.objects[0], artifact=obj)
obj = self.factory_class(name="out1", state=self.project.uuid)
ArtifactLineage.objects.create(
run=self.objects[0],
artifact=obj,
is_input=False,
)
self.query = self.queryset.filter(run__project=self.project)
def setUp(self):
super().setUp()
self.user = UserFactory()
self.project = ProjectFactory()
self.run = RunFactory(project=self.project)
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.url = "/{}/polyaxon/{}/runs/".format(API_V1, self.project.name)
