vandonr-amz commented on code in PR #30853:
URL: https://github.com/apache/airflow/pull/30853#discussion_r1189024588
##########
tests/providers/amazon/aws/triggers/test_redshift_cluster.py:
##########
@@ -72,3 +76,143 @@ async def test_redshift_create_cluster_trigger_run(self,
mock_async_conn):
response = await generator.asend(None)
assert response == TriggerEvent({"status": "success", "message":
"Cluster Created"})
+
+
+class TestRedshiftPauseClusterTrigger:
+ def test_redshift_resume_cluster_trigger_serialize(self):
+ redshift_resume_cluster_trigger = RedshiftPauseClusterTrigger(
+ cluster_identifier=TEST_CLUSTER_IDENTIFIER,
+ poll_interval=TEST_POLL_INTERVAL,
+ max_attempts=TEST_MAX_ATTEMPT,
+ aws_conn_id=TEST_AWS_CONN_ID,
+ )
+ class_path, args = redshift_resume_cluster_trigger.serialize()
+ assert (
+ class_path ==
"airflow.providers.amazon.aws.triggers.redshift_cluster.RedshiftPauseClusterTrigger"
+ )
+ assert args["cluster_identifier"] == TEST_CLUSTER_IDENTIFIER
+ assert args["poll_interval"] == str(TEST_POLL_INTERVAL)
+ assert args["max_attempts"] == str(TEST_MAX_ATTEMPT)
+ assert args["aws_conn_id"] == TEST_AWS_CONN_ID
+
+ @pytest.mark.asyncio
+
@async_mock.patch("airflow.providers.amazon.aws.hooks.redshift_cluster.RedshiftHook.get_waiter")
+
@async_mock.patch("airflow.providers.amazon.aws.hooks.redshift_cluster.RedshiftHook.async_conn")
Review Comment:
same remark as above for this and the other mocks in the file
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
