@Override public String toString() { return "Type = " + header.getType(); }
private TryResult addImplicits(Message parent, String codeToExecute) { SimpleEvaluationObject seo = createSimpleEvaluationObject( codeToExecute, kernel, new Message(new Header(JupyterMessages.COMM_MSG, parent.getHeader().getSession())), 1); return kernel.executeCode(codeToExecute, seo); }
@Test public void serializeConfig_createConfigWithProperties() { //when Header header = MessageSerializer.parse(srcJson, Header.class); //then Assertions.assertThat(header.getVersion()).isEqualTo("5.0"); Assertions.assertThat(header.getType()).isEqualTo("execute_request"); Assertions.assertThat(header.getId()).isEqualTo("9CEE9EE3351643CD83A6C2A29401A942"); Assertions.assertThat(header.getUsername()).isEqualTo("username"); Assertions.assertThat(header.getSession()).isEqualTo("605DC3793F6F402698418908DBC7C572"); } }
public static Header initHeader(JupyterMessages jupyterMessages) { Header header = new Header(jupyterMessages, "sessionId" + jupyterMessages.getName()); header.setId("messageId"); header.setUsername("username"); header.setVersion("5.0"); return header; }
public static Message msg(JupyterMessages type) { return new Message(new Header(type, "session1")); } }
private void handleControlMsg() { Message message = readMessage(controlSocket); JupyterMessages type = message.getHeader().getTypeEnum(); if (type.equals(SHUTDOWN_REQUEST)) { Message reply = new Message(new Header(SHUTDOWN_REPLY, message.getHeader().getSession())); reply.setParentHeader(message.getHeader()); reply.setContent(message.getContent()); sendMsg(controlSocket, Collections.singletonList(reply)); shutdown(); } }
@Test public void handleMessage_firstSentMessageHasSessionId() throws Exception { //given String expectedSessionId = message.getHeader().getSession(); //when commMsgHandler.handle(message); //then Assertions.assertThat(kernel.getPublishedMessages()).isNotEmpty(); Message publishMessage = kernel.getPublishedMessages().get(0); Assertions.assertThat(publishMessage.getHeader().getSession()).isEqualTo(expectedSessionId); }
public static Message getExecuteRequestMessage(String code) { Header header = new Header(JupyterMessages.COMM_MSG,"session1"); Message message = new Message(header); header.setTypeEnum(JupyterMessages.EXECUTE_REQUEST); HashMap<String, Serializable> content = new HashMap<>(); content.put("code", code); message.setContent(content); return message; }
private synchronized void send(boolean isError, String text) { List<Message> list = new ArrayList<>(); list.add(getComm().createUpdateMessage(MSG_ID, getComm().getParentMessage().getHeader().getId())); Map<String, Serializable> asMap = addOutput(isError, text); list.add(getComm().createOutputContent(asMap)); list.add(getComm().createUpdateMessage(MSG_ID, "")); getComm().publish(list); }
public JupyterMessages type() { return (header != null && header.getTypeEnum() != null) ? header.getTypeEnum() : null; }
public static Message createMessage() { Header header = new Header(JupyterMessages.COMM_MSG, "session1"); Message message = new Message(header); message.getIdentities().add("identityStr".getBytes()); message.setParentHeader(header); message.setMetadata(new LinkedHashMap<>()); message.setContent(new LinkedHashMap<>()); return message; }
@Test public void handleMessage_sentMessageHasSessionId() throws Exception { //given String expectedSessionId = message.getHeader().getSession(); //when commInfoHandler.handle(message); //then Assertions.assertThat(kernel.getSentMessages()).isNotEmpty(); Message sendMessage = kernel.getSentMessages().get(0); Assertions.assertThat(sendMessage.getHeader().getSession()).isEqualTo(expectedSessionId); }
private void display(HashMap<String, Serializable> content) { List<Message> list = new ArrayList<>(); list.add(getComm().createUpdateMessage(MSG_ID, getComm().getParentMessage().getHeader().getId())); list.add(getComm().createMessage(DISPLAY_DATA, Comm.Buffer.EMPTY, new Comm.Data(content))); list.add(getComm().createUpdateMessage(MSG_ID, "")); getComm().publish(list); }
@Test public void handleMagicMessage_replyIsSent() throws InterruptedException { //when executeRequestHandler.handle(magicMessage); waitForIdleMessage(kernel); //then final List<Message> sentMessages = kernel.getSentMessages(); assertThat(sentMessages).isNotEmpty(); Message firstSentMessage = sentMessages.get(0); assertThat(firstSentMessage.getHeader().getTypeEnum()).isEqualTo(EXECUTE_REPLY); }
private TryResult initSparkContextInShell(KernelFunctionality kernel, Message parent) { String addSc = String.format(("import com.twosigma.beakerx.widget.SparkVariable\n" + "val %s = SparkVariable.getSparkSession()\n" + "val %s = %s.sparkContext\n" + "import org.apache.spark.SparkContext._\n" + "import %s.implicits._\n" + "import %s.sql\n" + "import org.apache.spark.sql.functions._\n"), SPARK_SESSION_NAME, SPARK_CONTEXT_NAME, SPARK_SESSION_NAME, SPARK_SESSION_NAME, SPARK_SESSION_NAME); SimpleEvaluationObject seo = createSimpleEvaluationObject(addSc, kernel, new Message(new Header(JupyterMessages.COMM_MSG, parent.getHeader().getSession())), 1); return kernel.executeCode(addSc, seo); }
private void verifyOpenCommMsgBeakerxWidget(List<Message> messages, String modelNameValue, String viewNameValue) { assertThat(messages.size()).isEqualTo(1); Message widget = messages.get(0); assertThat(widget.getHeader().getType()).isEqualTo(COMM_OPEN.getName()); Map data = TestWidgetUtils.getData(widget); assertThat(data.get(Widget.MODEL_MODULE)).isEqualTo(BeakerxWidget.MODEL_MODULE_VALUE); assertThat(data.get(Widget.VIEW_MODULE)).isEqualTo(BeakerxWidget.VIEW_MODULE_VALUE); assertThat(data.get(Widget.MODEL_NAME)).isEqualTo(modelNameValue); assertThat(data.get(Widget.VIEW_NAME)).isEqualTo(viewNameValue); }
@Before public void setUp() { this.kernel = new KernelTest(); sparkSupportActionOptions = new EnableSparkSupportOptionsTest.EnableSparkSupportActionMock(); EnableSparkSupportOptions enableSparkSupportOptions = new EnableSparkSupportOptions(sparkSupportActionOptions); sut = new RunOptionsCommand(enableSparkSupportOptions, "%%spark -v " + SPARK_VERSION, new Message(new Header(JupyterMessages.COMM_MSG, "id1"))); }
@Test public void handleMessage_secondSentMessageHasSessionId() throws Exception { //given String expectedSessionId = message.getHeader().getSession(); //when commMsgHandler.handle(message); //then Assertions.assertThat(kernel.getPublishedMessages()).isNotEmpty(); Message publishMessage = kernel.getPublishedMessages().get(1); Assertions.assertThat(publishMessage.getHeader().getSession()).isEqualTo(expectedSessionId); }
protected Message createReplyMessage(Message message, Serializable responceData) { Message ret = null; if (message != null) { ret = new Message(new Header(COMM_MSG, message.getHeader().getSession())); Map<String, Serializable> commMap = message.getContent(); HashMap<String, Serializable> map = new HashMap<>(); map.put(COMM_ID, getString(commMap, COMM_ID)); map.put(DATA, responceData); ret.setContent(map); } return ret; }