feat!: job/step/task API functions and session log assertions (#150)

Signed-off-by: Josh Usiskin <56369778+jusiskin@users.noreply.github.com> BREAKING CHANGE: Job.lifecycle_status changed to enum and TaskStatus.UNKNOWN removed
aws-deadline · Sep 5, 2024 · 18f7078 · 18f7078
1 parent 5f9b334
commit 18f7078
Show file tree

Hide file tree

Showing 4 changed files with 1,499 additions and 39 deletions.
diff --git a/src/deadline_test_fixtures/__init__.py b/src/deadline_test_fixtures/__init__.py
@@ -17,6 +17,10 @@
     PipInstall,
     Queue,
     QueueFleetAssociation,
+    Session,
+    SessionLog,
+    Step,
+    Task,
     TaskStatus,
 )
 from .fixtures import (
@@ -49,30 +53,34 @@
     "CloudWatchLogEvent",
     "CodeArtifactRepositoryInfo",
     "CommandResult",
-    "DeadlineResources",
     "DeadlineClient",
+    "DeadlineResources",
     "DeadlineWorker",
     "DeadlineWorkerConfiguration",
     "DockerContainerWorker",
     "EC2InstanceWorker",
-    "WindowsInstanceWorkerBase",
-    "WindowsInstanceBuildWorker",
-    "PosixInstanceWorkerBase",
-    "PosixInstanceBuildWorker",
     "Farm",
     "Fleet",
     "Job",
-    "JobAttachmentSettings",
     "JobAttachmentManager",
+    "JobAttachmentSettings",
     "JobRunAsUser",
+    "OperatingSystem",
     "PipInstall",
+    "PosixInstanceBuildWorker",
+    "PosixInstanceWorkerBase",
     "PosixSessionUser",
-    "S3Object",
-    "ServiceModel",
-    "OperatingSystem",
     "Queue",
     "QueueFleetAssociation",
+    "S3Object",
+    "ServiceModel",
+    "Session",
+    "SessionLog",
+    "Step",
+    "Task",
     "TaskStatus",
+    "WindowsInstanceBuildWorker",
+    "WindowsInstanceWorkerBase",
     "bootstrap_resources",
     "codeartifact",
     "deadline_client",
@@ -81,6 +89,6 @@
     "install_service_model",
     "service_model",
     "version",
-    "worker",
     "worker_config",
+    "worker",
 ]
diff --git a/src/deadline_test_fixtures/deadline/__init__.py b/src/deadline_test_fixtures/deadline/__init__.py
@@ -1,15 +1,29 @@
 # Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 
+from .client import DeadlineClient
 from .resources import (
     CloudWatchLogEvent,
+    DependencyCounts,
     Farm,
     Fleet,
+    FloatTaskParameterValue,
+    IntTaskParameterValue,
     Job,
+    JobLifecycleStatus,
+    LogConfiguration,
+    PathTaskParameterValue,
     Queue,
     QueueFleetAssociation,
+    Session,
+    SessionLifecycleStatus,
+    SessionLog,
+    Step,
+    StepLifecycleStatus,
+    StringTaskParameterValue,
+    Task,
+    TaskParameterValue,
     TaskStatus,
 )
-from .client import DeadlineClient
 from .worker import (
     CommandResult,
     DeadlineWorker,
@@ -29,17 +43,31 @@
     "DeadlineClient",
     "DeadlineWorker",
     "DeadlineWorkerConfiguration",
+    "DependencyCounts",
     "DockerContainerWorker",
     "EC2InstanceWorker",
-    "WindowsInstanceWorkerBase",
-    "WindowsInstanceBuildWorker",
-    "PosixInstanceWorkerBase",
-    "PosixInstanceBuildWorker",
     "Farm",
     "Fleet",
+    "FloatTaskParameterValue",
+    "IntTaskParameterValue",
     "Job",
+    "JobLifecycleStatus",
+    "LogConfiguration",
+    "PathTaskParameterValue",
     "PipInstall",
+    "PosixInstanceBuildWorker",
+    "PosixInstanceWorkerBase",
     "Queue",
     "QueueFleetAssociation",
+    "Session",
+    "SessionLifecycleStatus",
+    "SessionLog",
+    "Step",
+    "StepLifecycleStatus",
+    "StringTaskParameterValue",
+    "Task",
+    "TaskParameterValue",
     "TaskStatus",
+    "WindowsInstanceBuildWorker",
+    "WindowsInstanceWorkerBase",
 ]