File Manager

X7ROOT File Manager

Current Path: /opt/gsutil/gslib/tests

📁 ..
📄 __init__.py(808 B)
📄 mock_cloud_api.py(7.97 KB)
📄 mock_logging_handler.py(1.28 KB)
📄 rewrite_helper.py(2.8 KB)
📄 signurl_signatures.py(5.7 KB)
📄 test_Doption.py(9.78 KB)
📄 test_acl.py(55.99 KB)
📄 test_autoclass.py(6.85 KB)
📄 test_boto_util.py(9.65 KB)
📄 test_bucketconfig.py(4.98 KB)
📄 test_bucketpolicyonly.py(3.78 KB)
📄 test_cat.py(11.55 KB)
📄 test_cloud_api_delegator.py(2 KB)
📄 test_command.py(3.39 KB)
📄 test_command_runner.py(20.66 KB)
📄 test_compose.py(14.12 KB)
📄 test_context_config.py(18.78 KB)
📄 test_copy_helper_funcs.py(39.76 KB)
📄 test_copy_objects_iterator.py(4.49 KB)
📄 test_cors.py(12.45 KB)
📄 test_cp.py(215.86 KB)
📄 test_creds_config.py(8.64 KB)
📄 test_daisy_chain_wrapper.py(14.13 KB)
📁 test_data
📄 test_defacl.py(14.36 KB)
📄 test_defstorageclass.py(5.43 KB)
📄 test_du.py(10.61 KB)
📄 test_encryption_helper.py(4.62 KB)
📄 test_execution_util.py(3.88 KB)
📄 test_file_part.py(3.38 KB)
📄 test_gcs_json_api.py(2.9 KB)
📄 test_gcs_json_credentials.py(9.85 KB)
📄 test_gcs_json_media.py(7.44 KB)
📄 test_gsutil.py(4.68 KB)
📄 test_hash.py(9.58 KB)
📄 test_hashing_helper.py(10.78 KB)
📄 test_help.py(3.5 KB)
📄 test_hmac.py(23.9 KB)
📄 test_iam.py(90.67 KB)
📄 test_kms.py(16.68 KB)
📄 test_label.py(11.55 KB)
📄 test_lifecycle.py(13.8 KB)
📄 test_logging.py(3.5 KB)
📄 test_ls.py(53.04 KB)
📄 test_mb.py(23.29 KB)
📄 test_metrics.py(51.65 KB)
📄 test_mtls.py(2.01 KB)
📄 test_mv.py(13.08 KB)
📄 test_naming.py(63.11 KB)
📄 test_notification.py(5.9 KB)
📄 test_notification_pubsub.py(5.46 KB)
📄 test_pap.py(5.91 KB)
📄 test_parallel_cp.py(10.15 KB)
📄 test_parallelism_framework.py(33.09 KB)
📄 test_perfdiag.py(12.62 KB)
📄 test_plurality_checkable_iterator.py(7.53 KB)
📄 test_posix_util.py(2.03 KB)
📄 test_psc.py(5.88 KB)
📄 test_rb.py(2.93 KB)
📄 test_requester_pays.py(11.7 KB)
📄 test_resumable_streaming.py(12.36 KB)
📄 test_retention.py(28.9 KB)
📄 test_retention_util.py(5.49 KB)
📄 test_rewrite.py(31.62 KB)
📄 test_rm.py(33.72 KB)
📄 test_rpo.py(10.22 KB)
📄 test_rsync.py(149.64 KB)
📄 test_rsync_funcs.py(3.36 KB)
📄 test_seek_ahead_thread.py(8.79 KB)
📄 test_setmeta.py(12.54 KB)
📄 test_shim_util.py(64.19 KB)
📄 test_signurl.py(24.69 KB)
📄 test_stat.py(11.38 KB)
📄 test_stet_cp.py(5.77 KB)
📄 test_stet_util.py(7.38 KB)
📄 test_storage_url.py(7.02 KB)
📄 test_tabcomplete.py(14.31 KB)
📄 test_temporary_file_util.py(1.54 KB)
📄 test_trace.py(1.76 KB)
📄 test_tracker_file.py(9.9 KB)
📄 test_ubla.py(3.88 KB)
📄 test_ui.py(67.42 KB)
📄 test_update.py(10.37 KB)
📄 test_user_agent_helper.py(5.34 KB)
📄 test_util.py(19.85 KB)
📄 test_versioning.py(3.61 KB)
📄 test_web.py(6.54 KB)
📄 test_wildcard_iterator.py(22.18 KB)
📄 test_wrapped_credentials.py(13.95 KB)
📁 testcase
📄 util.py(29.01 KB)

Editing: test_user_agent_helper.py

# -*- 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 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")

@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$")

X7ROOT File Manager

Editing: test_user_agent_helper.py

Upload File

Create Folder