def setUp(self):
super().setUp()
self.user = UserFactory()
activitylogs.validate()
activitylogs.setup()
self.project = ProjectFactory()
activitylogs.record(ref_id=uuid.uuid4(),
event_type=USER_ACTIVATED,
instance=self.user,
actor_id=self.user.id,
actor_name=self.user.username)
activitylogs.record(ref_id=uuid.uuid4(),
event_type=PROJECT_DELETED_TRIGGERED,
instance=self.project,
actor_id=self.user.id,
actor_name=self.user.username)
self.experiment = ExperimentFactory()
activitylogs.record(ref_id=uuid.uuid4(),
event_type=EXPERIMENT_DELETED_TRIGGERED,
instance=self.experiment,
actor_id=self.user.id,
actor_name=self.user.username)
self.job = JobFactory()
activitylogs.record(ref_id=uuid.uuid4(),
event_type=JOB_VIEWED,
instance=self.job,
actor_id=self.user.id,
actor_name=self.user.username)
def setUp(self):
super().setUp()
self.experiment = ExperimentFactory()
self.admin = UserFactory(is_staff=True, is_superuser=True)
self.user = UserFactory()
activitylogs.validate()
activitylogs.setup()
def setUp(self):
super().setUp()
activitylogs.validate()
activitylogs.setup()
self.set_objects()
self.set_url()
self.set_queryset()
def setUp(self):
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super().setUp()
def setUp(self):
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorPermissionTest, self).setUp()
def setUp(self):
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorSuperUserTest, self).setUp()
def setUp(self):
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorPermissionTest, self).setUp()
def setUp(self):
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorSuperUserTest, self).setUp()
def setUp(self):
self.user = UserFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorUserTest, self).setUp()
def setUp(self):
self.notebook = NotebookJobFactory(project=ProjectFactory())
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorNotebookTest, self).setUp()
def setUp(self):
self.experiment_job = ExperimentJobFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorExperimentJobTest, self).setUp()
def setUp(self):
super().setUp()
self.experiment = ExperimentFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
def setUp(self):
self.tensorboard = TensorboardJobFactory(project=ProjectFactory())
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorTensorboardTest, self).setUp()
def setUp(self):
self.build_job = BuildJobFactory(project=ProjectFactory())
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super().setUp()
def setUp(self):
self.tensorboard = TensorboardJobFactory(project=ProjectFactory())
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorTensorboardTest, self).setUp()
def setUp(self):
super().setUp()
self.experiment_group = ExperimentGroupFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
def setUp(self):
self.project = ProjectFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorProjectTest, self).setUp()
def setUp(self):
self.project = RepoFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorRepoTest, self).setUp()
def setUp(self):
self.user = UserFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super().setUp()
def setUp(self):
self.project = ProjectFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super().setUp()
def setUp(self):
self.job = NotebookJobFactory(project=ProjectFactory())
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorJobTest, self).setUp()
def setUp(self):
self.cluster = Cluster.load()
self.cluster_node = ClusterNodeFactory(cluster=self.cluster)
self.node_gpu = GPUFactory(cluster_node=self.cluster_node)
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorClusterTest, self).setUp()
def setUp(self):
super().setUp()
self.experiment_job = ExperimentJobFactory()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
notifier.validate()
notifier.setup()
def setUp(self):
super().setUp()
activitylogs.validate()
activitylogs.setup()
self.set_objects()
self.set_url()
self.objects = self.model_class.objects.all()
self.queryset = self.model_class.objects.all()
self.queryset = self.queryset.order_by('-created_at')
def setUp(self):
Cluster.load()
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
notifier.validate()
notifier.setup()
super().setUp()
def setUp(self):
self.cluster = Cluster.load()
self.cluster_node = ClusterNodeFactory(cluster=self.cluster)
self.node_gpu = GPUFactory(cluster_node=self.cluster_node)
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
super(AuditorClusterTest, self).setUp()
def setup_auditor_services(self):
if not self.is_testing_env:
import activitylogs
import auditor
import tracker
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
def setUp(self):
super().setUp()
self.project = ProjectFactory()
self.user = self.project.user
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
notifier.validate()
notifier.setup()
def setup_auditor_services(self):
if not self.is_testing_env:
import activitylogs
import auditor
import tracker
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
def setUp(self):
# Force tasks autodiscover
from scheduler import tasks # noqa
from hpsearch.tasks import bo, grid, health, hyperband, random # noqa
from polyflow import health, tasks # noqa
from crons import tasks # noqa
from events_handlers import tasks # noqa
from k8s_events_handlers import tasks # noqa
from logs_handlers import tasks # noqa
# Flushing all redis databases
redis.StrictRedis(connection_pool=RedisPools.JOB_CONTAINERS).flushall()
redis.StrictRedis(connection_pool=RedisPools.TO_STREAM).flushall()
# Mock dirs
settings.REPOS_MOUNT_PATH = tempfile.mkdtemp()
settings.UPLOAD_MOUNT_PATH = tempfile.mkdtemp()
settings.PERSISTENCE_LOGS['mountPath'] = tempfile.mkdtemp()
settings.PERSISTENCE_OUTPUTS['outputs'][
'mountPath'] = tempfile.mkdtemp()
settings.ARCHIVES_ROOT_REPOS = tempfile.mkdtemp()
settings.ARCHIVES_ROOT_ARTIFACTS = tempfile.mkdtemp()
settings.DOWNLOADS_ROOT_ARTIFACTS = tempfile.mkdtemp()
settings.DOWNLOADS_ROOT_LOGS = tempfile.mkdtemp()
settings.ARCHIVES_ROOT_LOGS = tempfile.mkdtemp()
# Flush cache
cache.clear()
# Mock celery default sent task
self.mock_send_task()
conf.setup()
ownership.setup()
if self.DISABLE_RUNNER:
self.disable_experiment_groups_runner()
self.disable_experiments_runner()
self.plugin_jobs_runner()
super().setUp()
if not self.DISABLE_EXECUTOR or not self.DISABLE_AUDITOR:
auditor.validate()
auditor.setup()
if not self.DISABLE_AUDITOR:
tracker.validate()
tracker.setup()
activitylogs.validate()
activitylogs.setup()
notifier.validate()
notifier.setup()
if not self.DISABLE_EXECUTOR:
executor.validate()
executor.setup()
def setUp(self):
super().setUp()
self.user = UserFactory()
activitylogs.validate()
activitylogs.setup()
self.project = ProjectFactory()
activitylogs.record(event_type=PROJECT_DELETED_TRIGGERED,
instance=self.project,
actor_id=self.user.id)
self.experiment = ExperimentFactory()
activitylogs.record(event_type=EXPERIMENT_DELETED_TRIGGERED,
instance=self.experiment,
actor_id=self.user.id)
self.job = JobFactory()
activitylogs.record(event_type=JOB_VIEWED,
instance=self.job,
actor_id=self.user.id)
def setup_auditor_services(self) -> None:
if not self.is_testing_env:
import activitylogs
import auditor
import executor
import notifier
import tracker
auditor.validate()
auditor.setup()
tracker.validate()
tracker.setup()
if self._enable_activitylogs:
activitylogs.validate()
activitylogs.setup()
if self._enable_notifier:
notifier.validate()
notifier.setup()
executor.validate()
executor.setup()
