class ManagementCertificateManagementServiceTest(AzureTestCase):
def setUp(self):
self.sms = ServiceManagementService(
credentials.getSubscriptionId(),
credentials.getManagementCertFile())
set_service_options(self.sms)
self.certificate_thumbprints = []
def tearDown(self):
for thumbprint in self.certificate_thumbprints:
try:
self.sms.delete_management_certificate(thumbprint)
except:
pass
#--Helpers-----------------------------------------------------------------
def _create_management_certificate(self, cert):
self.certificate_thumbprints.append(cert.thumbprint)
result = self.sms.add_management_certificate(cert.public_key,
cert.thumbprint,
cert.data)
self.assertIsNone(result)
def _management_certificate_exists(self, thumbprint):
try:
props = self.sms.get_management_certificate(thumbprint)
return props is not None
except:
return False
#--Test cases for management certificates ----------------------------
def test_list_management_certificates(self):
# Arrange
local_cert = _local_certificate()
self._create_management_certificate(local_cert)
# Act
result = self.sms.list_management_certificates()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
cert = None
for temp in result:
if temp.subscription_certificate_thumbprint == \
local_cert.thumbprint:
cert = temp
break
self.assertIsNotNone(cert)
self.assertIsNotNone(cert.created)
self.assertEqual(cert.subscription_certificate_public_key,
local_cert.public_key)
self.assertEqual(cert.subscription_certificate_data, local_cert.data)
self.assertEqual(cert.subscription_certificate_thumbprint,
local_cert.thumbprint)
def test_get_management_certificate(self):
# Arrange
local_cert = _local_certificate()
self._create_management_certificate(local_cert)
# Act
result = self.sms.get_management_certificate(local_cert.thumbprint)
# Assert
self.assertIsNotNone(result)
self.assertIsNotNone(result.created)
self.assertEqual(result.subscription_certificate_public_key,
local_cert.public_key)
self.assertEqual(result.subscription_certificate_data, local_cert.data)
self.assertEqual(result.subscription_certificate_thumbprint,
local_cert.thumbprint)
def test_add_management_certificate(self):
# Arrange
local_cert = _local_certificate()
# Act
self.certificate_thumbprints.append(local_cert.thumbprint)
result = self.sms.add_management_certificate(local_cert.public_key,
local_cert.thumbprint,
local_cert.data)
# Assert
self.assertIsNone(result)
self.assertTrue(
self._management_certificate_exists(local_cert.thumbprint))
def test_delete_management_certificate(self):
# Arrange
local_cert = _local_certificate()
self._create_management_certificate(local_cert)
# Act
result = self.sms.delete_management_certificate(local_cert.thumbprint)
# Assert
self.assertIsNone(result)
self.assertFalse(
self._management_certificate_exists(local_cert.thumbprint))
class ManagementCertificateManagementServiceTest(AzureTestCase):
def setUp(self):
proxy_host = credentials.getProxyHost()
proxy_port = credentials.getProxyPort()
self.sms = ServiceManagementService(credentials.getSubscriptionId(), credentials.getManagementCertFile())
if proxy_host:
self.sms.set_proxy(proxy_host, proxy_port)
self.management_certificate_name = getUniqueNameBasedOnCurrentTime('utmgmtcert')
def tearDown(self):
try:
self.sms.delete_management_certificate(self.management_certificate_name)
except: pass
#--Helpers-----------------------------------------------------------------
def _create_management_certificate(self, thumbprint):
result = self.sms.add_management_certificate(MANAGEMENT_CERT_PUBLICKEY, thumbprint, MANAGEMENT_CERT_DATA)
self.assertIsNone(result)
def _management_certificate_exists(self, thumbprint):
try:
props = self.sms.get_management_certificate(thumbprint)
return props is not None
except:
return False
#--Test cases for management certificates ----------------------------
def test_list_management_certificates(self):
# Arrange
self._create_management_certificate(self.management_certificate_name)
# Act
result = self.sms.list_management_certificates()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
cert = None
for temp in result:
if temp.subscription_certificate_thumbprint == self.management_certificate_name:
cert = temp
break
self.assertIsNotNone(cert)
self.assertIsNotNone(cert.created)
self.assertEqual(cert.subscription_certificate_public_key, MANAGEMENT_CERT_PUBLICKEY)
self.assertEqual(cert.subscription_certificate_data, MANAGEMENT_CERT_DATA)
self.assertEqual(cert.subscription_certificate_thumbprint, self.management_certificate_name)
def test_get_management_certificate(self):
# Arrange
self._create_management_certificate(self.management_certificate_name)
# Act
result = self.sms.get_management_certificate(self.management_certificate_name)
# Assert
self.assertIsNotNone(result)
self.assertIsNotNone(result.created)
self.assertEqual(result.subscription_certificate_public_key, MANAGEMENT_CERT_PUBLICKEY)
self.assertEqual(result.subscription_certificate_data, MANAGEMENT_CERT_DATA)
self.assertEqual(result.subscription_certificate_thumbprint, self.management_certificate_name)
def test_add_management_certificate(self):
# Arrange
public_key = MANAGEMENT_CERT_PUBLICKEY
data = MANAGEMENT_CERT_DATA
# Act
result = self.sms.add_management_certificate(public_key, self.management_certificate_name, data)
# Assert
self.assertIsNone(result)
self.assertTrue(self._management_certificate_exists(self.management_certificate_name))
def test_delete_management_certificate(self):
# Arrange
self._create_management_certificate(self.management_certificate_name)
# Act
result = self.sms.delete_management_certificate(self.management_certificate_name)
# Assert
self.assertIsNone(result)
self.assertFalse(self._management_certificate_exists(self.management_certificate_name))
class ManagementCertificateManagementServiceTest(AzureTestCase):
def setUp(self):
self.sms = ServiceManagementService(credentials.getSubscriptionId(),
credentials.getManagementCertFile())
self.sms.set_proxy(credentials.getProxyHost(),
credentials.getProxyPort(),
credentials.getProxyUser(),
credentials.getProxyPassword())
self.certificate_thumbprints = []
def tearDown(self):
for thumbprint in self.certificate_thumbprints:
try:
self.sms.delete_management_certificate(thumbprint)
except: pass
#--Helpers-----------------------------------------------------------------
def _create_management_certificate(self, cert):
self.certificate_thumbprints.append(cert.thumbprint)
result = self.sms.add_management_certificate(cert.public_key,
cert.thumbprint,
cert.data)
self.assertIsNone(result)
def _management_certificate_exists(self, thumbprint):
try:
props = self.sms.get_management_certificate(thumbprint)
return props is not None
except:
return False
#--Test cases for management certificates ----------------------------
def test_list_management_certificates(self):
# Arrange
local_cert = _local_certificate()
self._create_management_certificate(local_cert)
# Act
result = self.sms.list_management_certificates()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
cert = None
for temp in result:
if temp.subscription_certificate_thumbprint == local_cert.thumbprint:
cert = temp
break
self.assertIsNotNone(cert)
self.assertIsNotNone(cert.created)
self.assertEqual(cert.subscription_certificate_public_key, local_cert.public_key)
self.assertEqual(cert.subscription_certificate_data, local_cert.data)
self.assertEqual(cert.subscription_certificate_thumbprint, local_cert.thumbprint)
def test_get_management_certificate(self):
# Arrange
local_cert = _local_certificate()
self._create_management_certificate(local_cert)
# Act
result = self.sms.get_management_certificate(local_cert.thumbprint)
# Assert
self.assertIsNotNone(result)
self.assertIsNotNone(result.created)
self.assertEqual(result.subscription_certificate_public_key, local_cert.public_key)
self.assertEqual(result.subscription_certificate_data, local_cert.data)
self.assertEqual(result.subscription_certificate_thumbprint, local_cert.thumbprint)
def test_add_management_certificate(self):
# Arrange
local_cert = _local_certificate()
# Act
self.certificate_thumbprints.append(local_cert.thumbprint)
result = self.sms.add_management_certificate(local_cert.public_key,
local_cert.thumbprint,
local_cert.data)
# Assert
self.assertIsNone(result)
self.assertTrue(self._management_certificate_exists(local_cert.thumbprint))
def test_delete_management_certificate(self):
# Arrange
local_cert = _local_certificate()
self._create_management_certificate(local_cert)
# Act
result = self.sms.delete_management_certificate(local_cert.thumbprint)
# Assert
self.assertIsNone(result)
self.assertFalse(self._management_certificate_exists(local_cert.thumbprint))
