class ServiceBusManagementServiceTest(AzureTestCase):
def setUp(self):
self.sms = ServiceBusManagementService(
credentials.getSubscriptionId(),
credentials.getManagementCertFile())
set_service_options(self.sms)
self.sb_namespace = getUniqueName('uts')
def tearDown(self):
try:
self.sms.delete_namespace(self.sb_namespace)
except:
pass
#--Helpers-----------------------------------------------------------------
def _namespace_exists(self, name):
try:
ns = self.sms.get_namespace(name)
# treat it as non-existent if it is in process of being removed
return ns.status != 'Removing'
except:
return False
def _wait_for_namespace_active(self, name):
count = 0
ns = self.sms.get_namespace(name)
while ns.status != 'Active':
count = count + 1
if count > 120:
self.assertTrue(
False,
'Timed out waiting for service bus namespace activation.')
time.sleep(5)
ns = self.sms.get_namespace(name)
#--Operations for service bus ----------------------------------------
def test_get_regions(self):
# Arrange
# Act
result = self.sms.get_regions()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
for region in result:
self.assertTrue(len(region.code) > 0)
self.assertTrue(len(region.fullname) > 0)
def test_list_namespaces(self):
# Arrange
# Act
result = self.sms.list_namespaces()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
for ns in result:
self.assertTrue(len(ns.name) > 0)
self.assertTrue(len(ns.region) > 0)
def test_get_namespace(self):
# Arrange
name = credentials.getServiceBusNamespace()
# Act
result = self.sms.get_namespace(name)
# Assert
self.assertIsNotNone(result)
self.assertEqual(result.name, name)
self.assertIsNotNone(result.region)
self.assertIsNotNone(result.default_key)
self.assertIsNotNone(result.status)
self.assertIsNotNone(result.created_at)
self.assertIsNotNone(result.acs_management_endpoint)
self.assertIsNotNone(result.servicebus_endpoint)
self.assertIsNotNone(result.connection_string)
self.assertEqual(result.subscription_id,
credentials.getSubscriptionId().replace('-', ''))
self.assertTrue(result.enabled)
def test_get_namespace_with_non_existing_namespace(self):
# Arrange
name = self.sb_namespace
# Act
with self.assertRaises(WindowsAzureMissingResourceError):
self.sms.get_namespace(name)
# Assert
def test_check_namespace_availability_not_available(self):
# arrange
name = credentials.getServiceBusNamespace()
# act
availability = self.sms.check_namespace_availability(name)
# assert
self.assertFalse(availability.result)
def test_check_namespace_availability_available(self):
# arrange
name = 'someunusedname'
# act
availability = self.sms.check_namespace_availability(name)
# assert
self.assertTrue(availability.result)
def test_create_namespace(self):
# Arrange
name = self.sb_namespace
region = 'West US'
# Act
result = self.sms.create_namespace(name, region)
self._wait_for_namespace_active(name)
# Assert
self.assertIsNone(result)
self.assertTrue(self._namespace_exists(name))
def test_create_namespace_with_existing_namespace(self):
# Arrange
name = self.sb_namespace
region = 'West US'
self.sms.create_namespace(name, region)
self._wait_for_namespace_active(name)
# Act
with self.assertRaises(WindowsAzureError):
self.sms.create_namespace(name, region)
# Assert
def test_delete_namespace(self):
# Arrange
name = self.sb_namespace
region = 'West US'
self.sms.create_namespace(name, region)
self._wait_for_namespace_active(name)
# Act
result = self.sms.delete_namespace(name)
# Assert
self.assertIsNone(result)
self.assertFalse(self._namespace_exists(name))
def test_delete_namespace_with_non_existing_namespace(self):
# Arrange
name = self.sb_namespace
# Act
with self.assertRaises(WindowsAzureMissingResourceError):
self.sms.delete_namespace(name)
class ServiceBusManagementServiceTest(AzureTestCase):
def setUp(self):
proxy_host = credentials.getProxyHost()
proxy_port = credentials.getProxyPort()
self.sms = ServiceBusManagementService(credentials.getSubscriptionId(), credentials.getManagementCertFile())
if proxy_host:
self.sms.set_proxy(proxy_host, proxy_port)
self.sb_namespace = getUniqueNameBasedOnCurrentTime('uts')
def tearDown(self):
try:
self.sms.delete_namespace(self.sb_namespace)
except: pass
#--Helpers-----------------------------------------------------------------
def _namespace_exists(self, name):
try:
ns = self.sms.get_namespace(name)
# treat it as non-existent if it is in process of being removed
return ns.status != 'Removing'
except:
return False
def _wait_for_namespace_active(self, name):
count = 0
ns = self.sms.get_namespace(name)
while ns.status != 'Active':
count = count + 1
if count > 120:
self.assertTrue(False, 'Timed out waiting for service bus namespace activation.')
time.sleep(5)
ns = self.sms.get_namespace(name)
#--Operations for service bus ----------------------------------------
def test_get_regions(self):
# Arrange
# Act
result = self.sms.get_regions()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
for region in result:
self.assertTrue(len(region.code) > 0)
self.assertTrue(len(region.fullname) > 0)
def test_list_namespaces(self):
# Arrange
# Act
result = self.sms.list_namespaces()
# Assert
self.assertIsNotNone(result)
self.assertTrue(len(result) > 0)
for ns in result:
self.assertTrue(len(ns.name) > 0)
self.assertTrue(len(ns.region) > 0)
def test_get_namespace(self):
# Arrange
name = credentials.getServiceBusNamespace()
# Act
result = self.sms.get_namespace(name)
# Assert
self.assertIsNotNone(result)
self.assertEqual(result.name, name)
self.assertIsNotNone(result.region)
self.assertIsNotNone(result.default_key)
self.assertIsNotNone(result.status)
self.assertIsNotNone(result.created_at)
self.assertIsNotNone(result.acs_management_endpoint)
self.assertIsNotNone(result.servicebus_endpoint)
self.assertIsNotNone(result.connection_string)
self.assertEqual(result.subscription_id, credentials.getSubscriptionId().replace('-', ''))
self.assertTrue(result.enabled)
def test_get_namespace_with_non_existing_namespace(self):
# Arrange
name = self.sb_namespace
# Act
with self.assertRaises(WindowsAzureMissingResourceError):
self.sms.get_namespace(name)
# Assert
def test_check_namespace_availability_not_available(self):
# arrange
name = credentials.getServiceBusNamespace()
# act
availability = self.sms.check_namespace_availability(name)
# assert
self.assertFalse(availability.result)
def test_check_namespace_availability_available(self):
# arrange
name = 'someunusedname'
# act
availability = self.sms.check_namespace_availability(name)
# assert
self.assertTrue(availability.result)
def test_create_namespace(self):
# Arrange
name = self.sb_namespace
region = 'West US'
# Act
result = self.sms.create_namespace(name, region)
self._wait_for_namespace_active(name)
# Assert
self.assertIsNone(result)
self.assertTrue(self._namespace_exists(name))
def test_create_namespace_with_existing_namespace(self):
# Arrange
name = self.sb_namespace
region = 'West US'
self.sms.create_namespace(name, region)
self._wait_for_namespace_active(name)
# Act
with self.assertRaises(WindowsAzureError):
self.sms.create_namespace(name, region)
# Assert
def test_delete_namespace(self):
# Arrange
name = self.sb_namespace
region = 'West US'
self.sms.create_namespace(name, region)
self._wait_for_namespace_active(name)
# Act
result = self.sms.delete_namespace(name)
# Assert
self.assertIsNone(result)
self.assertFalse(self._namespace_exists(name))
def test_delete_namespace_with_non_existing_namespace(self):
# Arrange
name = self.sb_namespace
# Act
with self.assertRaises(WindowsAzureMissingResourceError):
self.sms.delete_namespace(name)
