CHG: caplog -> patch.object style

Author: kyungjunleeme

providers/microsoft/azure/tests/unit/microsoft/azure/transfers/test_sftp_to_wasb.py

@@ -59,34 +59,45 @@ def test_init(self):
         assert operator.create_container is False
 
     @pytest.mark.skipif(not AIRFLOW_V_3_0_PLUS, reason="dry_run only exists in Airflow 3.0+")
-    @mock.patch("airflow.providers.microsoft.azure.transfers.sftp_to_wasb.WasbHook")
-    @mock.patch("airflow.providers.microsoft.azure.transfers.sftp_to_wasb.SFTPHook")
-    def test_dry_run_logs_and_skips_real_action(self, mock_sftp_hook, mock_wasb_hook, caplog):
-        mock_sftp_hook.return_value.get_tree_map.return_value = [
-            ["main_dir/test_object3.json"],  # files
-            [],  # dirs
-            [],  # links
-        ]
+    @pytest.mark.parametrize(
+        "move_object, expect_delete_log",
+        [
+            (True, True),
+            (False, False),
+        ],
+    )
+    def test_dry_run_logs_and_skips_real_action(self, move_object, expect_delete_log):
         operator = SFTPToWasbOperator(
             task_id=TASK_ID,
             sftp_source_path=SOURCE_PATH_NO_WILDCARD,
             sftp_conn_id=SFTP_CONN_ID,
             container_name=CONTAINER_NAME,
             wasb_conn_id=WASB_CONN_ID,
             blob_prefix="sponge-bob/",
-            move_object=True,
+            move_object=move_object,
         )
+        with (
+            mock.patch.object(operator.log, "info") as mock_info,
+            mock.patch("airflow.providers.microsoft.azure.transfers.sftp_to_wasb.WasbHook"),
+            mock.patch("airflow.providers.microsoft.azure.transfers.sftp_to_wasb.SFTPHook") as mock_sftp_hook,
+        ):
+            mock_sftp_hook.return_value.get_tree_map.return_value = [
+                ["main_dir/test_object3.json"],
+                [],
+                [],
+            ]
 
-        caplog.clear()
-        with caplog.at_level("INFO"):
             operator.dry_run()
 
-        assert "Process will upload file from (SFTP) main_dir/test_object3.json" in caplog.text
-        assert "as sponge-bob/test_object3.json" in caplog.text
-        assert "Executing delete of" in caplog.text
+        logged_messages = [call.args[0] for call in mock_info.call_args_list]
+        assert "Dry run" in logged_messages
+        assert "Process will upload file from (SFTP) %s to wasb://%s as %s" in logged_messages
 
-        assert not mock_wasb_hook.return_value.load_file.called
-        assert not mock_sftp_hook.return_value.delete_file.called
+        delete_log = "Executing delete of %s"
+        if expect_delete_log:
+            assert delete_log in logged_messages
+        else:
+            assert delete_log not in logged_messages
 
     @mock.patch("airflow.providers.microsoft.azure.transfers.sftp_to_wasb.AIRFLOW_V_3_0_PLUS", False)
     def test_dry_run_raises_not_implemented(mock_version_tuple):