Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
fix for ut
Browse files Browse the repository at this point in the history
  • Loading branch information
knakazawa99 committed Mar 17, 2024
1 parent 5a4fd2d commit 12781be
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions tests/test_glue_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,9 @@ async def test_fetch_result(aws_credentials, glue_job_client):
JobName="test_job_name",
Arguments={},
)["JobRunId"]
glue_job_run = GlueJobRun(job_name="test_job_name", job_id="test_job_name")
glue_job_run.job_id = job_run_id
glue_job_run.client = glue_job_client
glue_job_run = GlueJobRun(
job_name="test_job_name", job_id=job_run_id, client=glue_job_client
)
result = await glue_job_run.fetch_result()
assert result == "SUCCEEDED"

Expand All @@ -42,6 +42,7 @@ def test_wait_for_completion(aws_credentials, glue_job_client):
job_name="test_job_name",
job_id=job_run_id,
job_watch_poll_interval=0.1,
client=glue_job_client,
)

glue_job_client.get_job_run = MagicMock(
Expand All @@ -60,8 +61,6 @@ def test_wait_for_completion(aws_credentials, glue_job_client):
},
]
)
glue_job_run.client = glue_job_client
glue_job_run.job_id = job_run_id
glue_job_run.wait_for_completion()


Expand Down

0 comments on commit 12781be

Please sign in to comment.