@@ -528,26 +528,21 @@ def test_rewrite_exposes_lineage(self, mock_service, hook_lineage_collector):
528528
529529 @mock .patch ("google.cloud.storage.Bucket" )
530530 @mock .patch (GCS_STRING .format ("GCSHook.get_conn" ))
531- def test_delete (self , mock_service , mock_bucket , caplog ):
531+ def test_delete (self , mock_service , mock_bucket ):
532532 test_bucket = "test_bucket"
533533 test_object = "test_object"
534534 blob_to_be_deleted = storage .Blob (name = test_object , bucket = mock_bucket )
535535
536- bucket_method = mock_service .return_value .bucket
537- blob = bucket_method .return_value .blob
538- delete_method = blob .return_value .delete
536+ get_bucket_method = mock_service .return_value .get_bucket
537+ get_blob_method = get_bucket_method .return_value .get_blob
538+ delete_method = get_blob_method .return_value .delete
539539 delete_method .return_value = blob_to_be_deleted
540540
541- with caplog .at_level (logging .INFO ):
542- response = self .gcs_hook .delete (bucket_name = test_bucket , object_name = test_object )
541+ response = self .gcs_hook .delete (bucket_name = test_bucket , object_name = test_object )
543542 assert response is None
544- bucket_method .assert_called_once_with (test_bucket )
545- blob .assert_called_once_with (blob_name = test_object )
546- delete_method .assert_called_once ()
547- assert "Blob test_object has been deleted" in caplog .text
548543
549544 @mock .patch (GCS_STRING .format ("GCSHook.get_conn" ))
550- def test_delete_nonexisting_object (self , mock_service , caplog ):
545+ def test_delete_nonexisting_object (self , mock_service ):
551546 test_bucket = "test_bucket"
552547 test_object = "test_object"
553548
@@ -556,12 +551,8 @@ def test_delete_nonexisting_object(self, mock_service, caplog):
556551 delete_method = blob .return_value .delete
557552 delete_method .side_effect = NotFound (message = "Not Found" )
558553
559- with caplog . at_level ( logging . INFO ):
554+ with pytest . raises ( NotFound ):
560555 self .gcs_hook .delete (bucket_name = test_bucket , object_name = test_object )
561- bucket_method .assert_called_once_with (test_bucket )
562- blob .assert_called_once_with (blob_name = test_object )
563- delete_method .assert_called_once ()
564- assert "Blob test_object does not exist" in caplog .text
565556
566557 @mock .patch (GCS_STRING .format ("GCSHook.get_conn" ))
567558 def test_delete_exposes_lineage (self , mock_service , hook_lineage_collector ):
0 commit comments