def _SetupBotoConfig(self): """Set the boto config so GSUtil works with provisioned service accounts.""" project_id = self._GetNumericProjectId() try: boto_config.BotoConfig(project_id, debug=self.debug) except (IOError, OSError) as e: self.logger.warning(str(e))
def testCreateConfig(self, mock_config, mock_logger, mock_watcher): mock_config_instance = mock.Mock() mock_config.ConfigManager.return_value = mock_config_instance mocks = mock.Mock() mocks.attach_mock(mock_config.ConfigManager, 'config') mocks.attach_mock(mock_config_instance.SetOption, 'set') mocks.attach_mock(mock_config_instance.WriteConfig, 'write') mocks.attach_mock(mock_logger, 'logger') mocks.attach_mock(mock_watcher, 'watcher') mock_logger_instance = mock.Mock() mock_logger.Logger.return_value = mock_logger_instance boto_config.BotoConfig(self.project_id) expected_calls = [ mock.call.logger.Logger(name=mock.ANY), mock.call.watcher.MetadataWatcher(logger=mock_logger_instance), mock.call.config(config_file='template', config_header='/tmp/test.py template'), mock.call.set('GSUtil', 'default_project_id', self.project_id), mock.call.set('GSUtil', 'default_api_version', '2'), mock.call.set('GoogleCompute', 'service_account', 'default'), mock.call.set('Plugin', 'plugin_directory', '/tmp'), mock.call.write(config_file='config'), ] self.assertEqual(mocks.mock_calls, expected_calls)
def testCreateConfigExit(self, mock_config, mock_watcher): mock_config_instance = mock.Mock() mock_config.ConfigManager.return_value = mock_config_instance mock_watcher_instance = mock.Mock() mock_watcher.MetadataWatcher.return_value = mock_watcher_instance mock_watcher_instance.GetMetadata.return_value = None boto_config.BotoConfig() mock_watcher_instance.GetMetadata.assert_called_once_with( metadata_key='project/numeric-project-id', recursive=False) mock_config.SetOption.assert_not_called()
def testCreateConfigProjectId(self, mock_config, mock_watcher): mock_config_instance = mock.Mock() mock_config.ConfigManager.return_value = mock_config_instance mock_watcher_instance = mock.Mock() mock_watcher.MetadataWatcher.return_value = mock_watcher_instance mock_watcher_instance.GetMetadata.return_value = self.project_id boto_config.BotoConfig() mock_watcher_instance.GetMetadata.assert_called_once_with( metadata_key='project/numeric-project-id', recursive=False) expected_calls = [ mock.call('GSUtil', 'default_project_id', self.project_id), ] mock_config_instance.SetOption.assert_has_calls(expected_calls)