def test_project_config_constructor(self, project_name, bucket_name, expected_bucket_name): """Test the direct construction of ProjectConfig.""" test_config = common.ProjectConfig(project_name, 'test_client_id', 'test_client_secret', bucket_name) self.assertEqual(project_name, test_config.project) self.assertEqual('test_client_id', test_config.client_id) self.assertEqual('test_client_secret', test_config.client_secret) self.assertEqual(expected_bucket_name, test_config.bucket) self.assertEqual('{}/configs'.format(expected_bucket_name), test_config.configs) # Test that two objects with the same constructor args are equal. self.assertEqual( test_config, common.ProjectConfig(project_name, 'test_client_id', 'test_client_secret', bucket_name)) # Test that two object with different constructor args are not equal. self.assertNotEqual( test_config, common.ProjectConfig(project_name, 'test_client_id', 'test_client_secret', 'INCORRECT')) # Test the string and representations for ProjectConfig. self.assertEqual('ProjectConfig for project: test_project.', str(test_config)) self.assertEqual( '<ProjectConfig({}, test_client_id, test_client_secret, {})>'. format(project_name, expected_bucket_name), repr(test_config))
def test_project_config_constructor(self, project_name, bucket_name, expected_bucket_name): """Test the direct construction of ProjectConfig.""" test_config = common.ProjectConfig('dev', project_name, 'test_client_id', 'test_client_secret', bucket_name, '/this/config/file.yaml') self.assertEqual(project_name, test_config.project) self.assertEqual('test_client_id', test_config.client_id) self.assertEqual('test_client_secret', test_config.client_secret) self.assertEqual(expected_bucket_name, test_config.bucket) self.assertEqual('configs/constants.json', test_config.constants_storage_path) self.assertEqual('/this/config/file.yaml', test_config.path) # Test that two objects with the same constructor args are equal. self.assertEqual( test_config, common.ProjectConfig('dev', project_name, 'test_client_id', 'test_client_secret', bucket_name, '/this/config/file.yaml')) # Test that two object with different constructor args are not equal. self.assertNotEqual( test_config, common.ProjectConfig('dev', project_name, 'test_client_id', 'test_client_secret', 'INCORRECT', '/this/config/file.yaml')) # Test the string and representations for ProjectConfig. self.assertEqual("ProjectConfig for project 'test_project'.", str(test_config)) self.assertEqual( '<ProjectConfig(dev, {}, test_client_id, test_client_secret, {}, ' '/this/config/file.yaml)>'.format(project_name, expected_bucket_name), repr(test_config))
def test_manager_init(self): test_config = common.ProjectConfig('KEY', 'PROJECT', 'ID', 'SECRET', 'BUCKET', self._blank_config_path) test_constants = app_constants.get_default_constants() test_manager = gng_impl._Manager(test_config, test_constants, None) self.assertEqual(str(test_manager), "_Manager for project 'PROJECT'") self.assertEqual(repr(test_manager), '<_Manager.new(/testdata/blank_config.yaml, KEY)>')
def test_project_config_from_yaml(self): """Test the construction of ProjectConfig when loaded from a config file.""" expected_config = common.ProjectConfig('test_project', 'test_client_id', 'test_client_secret', None) config_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'config.yaml') self.fs.CreateFile(config_file, contents=_CONFIG) test_config = common.ProjectConfig.from_yaml('dev', config_file) self.assertEqual(expected_config, test_config)
def setUp(self): super(AuthTest, self).setUp() self._test_project = 'test_project' self._test_client_id = 'test_client_id' self._test_client_secret = 'test_client_secret' self._test_config = common.ProjectConfig( 'test_key', self._test_project, self._test_client_id, self._test_client_secret, None, '/test/path.yaml') # Save the real modules for clean up. self.real_open = builtins.open # Create a fake file system and stub out builtin modules. self.fs = fake_filesystem.FakeFilesystem() self.os = fake_filesystem.FakeOsModule(self.fs) self.open = fake_filesystem.FakeFileOpen(self.fs) self.stubs = mox3_stubout.StubOutForTesting() self.stubs.SmartSet(builtins, 'open', self.open) self.stubs.SmartSet(auth, 'os', self.os)
def test_write(self): expected_dict = { 'project_id': 'test_project', 'client_id': 'test_client_id', 'client_secret': 'test_client_secret', 'bucket': 'test_project.appspot.com', } self.fs.CreateFile('/this/config/file.yaml', contents=_EMPTY_CONFIG) test_config = common.ProjectConfig( 'asdf', 'test_project', 'test_client_id', 'test_client_secret', None, '/this/config/file.yaml', ) test_config.write() with open('/this/config/file.yaml') as config_file: config = yaml.safe_load(config_file) self.assertEqual(config['asdf'], expected_dict)
def test_from_prompts(self): expected_response = common.ProjectConfig( 'test', 'fake_project', 'fake_client_id', 'fake_client_secret', None, '/this/config/file.yaml', ) prompt_return = [ 'fake_project', 'fake_client_id', 'fake_client_secret' ] with mock.patch.object(utils, 'prompt_string', side_effect=prompt_return) as mock_prompt: self.assertEqual( common.ProjectConfig.from_prompts('test', '/this/config/file.yaml'), expected_response) self.assertEqual(mock_prompt.call_count, 3)
def setUp(self): super(AppEngineTest, self).setUp() self.config = common.ProjectConfig( 'TEST_PROJECT', 'TEST_CLIENT_ID', 'TEST_CLIENT_SECRET', 'TEST_BUCKET')
def setUp(self): super(GoogleApiTest, self).setUp() self.config = common.ProjectConfig('TEST_KEY', 'TEST_PROJECT', 'TEST_CLIENT_ID', 'TEST_CLIENT_SECRET', 'TEST_BUCKET', '/this/path.yaml')
def setUp(self): super(CloudStorageAPITest, self).setUp() self.config = common.ProjectConfig('TEST_PROJECT', 'TEST_CLIENT_ID', 'TEST_CLIENT_SECRET', 'TEST_BUCKET')
def setUp(self): super(DatastoreAPITest, self).setUp() self.config = common.ProjectConfig('TEST_KEY', 'TEST_PROJECT', 'TEST_CLIENT_ID', 'TEST_CLIENT_SECRET', 'TEST_BUCKET', '/test/path.yaml')