def test_check_for_file(self, mock_service): mock_instance = mock_service.return_value mock_instance.exists.return_value = True hook = AzureFileShareHook(azure_fileshare_conn_id='azure_fileshare_extras') assert hook.check_for_file('share', 'directory', 'file', timeout=3) mock_instance.exists.assert_called_once_with('share', 'directory', 'file', timeout=3)