def _test_getters(self, test_data, mock_restutil, MockCryptUtil, _): mock_restutil.http_get.side_effect = test_data.mock_http_get MockCryptUtil.side_effect = test_data.mock_crypt_util protocol = WireProtocol("foo.bar") protocol.detect() protocol.get_vminfo() protocol.get_certs() ext_handlers, etag = protocol.get_ext_handlers() for ext_handler in ext_handlers.extHandlers: protocol.get_ext_handler_pkgs(ext_handler) crt1 = os.path.join(self.tmp_dir, '33B0ABCE4673538650971C10F7D7397E71561F35.crt') crt2 = os.path.join(self.tmp_dir, '4037FBF5F1F3014F99B5D6C7799E9B20E6871CB3.crt') prv2 = os.path.join(self.tmp_dir, '4037FBF5F1F3014F99B5D6C7799E9B20E6871CB3.prv') self.assertTrue(os.path.isfile(crt1)) self.assertTrue(os.path.isfile(crt2)) self.assertTrue(os.path.isfile(prv2))
def _test_getters(self, test_data, mock_restutil, MockCryptUtil, _): mock_restutil.http_get.side_effect = test_data.mock_http_get MockCryptUtil.side_effect = test_data.mock_crypt_util protocol = WireProtocol("foo.bar") protocol.detect() protocol.get_vminfo() protocol.get_certs() ext_handlers, etag = protocol.get_ext_handlers() for ext_handler in ext_handlers.extHandlers: protocol.get_ext_handler_pkgs(ext_handler) crt1 = os.path.join(self.tmp_dir, '33B0ABCE4673538650971C10F7D7397E71561F35.crt') crt2 = os.path.join(self.tmp_dir, '4037FBF5F1F3014F99B5D6C7799E9B20E6871CB3.crt') prv2 = os.path.join(self.tmp_dir, '4037FBF5F1F3014F99B5D6C7799E9B20E6871CB3.prv') self.assertTrue(os.path.isfile(crt1)) self.assertTrue(os.path.isfile(crt2)) self.assertTrue(os.path.isfile(prv2))
def test_it_should_report_an_error_if_the_wireserver_cannot_be_reached( self, patch_is_triggered, patch_add_event): test_message = "TEST MESSAGE" patch_is_triggered.return_value = True # protocol errors are reported only after a delay; force the error to be reported now protocol = WireProtocol("foo.bar") protocol.get_ext_handlers = MagicMock( side_effect=ProtocolError(test_message)) get_exthandlers_handler(protocol).run() self.assertEquals(patch_add_event.call_count, 2) _, first_call_args = patch_add_event.call_args_list[0] self.assertEquals(first_call_args['op'], WALAEventOperation.GetArtifactExtended) self.assertEquals(first_call_args['is_success'], False) _, second_call_args = patch_add_event.call_args_list[1] self.assertEquals(second_call_args['op'], WALAEventOperation.ExtensionProcessing) self.assertEquals(second_call_args['is_success'], False) self.assertIn(test_message, second_call_args['message'])