solr/core/src/test/org/apache/solr/handler/tagger/EmbeddedSolrNoSerializeTest.java

/*
 * This software was produced for the U. S. Government
 * under Contract No. W15P7T-11-C-F600, and is
 * subject to the Rights in Noncommercial Computer Software
 * and Noncommercial Computer Software Documentation
 * Clause 252.227-7014 (JUN 1995)
 *
 * Copyright 2013 The MITRE Corporation. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.solr.handler.tagger;

import java.io.IOException;
import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiFunction;

import org.apache.lucene.document.Field;
import org.apache.solr.SolrTestCaseJ4;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.StreamingResponseCallback;
import org.apache.solr.client.solrj.embedded.EmbeddedSolrServer;
import org.apache.solr.client.solrj.request.QueryRequest;
import org.apache.solr.client.solrj.request.RequestWriter;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.common.params.SolrParams;
import org.apache.solr.common.util.ContentStream;
import org.apache.solr.common.util.ContentStreamBase;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Ignore;
import org.junit.Test;

/**
 * Tests that we can skip serialization of the documents when embedding
 * Solr.
 */
public class EmbeddedSolrNoSerializeTest extends SolrTestCaseJ4 {

  static EmbeddedSolrServer solrServer;

  @BeforeClass
  public static void init() throws Exception {
    initCore("solrconfig-tagger.xml", "schema-tagger.xml");
    solrServer = new EmbeddedSolrServer(h.getCoreContainer(), "collection1");
    //we don't need to close the EmbeddedSolrServer because SolrTestCaseJ4 closes the core
  }

  @AfterClass
  public static void cleanUpAfterClass() throws Exception {
    solrServer = null;
  }

  @Before
  public void setUp() throws Exception {
    super.setUp();
    clearIndex();
    assertU(adoc("id", "9999", "name", "Boston"));
    assertU(commit());
  }

  @Test
  public void testTag() throws SolrServerException, IOException {
    ModifiableSolrParams params = params();
    String input = "foo boston bar";//just one tag;
    QueryRequest req = new SolrTaggerRequest(params, input);
    req.setPath("/tag");

    QueryResponse rsp = req.process(solrServer);
    SolrDocumentList results= (SolrDocumentList) rsp.getResponse().get("response");
    assertNotNull(rsp.getResponse().get("tags"));
    assertNotNull(results.get(0));
  }

  @SuppressWarnings("serial")
  public static class SolrTaggerRequest extends QueryRequest {

    private final String input;

    public SolrTaggerRequest(SolrParams p, String input) {
      super(p, METHOD.POST);
      this.input = input;
    }

    // Deprecated in 7.2 but should live on until 8.x
    @SuppressWarnings("deprecation")
    @Override
    public Collection<ContentStream> getContentStreams() {
      return Collections.singleton(new ContentStreamBase.StringStream(input));
    }

    //     As of 7.2.  But won't work until: https://issues.apache.org/jira/browse/SOLR-12142
    @Override
    public RequestWriter.ContentWriter getContentWriter(String expectedType) {
      return new RequestWriter.StringPayloadContentWriter(input, "text/plain; charset=UTF8");
    }
  }

  @Test
  public void testSearch() throws Exception {
    QueryResponse rsp = solrServer.query(params("q", "name:Boston"));
    assertNotNull(rsp.getResults().get(0));
  }

  @Test
  public void testAssertTagStreamingWithSolrTaggerRequest() throws Exception {
    doTestAssertTagStreaming(SolrTaggerRequest::new);
  }

  @Test
  @Ignore("As of Solr 7, stream.body is disabled by default for security ") // DWS: dubious, IMO
  // and it can't be enabled with EmbeddedSolrServer until SOLR-12126
  public void testAssertTagStreamingWithStreamBodyParam() throws Exception {
    doTestAssertTagStreaming((params, input) -> {
      params.set("stream.body", input);
      return new QueryRequest(params);
    });
  }

  public void doTestAssertTagStreaming(BiFunction<ModifiableSolrParams,String,QueryRequest> newQueryRequest) throws IOException, SolrServerException {
    ModifiableSolrParams params = params();
    String input = "foo boston bar";//just one tag;
    QueryRequest req = newQueryRequest.apply(params, input);
    req.setPath("/tag");

    final AtomicReference<SolrDocument> refDoc = new AtomicReference<>();
    req.setStreamingResponseCallback(new StreamingResponseCallback() {
      @Override
      public void streamSolrDocument(SolrDocument doc) {
        refDoc.set(doc);
      }

      @Override
      public void streamDocListInfo(long numFound, long start, Float maxScore) {

      }
    });
    QueryResponse rsp = req.process(solrServer);
    assertNotNull(rsp.getResponse().get("tags"));
    assertNotNull(refDoc.get());
    assertEquals("Boston", ((Field)refDoc.get().getFieldValue("name")).stringValue());
  }
}