SolrTermQueryCachePreloadTest.java

package querqy.solr;

import org.apache.solr.SolrTestCaseJ4;
import org.apache.solr.common.params.CommonParams;
import org.apache.solr.common.params.DisMaxParams;
import org.apache.solr.request.SolrQueryRequest;
import org.apache.solr.search.QueryParsing;
import org.junit.BeforeClass;
import org.junit.Test;

@SolrTestCaseJ4.SuppressSSL
public class SolrTermQueryCachePreloadTest extends SolrTestCaseJ4 {

    @BeforeClass
    public static void beforeTest() throws Exception{
        initCore("solrconfig-cache-preloaded.xml", "schema.xml");
    }
     
    @Test
    public void testThatCacheIsAvailableAndPrefilledAndNotUpdated() throws Exception {

        // firstSearcher
        SolrQueryRequest req = req(
               CommonParams.QT, "/admin/mbeans",
               "cat", "CACHE",
               "stats", "true"
               );
        // the cache is prefilled asynchronously - retry 3 times to see the cache before giving up
        int attempts = 3;
        try {
            
            do {
                
                try {
                    assertQ("Missing querqy cache",
                       req,
                       "//lst[@name='CACHE']/lst[@name='querqyTermQueryCache']");
                    attempts = 0;
                }  catch (RuntimeException e) {
                    if ((!"Exception during query".equals(e.getMessage())) || (attempts <= 1)) {
                        throw e;
                    }
                    attempts--;
                    synchronized(this) {
                        wait(100L);
                    }
                }   
            } while (attempts > 0);
            
             // only one generated term in one field is preloaded for firstSearcher:
            assertQ("Querqy cache not prefilled",
                     req,
                   "//lst[@name='CACHE']/lst[@name='querqyTermQueryCache']"
                           + "/lst[@name='stats']/long[@name='size'][text()='1']");
        } finally {
            req.close();
        }
        
        assertU(adoc("id", "1", "f1", "a"));
        assertU(commit());
         
        // newSearcher
        SolrQueryRequest req2 = req(
                 CommonParams.QT, "/admin/mbeans",
                 "cat", "CACHE",
                 "stats", "true"
                 );
         
        // one generated term in two fields is preloaded for newSearcher:
        assertQ("Querqy cache not prefilled",
                 req2,
                 "//lst[@name='CACHE']/lst[@name='querqyTermQueryCache']"
                         + "/lst[@name='stats']/long[@name='size'][text()='2']");

        req2.close();
         
        String q = "a b c";
        SolrQueryRequest req3 = req(
                 
                 CommonParams.Q, q,
                 DisMaxParams.QF, "f1 f2",
                 QueryParsing.OP, "AND",
                 "defType", "querqy",
                 "debugQuery", "true"
                 );
         
        // f1:b and f2:b would be produced by synonym rule, but
        // due to pre-testing for hits in preload they should not
        // occur in the parsed query
        assertQ("Terms w/o hits found in parsedquery",
                 req3,
                 "//result[@name='response'][@numFound='0']",
                 "//str[@name='parsedquery'][not(contains(.,'f1:b'))]",
                 "//str[@name='parsedquery'][not(contains(.,'f2:b'))]"
                );

        req3.close();
         
         
        SolrQueryRequest reqStats = req(
                 CommonParams.QT, "/admin/mbeans",
                 "cat", "CACHE",
                 "stats", "true"
                 );
         
        assertQ("Querqy cache was updated unexpectedly",
                 reqStats,
                 "//lst[@name='CACHE']/lst[@name='querqyTermQueryCache']"
                         + "/lst[@name='stats']/long[@name='size'][text()='2']");

        reqStats.close();
         
         
         
    }
   
}