def test_round_trip_multi_subscription(self):
store_file = tempfile.NamedTemporaryFile(delete=False)
store_file.close()
principals = [
('12345', '67890', 'foobar'),
('abcde', 'fghij', 'foobaz'),
]
# Store them all
for principal in principals:
store_acs_service_principal(
principal[0], principal[1], principal[2], config_path=store_file.name)
# Make sure it worked
for principal in principals:
obj = load_acs_service_principal(principal[0], config_path=store_file.name)
self.assertIsNotNone(obj, 'expected non-None for {}'.format(principal[0]))
self.assertEqual(obj.get('service_principal'), principal[2])
self.assertEqual(obj.get('client_secret'), principal[1])
# Change one
new_principal = 'foo'
new_secret = 'bar'
store_acs_service_principal(
principals[0][0], new_secret, new_principal, config_path=store_file.name)
obj = load_acs_service_principal(principals[0][0], config_path=store_file.name)
self.assertIsNotNone(obj, 'expected non-None for {}'.format(principals[0][0]))
self.assertEqual(obj.get('service_principal'), new_principal)
self.assertEqual(obj.get('client_secret'), new_secret)
os.remove(store_file.name)
def test_round_trip_one_subscription(self):
store_file = tempfile.NamedTemporaryFile(delete=False)
store_file.close()
service_principal = '12345'
sub_id = '67890'
client_secret = 'foobar'
store_acs_service_principal(
sub_id, client_secret, service_principal, config_path=store_file.name)
obj = load_acs_service_principal(sub_id, config_path=store_file.name)
self.assertIsNotNone(obj)
self.assertEqual(obj.get('service_principal'), service_principal)
self.assertEqual(obj.get('client_secret'), client_secret)
os.remove(store_file.name)
