Skip to content

Commit

Permalink
35: params_dict in JobSubmission should be struct instead of dict in …
Browse files Browse the repository at this point in the history
…test_main.py
  • Loading branch information
lfse-slafleur committed May 6, 2024
1 parent c5fb765 commit 69deb3d
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions unit_test/test_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
from omotes_sdk.workflow_type import WorkflowType
from omotes_sdk_protocol.job_pb2 import JobSubmission
from google.protobuf import json_format
from google.protobuf.struct_pb2 import Struct

from omotes_orchestrator.config import OrchestratorConfig
from omotes_orchestrator.db_models.job import JobStatus
Expand Down Expand Up @@ -188,7 +189,7 @@ def test__new_job_submitted_handler__fully_new_job(self) -> None:
timeout = 3000
workflow_type = "some-workflow"
esdl = "Some-esdl"
params_dict = {}
params_dict = Struct()
job_submission = JobSubmission(
uuid=str(job_id),
timeout_ms=timeout,
Expand Down Expand Up @@ -248,7 +249,7 @@ def test__new_job_submitted_handler__already_registered_but_not_submitted_new_jo
timeout = 3000
workflow_type = "some-workflow"
esdl = "Some-esdl"
params_dict = {}
params_dict = Struct()
job_submission = JobSubmission(
uuid=str(job_id),
timeout_ms=timeout,
Expand Down Expand Up @@ -302,7 +303,7 @@ def test__new_job_submitted_handler__already_registered_and_submitted_new_job(se
timeout = 3000
workflow_type = "some-workflow"
esdl = "Some-esdl"
params_dict = {}
params_dict = Struct()
job_submission = JobSubmission(
uuid=str(job_id),
timeout_ms=timeout,
Expand Down Expand Up @@ -351,7 +352,7 @@ def test__new_job_submitted_handler__already_running_new_job(self) -> None:
timeout = 3000
workflow_type = "some-workflow"
esdl = "Some-esdl"
params_dict = {}
params_dict = Struct()
job_submission = JobSubmission(
uuid=str(job_id),
timeout_ms=timeout,
Expand Down

0 comments on commit 69deb3d

Please sign in to comment.