tools/telemetry/third_party/gsutilz/gslib/tests/test_creds_config.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright 2014 Google Inc. All Rights Reserved.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 """Tests for various combinations of configured credentials."""
  16
  17 from gslib.cred_types import CredTypes
  18 from gslib.exception import CommandException
  19 from gslib.gcs_json_api import GcsJsonApi
  20 from gslib.tests.mock_logging_handler import MockLoggingHandler
  21 import gslib.tests.testcase as testcase
  22 from gslib.tests.util import SetBotoConfigForTest
  23
  24
  25 class TestCredsConfig(testcase.GsUtilUnitTestCase):
  26   """Tests for various combinations of configured credentials."""
  27
  28   def setUp(self):
  29     super(TestCredsConfig, self).setUp()
  30     self.log_handler = MockLoggingHandler()
  31     self.logger.addHandler(self.log_handler)
  32
  33   def testMultipleConfiguredCreds(self):
  34     with SetBotoConfigForTest([
  35         ('Credentials', 'gs_oauth2_refresh_token', 'foo'),
  36         ('Credentials', 'gs_service_client_id', 'bar'),
  37         ('Credentials', 'gs_service_key_file', 'baz')]):
  38
  39       try:
  40         GcsJsonApi(None, self.logger)
  41         self.fail('Succeeded with multiple types of configured creds.')
  42       except CommandException, e:
  43         msg = str(e)
  44         self.assertIn('types of configured credentials', msg)
  45         self.assertIn(CredTypes.OAUTH2_USER_ACCOUNT, msg)
  46         self.assertIn(CredTypes.OAUTH2_SERVICE_ACCOUNT, msg)
  47
  48   def testExactlyOneInvalid(self):
  49     with SetBotoConfigForTest([
  50         ('Credentials', 'gs_oauth2_refresh_token', 'foo'),
  51         ('Credentials', 'gs_service_client_id', None),
  52         ('Credentials', 'gs_service_key_file', None)]):
  53       succeeded = False
  54       try:
  55         GcsJsonApi(None, self.logger)
  56         succeeded = True  # If we self.fail() here, the except below will catch
  57       except:  # pylint: disable=bare-except
  58         warning_messages = self.log_handler.messages['warning']
  59         self.assertEquals(1, len(warning_messages))
  60         self.assertIn('credentials are invalid', warning_messages[0])
  61         self.assertIn(CredTypes.OAUTH2_USER_ACCOUNT, warning_messages[0])
  62       if succeeded:
  63         self.fail('Succeeded with invalid credentials, one configured.')