def on_post(self, req, resp):
body = load_body(req)
api_secret = body['api_secret']
coordinator_uri = body['coordinator_uri']
personality = body['personality']
#start pairing on a separate process
pairing_process = PairingProcess(
api_secret, coordinator_uri, personality)
pairing_process.run()
resp.status = falcon.HTTP_200
def on_post(self, req, resp, validated_body):
body = validated_body['pairing_configuration']
api_secret = body['api_secret']
coordinator_uri = body['coordinator_uri']
personality = body['personality']
#start pairing on a separate process
pairing_process = PairingProcess(
api_secret, coordinator_uri, personality)
pairing_process.run()
resp.status = falcon.HTTP_200
def on_post(self, req, resp, validated_body):
body = validated_body['pairing_configuration']
api_secret = body['api_secret']
coordinator_uri = body['coordinator_uri']
personality = body['personality']
#start pairing on a separate process
pairing_process = PairingProcess(api_secret, coordinator_uri,
personality)
pairing_process.run()
resp.status = falcon.HTTP_200
def setUp(self):
self.api_secret = "3F2504E0-4F89-11D3-9A0C-0305E82C3301"
self.coordinator_uri = "http://localhost:8080/v1"
self.personality = 'normalization'
self.pairing_process = PairingProcess(self.api_secret,
self.coordinator_uri,
self.personality)
self.native_proxy = MagicMock()
self.get_config = WorkerConfiguration(
personality='pairing',
personality_module='meniscus.personas.pairing.app',
worker_token='token_id',
worker_id='worker_id',
coordinator_uri="192.168.1.1:8080/v1")
self.resp = requests.Response()
self.http_request = MagicMock(return_value=self.resp)
self.registration = WorkerRegistration(personality='correlation')\
.format()
def setUp(self):
self.api_secret = "3F2504E0-4F89-11D3-9A0C-0305E82C3301"
self.coordinator_uri = "http://localhost:8080/v1"
self.personality = 'normalization'
self.pairing_process = PairingProcess(
self.api_secret, self.coordinator_uri, self.personality)
self.native_proxy = MagicMock()
self.get_config = WorkerConfiguration(
personality='pairing',
personality_module='meniscus.personas.pairing.app',
worker_token='token_id',
worker_id='worker_id',
coordinator_uri="192.168.1.1:8080/v1"
)
self.resp = requests.Response()
self.http_request = MagicMock(return_value=self.resp)
self.registration = WorkerRegistration(personality='correlation')\
.format()
class WhenTestingPairingProcess(unittest.TestCase):
def setUp(self):
self.api_secret = "3F2504E0-4F89-11D3-9A0C-0305E82C3301"
self.coordinator_uri = "http://localhost:8080/v1"
self.personality = 'normalization'
self.pairing_process = PairingProcess(
self.api_secret, self.coordinator_uri, self.personality)
self.native_proxy = MagicMock()
self.get_config = WorkerConfiguration(
personality='pairing',
personality_module='meniscus.personas.pairing.app',
worker_token='token_id',
worker_id='worker_id',
coordinator_uri="192.168.1.1:8080/v1"
)
self.resp = requests.Response()
self.http_request = MagicMock(return_value=self.resp)
self.registration = WorkerRegistration(personality='correlation')\
.format()
def test_process_start_on_run(self):
with patch.object(pairing_process.Process, 'start') as start:
self.pairing_process.run()
start.assert_called_once()
def test_server_restart_pair_with_coordinator(self):
sys_assist = MagicMock()
sys_assist.get_lan_ip.return_value = "192.168.1.1"
sys_assist.get_cpu_core_count.return_value = "4"
sys_assist.get_disk_size_GB.return_value = "20"
sys_assist.get_sys_mem_total_MB.return_value = "4090"
with patch.object(pairing_process.NativeProxy, 'restart') \
as server_restart, \
patch.object(pairing_process.PairingProcess,
'_register_with_coordinator') as register, \
patch.object(pairing_process.PairingProcess,
'_get_worker_routes') as routes:
self.pairing_process._pair_with_coordinator(
self.api_secret, self.coordinator_uri, self.personality)
server_restart.assert_called_once()
register.assert_called_once()
routes.assert_called_once()
def test_should_return_true_for_register_with_coordinator(self):
self.resp.status_code = httplib.ACCEPTED
self.resp._content = \
'{"personality_module": "meniscus.personas.pairing.app", ' \
'"worker_token": "3F2504E0-4F89-11D3-9A0C-0305E82C3301", ' \
'"worker_id": "3F2504E0-4F89-11D3-9A0C-0305E82C3301"}'
auth_header = {'X-AUTH-TOKEN': self.api_secret}
with patch('meniscus.api.pairing.pairing_process.'
'http_request', self.http_request):
self.assertTrue(
self.pairing_process._register_with_coordinator(
self.coordinator_uri, self.personality,
self.registration, auth_header))
self.http_request.assert_called_once_with(
self.coordinator_uri + '/pairing',
auth_header,
jsonutils.dumps(self.registration),
http_verb='POST')
def test_should_return_true_for_get_worker_routes(self):
with patch('meniscus.api.pairing.pairing_process.routing.'
'get_routes_from_coordinator',
MagicMock(return_value=True)) as get_routes:
self.assertTrue(self.pairing_process._get_worker_routes())
get_routes.assert_called_once()
class WhenTestingPairingProcess(unittest.TestCase):
def setUp(self):
self.api_secret = "3F2504E0-4F89-11D3-9A0C-0305E82C3301"
self.coordinator_uri = "http://localhost:8080/v1"
self.personality = 'normalization'
self.pairing_process = PairingProcess(self.api_secret,
self.coordinator_uri,
self.personality)
self.native_proxy = MagicMock()
self.get_config = WorkerConfiguration(
personality='pairing',
personality_module='meniscus.personas.pairing.app',
worker_token='token_id',
worker_id='worker_id',
coordinator_uri="192.168.1.1:8080/v1")
self.resp = requests.Response()
self.http_request = MagicMock(return_value=self.resp)
self.registration = WorkerRegistration(personality='correlation')\
.format()
def test_process_start_on_run(self):
with patch.object(pairing_process.Process, 'start') as start:
self.pairing_process.run()
start.assert_called_once()
def test_server_restart_pair_with_coordinator(self):
sys_assist = MagicMock()
sys_assist.get_lan_ip.return_value = "192.168.1.1"
sys_assist.get_cpu_core_count.return_value = "4"
sys_assist.get_disk_size_GB.return_value = "20"
sys_assist.get_sys_mem_total_MB.return_value = "4090"
with patch.object(pairing_process.NativeProxy, 'restart') \
as server_restart, \
patch.object(pairing_process.PairingProcess,
'_register_with_coordinator') as register:
self.pairing_process._pair_with_coordinator(
self.api_secret, self.coordinator_uri, self.personality)
server_restart.assert_called_once()
register.assert_called_once()
def test_should_return_true_for_register_with_coordinator(self):
self.resp.status_code = httplib.ACCEPTED
self.resp._content = jsonutils.dumps({
'worker_identity': {
"personality_module": "meniscus.personas.pairing.app",
"worker_token": "3F2504E0-4F89-11D3-9A0C-0305E82C3301",
"worker_id": "3F2504E0-4F89-11D3-9A0C-0305E82C3301"
}
})
auth_header = {'X-AUTH-TOKEN': self.api_secret}
with patch('meniscus.api.pairing.pairing_process.'
'http_request', self.http_request):
self.assertTrue(
self.pairing_process._register_with_coordinator(
self.coordinator_uri, self.personality, self.registration,
auth_header))
self.http_request.assert_called_once_with(
self.coordinator_uri + '/pairing',
auth_header,
jsonutils.dumps(self.registration),
http_verb='POST')
