File Manager

✏️ /

# -*- coding: utf-8 -*-
# Copyright 2014 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Unit tests for hashing helper functions and classes."""

from gslib.utils import system_util
from gslib.utils.user_agent_helper import GetUserAgent
import gslib.tests.testcase as testcase

import six
from six import add_move, MovedModule

add_move(MovedModule('mock', 'mock', 'unittest.mock'))
from six.moves import mock

class TestUserAgentHelper(testcase.GsUtilUnitTestCase):
  """Unit tests for the GetUserAgent helper function."""

@mock.patch('gslib.VERSION', '4_test')
  def testNoArgs(self):
    self.assertRegex(GetUserAgent([]), r"^ gsutil/4_test $[^$]+\)")

def testAnalyticsFlag(self):
    self.assertRegex(GetUserAgent([], False), r"analytics/enabled")
    self.assertRegex(GetUserAgent([], True), r"analytics/disabled")

@mock.patch.object(system_util, 'IsRunningInteractively')
  def testInteractiveFlag(self, mock_interactive):
    mock_interactive.return_value = True
    self.assertRegex(GetUserAgent([]), r"interactive/True")
    mock_interactive.return_value = False
    self.assertRegex(GetUserAgent([]), r"interactive/False")

def testHelp(self):
    self.assertRegex(GetUserAgent(['help']), r"command/help")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCp(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp', '-r', '-Z', '1.txt', 'gs://dst']),
                     r"command/cp$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCpWithMultiprocessing(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp', '-m', '1.txt', 'gs://dst']),
                     r"command/cp-m$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCpNotEnoughArgs(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp']), r"command/cp$")
    self.assertRegex(GetUserAgent(['cp', '1.txt']), r"command/cp$")
    self.assertRegex(GetUserAgent(['cp', '-r', '1.ts']), r"command/cp$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCpEncoding(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp', 'öne', 'twö']), r"command/cp$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testRsync(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['rsync', '1.txt', 'gs://dst']),
                     r"command/rsync$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testMv(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['mv', 'gs://src/1.txt', 'gs://dst/1.txt']),
                     r"command/mv$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCpCloudToCloud(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp', '-r', 'gs://src', 'gs://dst']),
                     r"command/cp$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCpForcedDaisyChain(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp', '-D', 'gs://src', 'gs://dst']),
                     r"command/cp$")

def testCpDaisyChain(self):
    self.assertRegex(GetUserAgent(['cp', '-r', '-Z', 'gs://src', 's3://dst']),
                     r"command/cp-DaisyChain")
    self.assertRegex(GetUserAgent(['mv', 'gs://src/1.txt', 's3://dst/1.txt']),
                     r"command/mv-DaisyChain")
    self.assertRegex(GetUserAgent(['rsync', '-r', 'gs://src', 's3://dst']),
                     r"command/rsync-DaisyChain")

def testCpDaisyChainWithMultiprocessing(self):
    self.assertRegex(
        GetUserAgent(['cp', '-r', '-Z', '-m', 'gs://src', 's3://dst']),
        r"command/cp-DaisyChain-m")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testPassOnInvalidUrlError(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['cp', '-r', '-Z', 'bad://src', 's3://dst']),
                     r"command/cp$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testRewriteEncryptionKey(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['rewrite', '-k', 'gs://dst']),
                     r"command/rewrite-k$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testRewriteStorageClass(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['rewrite', '-s', 'gs://dst']),
                     r"command/rewrite-s$")

@mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testRewriteEncryptionKeyAndStorageClass(self, mock_invoked):
    mock_invoked.return_value = False
    self.assertRegex(GetUserAgent(['rewrite', '-k', '-s', 'gs://dst']),
                     r"command/rewrite-k-s$")

@mock.patch.object(system_util, 'CloudSdkVersion')
  @mock.patch.object(system_util, 'InvokedViaCloudSdk')
  def testCloudSdk(self, mock_invoked, mock_version):
    mock_invoked.return_value = True
    mock_version.return_value = '500.1'
    self.assertRegex(GetUserAgent(['help']), r"google-cloud-sdk/500.1$")
    mock_invoked.return_value = False
    mock_version.return_value = '500.1'
    self.assertRegex(GetUserAgent(['help']), r"command/help$")

Batal

⬅ gslib 91 item

Nama	Tipe	Ukuran	Diubah	Aksi
📁test_data	dir	—	2025-06-25 08:19	📂 Buka
📁testcase	dir	—	2025-06-25 08:19	📂 Buka
🐍__init__.py	py	808 B	2025-06-25 08:19	✏️ 👁️
🐍mock_cloud_api.py	py	8 KB	2025-06-25 08:19	✏️ 👁️
🐍mock_logging_handler.py	py	1.3 KB	2025-06-25 08:19	✏️ 👁️
🐍rewrite_helper.py	py	2.8 KB	2025-06-25 08:19	✏️ 👁️
🐍signurl_signatures.py	py	5.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_Doption.py	py	9.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_acl.py	py	56 KB	2025-06-25 08:19	✏️ 👁️
🐍test_autoclass.py	py	6.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_boto_util.py	py	9.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_bucketconfig.py	py	5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_bucketpolicyonly.py	py	3.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_cat.py	py	11.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_cloud_api_delegator.py	py	2 KB	2025-06-25 08:19	✏️ 👁️
🐍test_command.py	py	3.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_command_runner.py	py	20.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_compose.py	py	14.1 KB	2025-06-25 08:19	✏️ 👁️
🐍test_context_config.py	py	18.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_copy_helper_funcs.py	py	39.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_copy_objects_iterator.py	py	4.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_cors.py	py	12.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_cp.py	py	215.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_creds_config.py	py	8.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_daisy_chain_wrapper.py	py	14.1 KB	2025-06-25 08:19	✏️ 👁️
🐍test_defacl.py	py	14.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_defstorageclass.py	py	5.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_du.py	py	10.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_encryption_helper.py	py	4.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_execution_util.py	py	3.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_file_part.py	py	3.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_gcs_json_api.py	py	2.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_gcs_json_credentials.py	py	9.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_gcs_json_media.py	py	7.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_gsutil.py	py	4.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_hash.py	py	9.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_hashing_helper.py	py	10.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_help.py	py	3.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_hmac.py	py	23.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_iam.py	py	90.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_kms.py	py	16.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_label.py	py	11.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_lifecycle.py	py	13.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_logging.py	py	3.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_ls.py	py	53 KB	2025-06-25 08:19	✏️ 👁️
🐍test_mb.py	py	23.3 KB	2025-06-25 08:19	✏️ 👁️
🐍test_metrics.py	py	51.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_mtls.py	py	2 KB	2025-06-25 08:19	✏️ 👁️
🐍test_mv.py	py	13.1 KB	2025-06-25 08:19	✏️ 👁️
🐍test_naming.py	py	63.1 KB	2025-06-25 08:19	✏️ 👁️
🐍test_notification.py	py	5.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_notification_pubsub.py	py	5.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_pap.py	py	5.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_parallel_cp.py	py	10.1 KB	2025-06-25 08:19	✏️ 👁️
🐍test_parallelism_framework.py	py	33.1 KB	2025-06-25 08:19	✏️ 👁️
🐍test_perfdiag.py	py	12.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_plurality_checkable_iterator.py	py	7.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_posix_util.py	py	2 KB	2025-06-25 08:19	✏️ 👁️
🐍test_psc.py	py	5.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_rb.py	py	2.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_requester_pays.py	py	11.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_resumable_streaming.py	py	12.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_retention.py	py	28.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_retention_util.py	py	5.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_rewrite.py	py	31.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_rm.py	py	33.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_rpo.py	py	10.2 KB	2025-06-25 08:19	✏️ 👁️
🐍test_rsync.py	py	149.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_rsync_funcs.py	py	3.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_seek_ahead_thread.py	py	8.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_setmeta.py	py	12.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_shim_util.py	py	64.2 KB	2025-06-25 08:19	✏️ 👁️
🐍test_signurl.py	py	24.7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_stat.py	py	11.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_stet_cp.py	py	5.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_stet_util.py	py	7.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_storage_url.py	py	7 KB	2025-06-25 08:19	✏️ 👁️
🐍test_tabcomplete.py	py	14.3 KB	2025-06-25 08:19	✏️ 👁️
🐍test_temporary_file_util.py	py	1.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_trace.py	py	1.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_tracker_file.py	py	9.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_ubla.py	py	3.9 KB	2025-06-25 08:19	✏️ 👁️
🐍test_ui.py	py	67.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_update.py	py	10.4 KB	2025-06-25 08:19	✏️ 👁️
🐍test_user_agent_helper.py	py	5.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_util.py	py	19.8 KB	2025-06-25 08:19	✏️ 👁️
🐍test_versioning.py	py	3.6 KB	2025-06-25 08:19	✏️ 👁️
🐍test_web.py	py	6.5 KB	2025-06-25 08:19	✏️ 👁️
🐍test_wildcard_iterator.py	py	22.2 KB	2025-06-25 08:19	✏️ 👁️
🐍test_wrapped_credentials.py	py	14 KB	2025-06-25 08:19	✏️ 👁️
🐍util.py	py	29 KB	2025-06-25 08:19	✏️ 👁️

✏️ Rename