CommandTest.java

package nl.knaw.huygens.alexandria.client;

/*
 * #%L
 * alexandria-java-client
 * =======
 * Copyright (C) 2015 - 2017 Huygens ING (KNAW)
 * =======
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import static org.assertj.core.api.Assertions.assertThat;

import java.util.Map;
import java.util.UUID;

import org.junit.Before;
import org.junit.Test;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;

import nl.knaw.huygens.Log;
import nl.knaw.huygens.alexandria.api.model.CommandResponse;
import nl.knaw.huygens.alexandria.api.model.CommandStatus;
import nl.knaw.huygens.alexandria.api.model.Commands;
import nl.knaw.huygens.alexandria.endpoint.command.XpathCommand.XPathResult;

public class CommandTest extends AlexandriaClientTest {

  @Before
  public void before() {
    client.setAuthKey(AUTHKEY);
    client.setAutoConfirm(true);
  }

  @Test
  public void testAddUniqueIdCommmandWorks() {
    UUID resourceUuid = createResource("xml");
    String xml = "<text><p>Alinea 1</p><p>Alinea 2</p></text>";
    String expectedXml = singleQuotesToDouble("<text xml:id='text-1'><p xml:id='p-1'>Alinea 1</p><p xml:id='p-2'>Alinea 2</p></text>");
    setResourceText(resourceUuid, xml);
    Map<String, Object> parameters = ImmutableMap.<String, Object> builder()//
        .put("resourceIds", ImmutableList.of(resourceUuid))//
        .put("elements", ImmutableList.of("text", "p"))//
        .build();
    RestResult<CommandResponse> result = client.doCommand(Commands.ADD_UNIQUE_ID, parameters);
    assertRequestSucceeded(result);
    CommandResponse commandResponse = result.get();
    assertThat(commandResponse.success()).isTrue();

    RestResult<String> xmlReadResult = client.getTextAsString(resourceUuid);
    assertRequestSucceeded(xmlReadResult);
    String xml2 = xmlReadResult.get();
    assertThat(xml2).isEqualTo(expectedXml);
  }

  @Test
  public void testXpathCommmandWorks() {
    UUID resourceUuid = createResource("xml");
    String xml = "<text><p>Alinea 1</p><p>Alinea 2</p></text>";
    String expected = "Alinea 2";
    setResourceText(resourceUuid, xml);
    Map<String, Object> parameters = ImmutableMap.<String, Object> builder()//
        .put("resourceIds", ImmutableList.of(resourceUuid))//
        .put("xpath", "string(/text/p[2])")//
        .build();
    RestResult<CommandResponse> result = client.doCommand(Commands.XPATH, parameters);
    assertRequestSucceeded(result);
    CommandResponse commandResponse = result.get();
    assertThat(commandResponse.success()).isTrue();
    Map<String, Map<String, String>> result2 = (Map<String, Map<String, String>>) commandResponse.getResult();
    Map<String, String> map = result2.get(resourceUuid.toString());
    XPathResult returnedXPathResult = new XPathResult(XPathResult.Type.valueOf(map.get("type")), map.get("result"));
    XPathResult expectedXPathResult = new XPathResult(XPathResult.Type.STRING, expected);
    assertThat(returnedXPathResult).isEqualToComparingFieldByField(expectedXPathResult);
  }

  @Test
  public void testAQL2CommmandWorks() throws InterruptedException {
    UUID resourceUuid = createResource("xml");
    String xml = "<text><p>Alinea 1</p><p>Alinea 2</p></text>";
    setResourceText(resourceUuid, xml);
    Map<String, Object> parameters = ImmutableMap.<String, Object> builder()//
        .put("command", "bye(\"world\")")//
        .build();
    RestResult<CommandResponse> result = client.doCommand(Commands.AQL2, parameters);
    assertRequestSucceeded(result);

    CommandResponse commandResponse = result.get();
    UUID statusId = commandResponse.getStatusId();
    Log.info("statusId={}", statusId);

    boolean done = false;
    RestResult<CommandStatus> statusResult = null;
    while (!done) {
      Thread.sleep(1000);
      statusResult = client.getCommandStatus(Commands.AQL2, statusId);
      assertRequestSucceeded(statusResult);
      done = statusResult.get().isDone();
    }
    Object result2 = statusResult.get().getResult();
    Log.info("{}", result2);
    assertThat(result2).isEqualTo("Goodbye world!");
  }

  // @Test
  public void testAnnotationCommandWorks() {
    UUID resourceUuid = createResource("xml");
    String xml = "<root>"//
        + "<p xml:id=\"p-1\">"//
        + "<seg xml:id=\"seg-1\">Some</seg>"//
        + " text</p>"//
        + "</root>";
    // String expectedXml = "<root>"//
    // + "<p xml:id=\"p-1\"><annotation xml:lang=\"?\">"//
    // + "<seg xml:id=\"seg-1\">Some</seg>"//
    // + " text</annotation></p>"//
    // + "</root>";
    String expectedXml = "<root>"//
        + "<p xml:id=\"p-1\"><annotation xml:lang=\"?\">"//
        + "<seg xml:id=\"seg-1\"><annotation xml:lang=\"fr\">Some</annotation></seg>"//
        + " text</annotation></p>"//
        + "</root>";
    setResourceText(resourceUuid, xml);

    Map<String, String> annotations = ImmutableMap.of(//
        "seg-1", "fr", //
        "p-1", "?");

    for (Map.Entry<String, String> entry : annotations.entrySet()) {
      Map<String, Object> attributes = ImmutableMap.<String, Object> builder()//
          .put("xml:lang", entry.getValue())//
          .build();
      Map<String, Object> element = ImmutableMap.<String, Object> builder()//
          .put("name", "annotation")//
          .put("attributes", attributes)//
          .build();
      Map<String, Object> parameters = ImmutableMap.<String, Object> builder()//
          .put("resourceIds", ImmutableList.of(resourceUuid))//
          .put("xmlIds", ImmutableList.of(entry.getKey()))//
          .put("element", element)//
          .build();
      RestResult<CommandResponse> result = client.doCommand(Commands.WRAP_CONTENT_IN_ELEMENT, parameters);
      assertRequestSucceeded(result);
      CommandResponse commandResponse = result.get();
      assertThat(commandResponse.success()).isTrue();
    }

    RestResult<String> xmlReadResult = client.getTextAsString(resourceUuid);
    assertRequestSucceeded(xmlReadResult);
    String xml2 = xmlReadResult.get();
    assertThat(xml2).isEqualTo(expectedXml);
  }

  // @Test
  public void testAnnotationCommandWorks2() {
    UUID resourceUuid = createResource("xml");
    String xml = "<p xml:id=\"p-10\">Tuus\n"//
        + "<lb/><persName key=\"beeckman.isaac.1588-1637\" resp=\"#ckcc\">Isacus Beeckmannus</persName>"//
        + "</p>";
    String expectedXml = "<p xml:id=\"p-10\"><p_lang value=\"la\" resp=\"#ckcc\"><p_type value=\"closer\" resp=\"#ckcc\">Tuus\n"//
        + "<lb/><persName key=\"beeckman.isaac.1588-1637\" resp=\"#ckcc\">Isacus Beeckmannus</persName>"//
        + "</p_type></p_lang></p>";
    setResourceText(resourceUuid, xml);

    annotate(resourceUuid, element("p_type", attributes("closer")));
    annotate(resourceUuid, element("p_lang", attributes("la")));

    RestResult<String> xmlReadResult = client.getTextAsString(resourceUuid);
    assertRequestSucceeded(xmlReadResult);
    String xml2 = xmlReadResult.get();
    assertThat(xml2).isEqualTo(expectedXml);
  }

  private void annotate(UUID resourceUuid, Map<String, Object> element) {
    Map<String, Object> parameters1 = ImmutableMap.<String, Object> builder()//
        .put("resourceIds", ImmutableList.of(resourceUuid))//
        .put("xmlIds", ImmutableList.of("p-10"))//
        .put("element", element)//
        .build();
    RestResult<CommandResponse> result = client.doCommand(Commands.WRAP_CONTENT_IN_ELEMENT, parameters1);
    assertRequestSucceeded(result);
    CommandResponse commandResponse = result.get();
    assertThat(commandResponse.success()).isTrue();
  }

  private Map<String, Object> attributes(String value) {
    return ImmutableMap.<String, Object> builder()//
        .put("value", value)//
        .put("resp", "#ckcc")//
        .build();
  }

  private Map<String, Object> element(String name, Map<String, Object> attributes) {
    return ImmutableMap.<String, Object> builder()//
        .put("name", name)//
        .put("attributes", attributes)//
        .build();
  }

}