MoreLikeThisHandler Unit Test

[Tom Morton]

I'm trying to write some unit tests for the MoreLikeThisHandler which test
the functionality.  I've added documents, committed them, but can't get the
query to return a single document from which mlt would get terms.  It's as
if the documents aren't making it to the index or I'm query them
incorrectly.  I'm stuck at this point.  Can someone take a look at the
attached unit tests and let me know if I'm missing something.

Thanks...Tom

/**
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

import org.apache.solr.common.params.CommonParams;
import org.apache.solr.common.params.MoreLikeThisParams;
import org.apache.solr.common.params.MultiMapSolrParams;
import org.apache.solr.common.params.SolrParams;
import org.apache.solr.common.util.ContentStream;
import org.apache.solr.common.util.ContentStreamBase;
import org.apache.solr.core.SolrCore;
import org.apache.solr.request.SolrQueryRequestBase;
import org.apache.solr.request.SolrQueryResponse;
import org.apache.solr.search.DocSlice;
import org.apache.solr.util.AbstractSolrTestCase;


/**
 * TODO -- this needs to actually test the results/query etc
 */
public class MoreLikeThisHandlerTest extends AbstractSolrTestCase {

  @Override public String getSchemaFile() { return "schema.xml"; }
  @Override public String getSolrConfigFile() { return "solrconfig.xml"; }
  @Override public void setUp() throws Exception {
    super.setUp();
    lrf = h.getRequestFactory("standard", 0, 20 );
  }
  
  public void testInterface()
  {
    SolrCore core = h.getCore();
    MoreLikeThisHandler mlt = new MoreLikeThisHandler();
    
    Map<String,String[]> params = new HashMap<String,String[]>();
    MultiMapSolrParams mmparams = new MultiMapSolrParams( params );
    SolrQueryRequestBase req = new SolrQueryRequestBase( core, 
(SolrParams)mmparams ) {};
    
    // requires 'q' or single content stream
    try {
      mlt.handleRequestBody( req, new SolrQueryResponse() );
    }
    catch( Exception ex ) {} // expected

    // requires 'q' or single content stream
    try {
      ArrayList<ContentStream> streams = new ArrayList<ContentStream>( 2 );
      streams.add( new ContentStreamBase.StringStream( "hello" ) );
      streams.add( new ContentStreamBase.StringStream( "there" ) );
      req.setContentStreams( streams );
      mlt.handleRequestBody( req, new SolrQueryResponse() );
    }
    catch( Exception ex ) {} // expected
    
    adoc(new String[]{"id","42","name","Tom Cruise","subword","Top 
Gun","subword","Risky Business","subword","The Color of 
Money","subword","Minority Report","subword", "Days of Thunder","subword", 
"Eyes Wide Shut","subword", "Far and Away"});
    adoc(new String[]{"id","43","name","Tom Hanks","subword","The Green 
Mile","subword","Forest Gump","subword","Philadelphia 
Story","subword","Big","subword","Cast Away"});
    adoc(new String[]{"id","44","name","Harrison Ford","subword","Star 
Wars","subword","Indiana Jones","subword","Patriot Games","subword","Regarding 
Henry"});
    adoc(new String[]{"id","45","name","George Harrison","subword","Yellow 
Submarine","subword","Help","subword","Magical Mystery Tour","subword","Sgt. 
Peppers Lonley Hearts Club Band"});
    adoc(new String[]{"id","46","name","Nicole 
Kidman","subword","Batman","subword","Days of Thunder","subword","Eyes Wide 
Shut","subword","Far and Away"});
    commit();

    params.put(CommonParams.Q, new String[]{"id:42"});
    params.put(MoreLikeThisParams.SIMILARITY_FIELDS, new 
String[]{"name,subword"});
    params.put(MoreLikeThisParams.INTERESTING_TERMS,new String[]{"details"});
    params.put(MoreLikeThisParams.MIN_TERM_FREQ,new String[]{"1"});
    params.put(MoreLikeThisParams.MIN_DOC_FREQ,new String[]{"1"});
    params.put(MoreLikeThisParams.MAX_QUERY_TERMS,new String[]{"5"});
    params.put(MoreLikeThisParams.MATCH_INCLUDE,new String[]{"true"});
    
    try {
      SolrQueryResponse rsp = new SolrQueryResponse();
      mlt.handleRequestBody(req, rsp);
      String response = (String) rsp.getValues().get("response");
      this.assertNotNull(response);
    } catch (Exception e) {
      e.printStackTrace();
    }
  }
}