Search in sources :

Example 21 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class IsCompleteRequestHandler method handleMsg.

private void handleMsg(Message message) {
    logger.debug("Processing is complete request");
    Message reply = new Message();
    HashMap<String, Serializable> map = new HashMap<>();
    map.put("status", "complete");
    reply.setContent(map);
    reply.setHeader(new Header(IS_COMPLETE_REPLY, message.getHeader().getSession()));
    reply.setParentHeader(message.getHeader());
    reply.setIdentities(message.getIdentities());
    send(reply);
}
Also used : Serializable(java.io.Serializable) Message(com.twosigma.beakerx.message.Message) Header(com.twosigma.beakerx.message.Header) HashMap(java.util.HashMap)

Example 22 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class CompleteHandler method createMsg.

private Message createMsg(Message message, int cursorPos, AutocompleteResult autocomplete) {
    Message reply = new Message();
    reply.setHeader(new Header(COMPLETE_REPLY, message.getHeader().getSession()));
    reply.setIdentities(message.getIdentities());
    reply.setParentHeader(message.getHeader());
    Map<String, Serializable> content = new HashMap<>();
    content.put(STATUS, "ok");
    content.put(MATCHES, autocomplete.getMatches().toArray());
    content.put(CURSOR_END, cursorPos);
    content.put(CURSOR_START, autocomplete.getStartIndex());
    reply.setContent(content);
    return reply;
}
Also used : Serializable(java.io.Serializable) Message(com.twosigma.beakerx.message.Message) Header(com.twosigma.beakerx.message.Header) HashMap(java.util.HashMap)

Example 23 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class MessageCreator method initMessage.

private static Message initMessage(JupyterMessages type, Message message) {
    Message reply = new Message();
    reply.setParentHeader(message.getHeader());
    reply.setIdentities(message.getIdentities());
    reply.setHeader(new Header(type, message.getHeader().getSession()));
    return reply;
}
Also used : Message(com.twosigma.beakerx.message.Message) Header(com.twosigma.beakerx.message.Header)

Example 24 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class GroovyKernelInfoHandlerTest method handle_sentMessageHasHeaderTypeIsKernelInfoReply.

@Test
public void handle_sentMessageHasHeaderTypeIsKernelInfoReply() throws Exception {
    // when
    handler.handle(message);
    // then
    Message sentMessage = kernel.getSentMessages().get(0);
    Header header = sentMessage.getHeader();
    Assertions.assertThat(header).isNotNull();
    Assertions.assertThat(header.getType()).isEqualTo(KERNEL_INFO_REPLY.getName());
}
Also used : Message(com.twosigma.beakerx.message.Message) Header(com.twosigma.beakerx.message.Header) Test(org.junit.Test) MessageTest(com.twosigma.beakerx.message.MessageTest)

Example 25 with Header

use of com.twosigma.beakerx.message.Header in project beakerx by twosigma.

the class ClojureKernelInfoHandlerTest method handle_sentMessageHasHeaderTypeIsKernelInfoReply.

@Test
public void handle_sentMessageHasHeaderTypeIsKernelInfoReply() throws Exception {
    // when
    handler.handle(message);
    // then
    Message sentMessage = kernel.getSentMessages().get(0);
    Header header = sentMessage.getHeader();
    Assertions.assertThat(header).isNotNull();
    Assertions.assertThat(header.getType()).isEqualTo(KERNEL_INFO_REPLY.getName());
}
Also used : Message(com.twosigma.beakerx.message.Message) Header(com.twosigma.beakerx.message.Header) MessageTest(com.twosigma.beakerx.message.MessageTest) Test(org.junit.Test)

Aggregations

Header (com.twosigma.beakerx.message.Header)25 Message (com.twosigma.beakerx.message.Message)23 Serializable (java.io.Serializable)12 HashMap (java.util.HashMap)12 Test (org.junit.Test)6 MessageTest (com.twosigma.beakerx.message.MessageTest)5 EvaluatorResultTestWatcher.waitForIdleMessage (com.twosigma.beakerx.evaluator.EvaluatorResultTestWatcher.waitForIdleMessage)2 Comm (com.twosigma.beakerx.kernel.comm.Comm)2 LinkedHashMap (java.util.LinkedHashMap)2 KernelTest (com.twosigma.beakerx.KernelTest)1 MessageFactoryTest.getExecuteRequestMessage (com.twosigma.beakerx.MessageFactoryTest.getExecuteRequestMessage)1 EvaluatorResultTestWatcher.waitForErrorMessage (com.twosigma.beakerx.evaluator.EvaluatorResultTestWatcher.waitForErrorMessage)1 EvaluatorResultTestWatcher.waitForUpdateMessage (com.twosigma.beakerx.evaluator.EvaluatorResultTestWatcher.waitForUpdateMessage)1 EvaluatorTest (com.twosigma.beakerx.evaluator.EvaluatorTest)1 KernelHandler (com.twosigma.beakerx.handler.KernelHandler)1 KernelHandlerWrapper.wrapBusyIdle (com.twosigma.beakerx.handler.KernelHandlerWrapper.wrapBusyIdle)1 KernelFunctionality (com.twosigma.beakerx.kernel.KernelFunctionality)1 COMMS (com.twosigma.beakerx.kernel.comm.Comm.COMMS)1 TARGET_NAME (com.twosigma.beakerx.kernel.comm.Comm.TARGET_NAME)1 JupyterMessages (com.twosigma.beakerx.kernel.msg.JupyterMessages)1