private static boolean isStderr(Message message) { return message.getContent() != null && message.getContent().get("name") != null && message.getContent().get("name").equals("stderr") && message.getContent().get("text") != null; }
public static boolean isIdleMessage(Message message) { return message.getContent() != null && message.getContent().get("execution_state") != null && message.getContent().get("execution_state").equals("idle"); }
private void verifyResultValue(Message message, String value) { Map actual = ((Map) message.getContent().get(Comm.DATA)); String result = (String) actual.get("text/plain"); assertThat(result).isEqualTo(value); } }
private void verifyFirstInput(Message message, String result) { Map actual = ((Map) message.getContent().get(Comm.DATA)); String value = (String) actual.get("text/plain"); assertThat(result).contains(value); }
private void verifySetAndGetResult(Message message) { Map actual = ((Map) message.getContent().get(Comm.DATA)); String value = (String) actual.get("text/plain"); assertThat(value).isNotEmpty(); assertThat(value).contains("123"); }
public static String initKernelCommMapWithOneComm(KernelTest kernelTest) { Message openMessage = initOpenMessage(); String commId = (String) openMessage.getContent().get(COMM_ID); String targetName = (String) openMessage.getContent().get(TARGET_NAME); Comm comm = new Comm(commId, targetName) { @Override public void handleMsg(Message parentMessage) { } }; kernelTest.addComm(commId, comm); return commId; }
public static void verifyInternalOpenCommMsgWitLayout(List<Message> messages, String modelNameValue, String viewNameValue, String modelModule, String viewModule) { Message widget = SearchMessages.getListWidgetsByViewName(messages, viewNameValue).get(0); Message layout = SearchMessages.getLayoutForWidget(messages, widget); verifyTypeMsg(widget, COMM_OPEN); Map data = getState(widget); assertThat(data.get(Layout.LAYOUT)) .isEqualTo(Layout.IPY_MODEL + layout.getContent().get(Comm.COMM_ID)); assertThat(data.get(Widget.MODEL_NAME)).isEqualTo(modelNameValue); assertThat(data.get(Widget.VIEW_NAME)).isEqualTo(viewNameValue); }
@Test public void handleMessage_sentMessageHasTargetName() throws Exception { //given String expectedTargetName = (String) message.getContent().get(TARGET_NAME); //when commOpenHandler.handle(message); //then Assertions.assertThat(kernel.getSentMessages()).isNotEmpty(); Message sendMessage = kernel.getSentMessages().get(0); Assertions.assertThat(sendMessage.getContent().get(TARGET_NAME)).isEqualTo(expectedTargetName); }
@Test public void handle_messageContentHasMatchesField() throws Exception { //when completeHandler.handle(message); //then Map<String, Serializable> content = kernel.getSentMessages().get(0).getContent(); Assertions.assertThat(content.get(CompleteHandler.MATCHES)).isNotNull(); }
@Test public void testJavaKernel() throws Exception { //given String javaCode = "%%kernel java" + System.lineSeparator() + "System.out.println(2+2);"; Code code = CodeFactory.create(javaCode, commMsg(), kernel); //when code.execute(this.kernel, 1); //then assertThat(kernel.getPublishedMessages().get(2).getContent().get("text")).isEqualTo("4"); }
@Test public void testKernelNotExist() throws Exception { //given String javaCode = "%%kernel not_exist" + System.lineSeparator() + "System.out.println(2+2);"; Code code = CodeFactory.create(javaCode, commMsg(), kernel); //when code.execute(this.kernel, 1); //then assertThat(kernel.getPublishedMessages().get(0).getContent().get("text")) .isEqualTo("Kernel not_exist is not available!\n"); }
@Test public void handleOpenCommMessage_shouldAddCommMessageToStorageMap() throws Exception { //given Message message = initCloseMessage(); String commId = (String) message.getContent().get(COMM_ID); //when commOpenHandler.handle(message); //then Assertions.assertThat(kernel.isCommPresent(commId)).isTrue(); }
public void addMvnLocal() { //given String allCode = CLASSPATH_CONFIG_RESOLVER + " " + MVN_LOCAL; Code code = CodeFactory.create(allCode, commMsg(), kernel); //when code.execute(kernel, 1); //then List<Message> std = EvaluatorResultTestWatcher.getStdouts(kernel.getPublishedMessages()); String text = (String) std.get(0).getContent().get("text"); assertThat(text).contains(ADDED_NEW_REPO + MVN_LOCAL); }
@Test public void handle_sentMessageHasContent() throws Exception { //when handler.handle(message); //then Message sentMessage = kernel.getSentMessages().get(0); Assertions.assertThat(sentMessage.getContent()).isNotEmpty(); }
@Test public void handle_sentMessageHasContent() throws Exception { //when handler.handle(message); //then Message sentMessage = kernel.getSentMessages().get(0); Assertions.assertThat(sentMessage.getContent()).isNotEmpty(); }
@Test public void commClose_sentMessageHasEmptyData() throws NoSuchAlgorithmException { initCommData(comm); //when comm.close(); //then assertThat(kernel.getPublishedMessages()).isNotEmpty(); Message sendMessage = kernel.getPublishedMessages().get(0); assertThat((Map) sendMessage.getContent().get(Comm.DATA)).isEmpty(); }
@Test public void noKeepVariablesWhenAddJar() throws Exception { //given //when runStatement("val aaa = 1 "); addDemoJar(); runStatement("aaa"); //then Optional<Message> result = waitForErrorMessage(getKernelSocketsService().getKernelSockets()); assertThat(result).isPresent(); assertThat((String) result.get().getContent().get("text")).contains("not found"); }
private void loadMagicCommandByClass() throws InterruptedException { String allCode = LOAD_MAGIC + " com.twosigma.beakerx.custom.magic.command.ShowEnvsCustomMagicCommand"; Code code = CodeFactory.create(allCode, commMsg(), getKernel()); code.execute(getKernel(), 2); List<Message> std = waitForStdouts(getKernelSocketsService().getKernelSockets()); String text = (String) std.get(0).getContent().get("text"); assertThat(text).contains("Magic command %showEnvs was successfully added."); }
@Test public void handleInfoCommMessages_replyCommMessageHasCommsInfoContent() throws Exception { //given initKernelCommMapWithOneComm(kernel); //when commInfoHandler.handle(initInfoMessage()); //then Assertions.assertThat(kernel.getSentMessages()).isNotEmpty(); Message sendMessage = kernel.getSentMessages().get(0); Assertions.assertThat((Map) sendMessage.getContent().get(COMMS)).isNotEmpty(); }
@Test public void shouldNotImportUnknownClassByMagicCommand() throws Exception { //given String allCode = IMPORT + " " + pathToDemoClassFromAddedDemoJar() + "UnknownClass"; //when Code code = CodeFactory.create(allCode, commMsg(), getKernel()); code.execute(kernel, 1); //then List<Message> std = waitForStderr(getKernelSocketsService().getKernelSockets()); String text = (String) std.get(0).getContent().get("text"); assertThat(text).contains("Could not import"); }