def test_requestComponentSecurityState(self, FileCache_mock,
runCommand_mock):
FileCache_mock.return_value = None
status_command = {
"serviceName": 'HDFS',
"commandType": "STATUS_COMMAND",
"clusterName": "",
"componentName": "DATANODE",
'configurations': {}
}
dummy_controller = MagicMock()
orchestrator = CustomServiceOrchestrator(self.config, dummy_controller)
# Test securityState
runCommand_mock.return_value = {
'exitcode': 0,
'structuredOut': {
'securityState': 'UNSECURED'
}
}
status = orchestrator.requestComponentSecurityState(status_command)
self.assertEqual('UNSECURED', status)
# Test case where exit code indicates failure
runCommand_mock.return_value = {"exitcode": 1}
status = orchestrator.requestComponentSecurityState(status_command)
self.assertEqual('UNKNOWN', status)
def test_requestComponentSecurityState(self, FileCache_mock, runCommand_mock):
FileCache_mock.return_value = None
status_command = {
"serviceName" : 'HDFS',
"commandType" : "STATUS_COMMAND",
"clusterName" : "",
"componentName" : "DATANODE",
'configurations':{}
}
dummy_controller = MagicMock()
orchestrator = CustomServiceOrchestrator(self.config, dummy_controller)
# Test securityState
runCommand_mock.return_value = {
'exitcode' : 0,
'structuredOut' : {'securityState': 'UNSECURED'}
}
status = orchestrator.requestComponentSecurityState(status_command)
self.assertEqual('UNSECURED', status)
# Test case where exit code indicates failure
runCommand_mock.return_value = {
"exitcode" : 1
}
status = orchestrator.requestComponentSecurityState(status_command)
self.assertEqual('UNKNOWN', status)
def test_requestComponentSecurityState_realFailure(self, FileCache_mock):
'''
Tests the case where the CustomServiceOrchestrator attempts to call a service's security_status
method, but fails to do so because the script or method was not found.
:param FileCache_mock:
:return:
'''
FileCache_mock.return_value = None
status_command = {
"serviceName": 'BOGUS_SERVICE',
"commandType": "STATUS_COMMAND",
"clusterName": "",
"componentName": "DATANODE",
'configurations': {}
}
dummy_controller = MagicMock()
orchestrator = CustomServiceOrchestrator(self.config, dummy_controller)
status = orchestrator.requestComponentSecurityState(status_command)
self.assertEqual('UNKNOWN', status)
def test_requestComponentSecurityState_realFailure(self, FileCache_mock):
'''
Tests the case where the CustomServiceOrchestrator attempts to call a service's security_status
method, but fails to do so because the script or method was not found.
:param FileCache_mock:
:return:
'''
FileCache_mock.return_value = None
status_command = {
"serviceName" : 'BOGUS_SERVICE',
"commandType" : "STATUS_COMMAND",
"clusterName" : "",
"componentName" : "DATANODE",
'configurations':{}
}
dummy_controller = MagicMock()
orchestrator = CustomServiceOrchestrator(self.config, dummy_controller)
status = orchestrator.requestComponentSecurityState(status_command)
self.assertEqual('UNKNOWN', status)
