Examples with Header com.twosigma.beakerx.message.Header used on opensource projects

Example 11 with Header

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

the class JupyterHandlerTest method initHeader.

public static Header initHeader(JupyterMessages jupyterMessages) {
    Header header = new Header();
    header.setId("messageId");
    header.setUsername("username");
    header.setSession("sessionId" + jupyterMessages.getName());
    header.setType(jupyterMessages.getName());
    header.setVersion("5.0");
    return header;
}

Example 12 with Header

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

the class ScalaKernelInfoHandlerTest 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());
}

Example 13 with Header

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

the class ExecuteRequestHandlerTest method copyMessage.

private static Message copyMessage(Message origin) {
    Message copy = new Message();
    for (byte[] list : origin.getIdentities()) {
        copy.getIdentities().add(list.clone());
    }
    String header = toJson(origin.getHeader());
    String parent = toJson(origin.getParentHeader());
    String metadata = toJson(origin.getMetadata());
    String content = toJson(origin.getContent());
    copy.setHeader(parse(header, Header.class));
    copy.setParentHeader(parse(parent, Header.class));
    copy.setMetadata(parse(metadata, LinkedHashMap.class));
    copy.setContent(parse(content, LinkedHashMap.class));
    return copy;
}

Example 14 with Header

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

the class CompleteHandlerTest method handle_sentMessageHasHeaderTypeIsCompleteReply.

@Test
public void handle_sentMessageHasHeaderTypeIsCompleteReply() throws Exception {
    // when
    completeHandler.handle(message);
    // then
    Header header = kernel.getSentMessages().get(0).getHeader();
    Assertions.assertThat(header).isNotNull();
    Assertions.assertThat(header.getType()).isEqualTo(COMPLETE_REPLY.getName());
}

Example 15 with Header

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

the class GroovyKernelTest method outputWidgetUpdateMessage.

private Message outputWidgetUpdateMessage(String outputCommId) {
    Message message = new Message();
    Header header = new Header();
    header.setTypeEnum(JupyterMessages.COMM_MSG);
    message.setHeader(header);
    HashMap<String, Serializable> content = new HashMap<>();
    HashMap<String, Serializable> data = new HashMap<>();
    data.put("method", "update");
    HashMap<String, Serializable> state = new HashMap<>();
    state.put("outputs", new ArrayList<>());
    data.put("state", state);
    content.put("comm_id", outputCommId);
    content.put("data", data);
    message.setContent(content);
    return message;
}