Fix mocked describe_processing_job

ajaykarpur · ajaykarpur · commit 6d77969cf68d · 2019-12-19T13:49:47.000-08:00
diff --git a/tests/unit/test_processing.py b/tests/unit/test_processing.py
@@ -88,8 +88,8 @@ def sagemaker_session():
     )
     session_mock.download_data = Mock(name="download_data")
     session_mock.expand_role.return_value = ROLE
-    session_mock.sagemaker_client.describe_processing_job = MagicMock(
-        return_value=PROCESSING_JOB_DESCRIPTION
+    session_mock.describe_processing_job = MagicMock(
+        name="describe_processing_job", return_value=PROCESSING_JOB_DESCRIPTION
     )
     return session_mock
 

Original file line number	Diff line number	Diff line change
`@@ -88,8 +88,8 @@ def sagemaker_session():`
`88`	`88`	`)`
`89`	`89`	`session_mock.download_data = Mock(name="download_data")`
`90`	`90`	`session_mock.expand_role.return_value = ROLE`
`91`		`- session_mock.sagemaker_client.describe_processing_job = MagicMock(`
`92`		`- return_value=PROCESSING_JOB_DESCRIPTION`
	`91`	`+ session_mock.describe_processing_job = MagicMock(`
	`92`	`+ name="describe_processing_job", return_value=PROCESSING_JOB_DESCRIPTION`
`93`	`93`	`)`
`94`	`94`	`return session_mock`
`95`	`95`