def testAddToApisMap_NoDefault(self):
apis.AddToApisMap('hello', 'v1')
self.assertTrue(
apis_internal._GetApiDef('hello', 'v1').default_version)
apis.AddToApisMap('hello', 'v2')
self.assertFalse(
apis_internal._GetApiDef('hello', 'v2').default_version)
self.assertTrue(
apis_internal._GetApiDef('hello', 'v1').default_version)
def SetUp(self):
self.StartObjectPatch(apis_map, 'MAP', {})
apis.AddToApisMap('compute', 'v1', True)
apis.AddToApisMap('compute', 'alpha', False)
apis.AddToApisMap('dns', 'v1', True)
apis.AddToApisMap('dns', 'v1beta1', False)
apis.AddToApisMap('sql', 'v1beta4', True)
self.StartObjectPatch(apis_util, '_API_NAME_ALIASES', {})
def testAddToApisMap(self):
with self.assertRaisesRegex(
apis_util.UnknownAPIError,
r'API named \[hello\] does not exist in the APIs map'):
apis_internal._GetApiDef('hello', 'v1')
expected_api_def = apis.ConstructApiDef('hello', 'v1', True)
apis.AddToApisMap('hello', 'v1', True)
self.assertEqual('v1', apis_internal._GetDefaultVersion('hello'))
self.assertEqual(expected_api_def,
apis_internal._GetApiDef('hello', 'v1'))