class NodeManagerTestCase(TestCase):
def setUp(self):
self.nm = NodeManager('localhost')
def test_node_information(self, request_mock):
self.nm.node_information()
request_mock.assert_called_with('/ws/v1/node/info')
def test_node_applications(self, request_mock):
self.nm.node_applications('RUNNING', 'root')
request_mock.assert_called_with('/ws/v1/node/apps',
params={"state":'RUNNING', "user":'******'})
self.nm.node_applications()
request_mock.assert_called_with('/ws/v1/node/apps', params={})
with self.assertRaises(IllegalArgumentError):
self.nm.node_applications('ololo', 'root')
def test_node_application(self, request_mock):
self.nm.node_application('app_1')
request_mock.assert_called_with('/ws/v1/node/apps/app_1')
def test_node_containers(self, request_mock):
self.nm.node_containers()
request_mock.assert_called_with('/ws/v1/node/containers')
def test_node_container(self, request_mock):
self.nm.node_container('container_1')
request_mock.assert_called_with('/ws/v1/node/containers/container_1')
def setUp(self):
self.nm = NodeManager('localhost')
def setUp(self):
self.nm = NodeManager('localhost')
class NodeManagerTestCase(TestCase):
def setUp(self):
self.nm = NodeManager('localhost')
def test_node_information(self, request_mock):
self.nm.node_information()
request_mock.assert_called_with('/ws/v1/node/info')
def test_node_applications(self, request_mock):
self.nm.node_applications('RUNNING', 'root')
request_mock.assert_called_with('/ws/v1/node/apps',
params={
"state": 'RUNNING',
"user": '******'
})
self.nm.node_applications()
request_mock.assert_called_with('/ws/v1/node/apps', params={})
with self.assertRaises(IllegalArgumentError):
self.nm.node_applications('ololo', 'root')
def test_node_application(self, request_mock):
self.nm.node_application('app_1')
request_mock.assert_called_with('/ws/v1/node/apps/app_1')
def test_node_containers(self, request_mock):
self.nm.node_containers()
request_mock.assert_called_with('/ws/v1/node/containers')
def test_node_container(self, request_mock):
self.nm.node_container('container_1')
request_mock.assert_called_with('/ws/v1/node/containers/container_1')
def test_auxiliary_services(self, request_mock):
self.nm.auxiliary_services()
request_mock.assert_called_with('/ws/v1/node/auxiliaryservices')
def test_auxiliary_services_update(self, request_mock):
self.nm.auxiliary_services_update({
"services": [{
"name": "mapreduce_shuffle",
"version": "2",
"configuration": {
"properties": {
"class.name":
"org.apache.hadoop.mapred.ShuffleHandler",
"mapreduce.shuffle.transfer.buffer.size": "102400",
"mapreduce.shuffle.port": "13563"
}
}
}]
})
request_mock.assert_called_with(
'/ws/v1/node/auxiliaryservices',
'PUT',
json={
"services": [{
"name": "mapreduce_shuffle",
"version": "2",
"configuration": {
"properties": {
"class.name":
"org.apache.hadoop.mapred.ShuffleHandler",
"mapreduce.shuffle.transfer.buffer.size": "102400",
"mapreduce.shuffle.port": "13563"
}
}
}]
})
class NodeManagerTestCase(TestCase):
def setUp(self):
self.nm = NodeManager('localhost')
def test_node_information(self, request_mock):
self.nm.node_information()
request_mock.assert_called_with('/ws/v1/node/info')
def test_node_applications(self, request_mock):
self.nm.node_applications('RUNNING', 'root')
request_mock.assert_called_with('/ws/v1/node/apps',
state='RUNNING',
user='******')
self.nm.node_applications()
request_mock.assert_called_with('/ws/v1/node/apps')
with self.assertRaises(IllegalArgumentError):
self.nm.node_applications('ololo', 'root')
def test_node_application(self, request_mock):
self.nm.node_application('app_1')
request_mock.assert_called_with('/ws/v1/node/apps/app_1')
def test_node_containers(self, request_mock):
self.nm.node_containers()
request_mock.assert_called_with('/ws/v1/node/containers')
def test_node_container(self, request_mock):
self.nm.node_container('container_1')
request_mock.assert_called_with('/ws/v1/node/containers/container_1')