def test_aws_ecsclusters(self, mock_session):
mock_data = {'clusterArns': ['cluster/production','cluster/staging'],'ResponseMetadata': {}}
mock_session.return_value.client.return_value.list_clusters.return_value = mock_data
foo = AdsAws()
return_msg = foo.aws_ecsclusters()
expected = '**ADS AWS ECS Clusters**\n> production: cluster/production\n> staging: cluster/staging\n'
self.assertEqual(return_msg, expected)
def test_aws_ecsclusters(self, mock_session):
mock_data = {u'clusters': [{u'status': u'ACTIVE',
u'clusterName': u'staging',
u'registeredContainerInstancesCount': 3,
u'pendingTasksCount': 0,
u'runningTasksCount': 11,
u'activeServicesCount': 11}],
'ResponseMetadata': {}}
mock_session.return_value.client.return_value.describe_clusters.return_value = mock_data
foo = AdsAws()
return_msg = foo.aws_ecsclusterinfo('ecsclusterinfo','staging')
expected = '**staging**\n>Status: ACTIVE\n># Registered Container Instances: 3\n># running Tasks: 11\n># pending Tasks: 0\n># active Servies: 11\n'
self.assertEqual(return_msg, expected)
def test_aws_ecsclusterstatus(self, mock_session):
mock_data = {u'failures': [], u'containerInstances': [{u'status': u'ACTIVE',
u'registeredResources': [{u'integerValue': 1024, u'longValue': 0, u'type': u'INTEGER', u'name': u'CPU', u'doubleValue': 0.0},
{u'integerValue': 2004, u'longValue': 0, u'type': u'INTEGER', u'name': u'MEMORY', u'doubleValue': 0.0},
{u'name': u'PORTS', u'longValue': 0, u'doubleValue': 0.0, u'stringSetValue': [u'22', u'2376', u'2375', u'51678'],
u'type': u'STRINGSET', u'integerValue': 0}, {u'name': u'PORTS_UDP', u'longValue': 0, u'doubleValue': 0.0, u'stringSetValue': [],
u'type': u'STRINGSET', u'integerValue': 0}],
u'ec2InstanceId': u'i-3889f693',
u'agentConnected': True,
u'containerInstanceArn': u'ARN',
u'pendingTasksCount': 0,
u'remainingResources': [{u'integerValue': 1024, u'longValue': 0, u'type': u'INTEGER', u'name': u'CPU', u'doubleValue': 0.0},
{u'integerValue': 504, u'longValue': 0, u'type': u'INTEGER', u'name': u'MEMORY', u'doubleValue': 0.0},
{u'name': u'PORTS', u'longValue': 0, u'doubleValue': 0.0, u'stringSetValue': [u'22', u'2376', u'2375', u'51678'],
u'type': u'STRINGSET', u'integerValue': 0}, {u'name': u'PORTS_UDP', u'longValue': 0, u'doubleValue': 0.0, u'stringSetValue': [],
u'type': u'STRINGSET', u'integerValue': 0}],
u'runningTasksCount': 5,
u'versionInfo': {u'agentVersion': u'1.3.0', u'agentHash': u'097e4af', u'dockerVersion': u'DockerVersion: 1.6.2'}}],
'ResponseMetadata': {}}
mock_session.return_value.client.return_value.describe_container_instances.return_value = mock_data
foo = AdsAws()
return_msg = foo.aws_ecsclusterstatus('ecsclusterstatus', 'staging')
expected = '**Cluster Container info for: staging**\n>Container: ARN\n>ec2InstanceId: i-3889f693\n>Container status: ACTIVE\n>Docker version: DockerVersion: 1.6.2\n>Agent version: 1.3.0\n>Agent connected: True\n>+++++++++++++++++++++++++++++++++++++++++++++\n'
self.assertEqual(return_msg, expected)
