| package org.apache.solr.search.function; |
| /* |
| * 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. |
| */ |
| |
| import org.apache.solr.util.AbstractSolrTestCase; |
| import org.junit.BeforeClass; |
| |
| |
| /** |
| * |
| * |
| **/ |
| public class SortByFunctionTest extends AbstractSolrTestCase { |
| |
| |
| @BeforeClass |
| public static void beforeClass() throws Exception { |
| initCore("solrconfig.xml", "schema.xml"); |
| } |
| |
| |
| @Override |
| public void setUp() throws Exception { |
| super.setUp(); |
| |
| } |
| |
| public void test() throws Exception { |
| assertU(adoc("id", "1", "x_td1", "0", "y_td1", "2", "w_td1", "25", "z_td1", "5", "f_t", "ipod")); |
| assertU(adoc("id", "2", "x_td1", "2", "y_td1", "2", "w_td1", "15", "z_td1", "5", "f_t", "ipod ipod ipod ipod ipod")); |
| assertU(adoc("id", "3", "x_td1", "3", "y_td1", "2", "w_td1", "55", "z_td1", "5", "f_t", "ipod ipod ipod ipod ipod ipod ipod ipod ipod")); |
| assertU(adoc("id", "4", "x_td1", "4", "y_td1", "2", "w_td1", "45", "z_td1", "5", "f_t", "ipod ipod ipod ipod ipod ipod ipod")); |
| assertU(commit()); |
| |
| assertQ(req("fl", "*,score", "q", "*:*"), |
| "//*[@numFound='4']", |
| "//float[@name='score']='1.0'", |
| "//result/doc[1]/int[@name='id'][.='1']", |
| "//result/doc[2]/int[@name='id'][.='2']", |
| "//result/doc[3]/int[@name='id'][.='3']", |
| "//result/doc[4]/int[@name='id'][.='4']" |
| ); |
| assertQ(req("fl", "*,score", "q", "*:*", "sort", "score desc"), |
| "//*[@numFound='4']", |
| "//float[@name='score']='1.0'", |
| "//result/doc[1]/int[@name='id'][.='1']", |
| "//result/doc[2]/int[@name='id'][.='2']", |
| "//result/doc[3]/int[@name='id'][.='3']", |
| "//result/doc[4]/int[@name='id'][.='4']" |
| ); |
| assertQ(req("fl", "id,score", "q", "f_t:ipod", "sort", "score desc"), |
| "//*[@numFound='4']", |
| "//result/doc[1]/int[@name='id'][.='1']", |
| "//result/doc[2]/int[@name='id'][.='4']", |
| "//result/doc[3]/int[@name='id'][.='2']", |
| "//result/doc[4]/int[@name='id'][.='3']" |
| ); |
| |
| |
| assertQ(req("fl", "*,score", "q", "*:*", "sort", "sum(x_td1, y_td1) desc"), |
| "//*[@numFound='4']", |
| "//float[@name='score']='1.0'", |
| "//result/doc[1]/int[@name='id'][.='4']", |
| "//result/doc[2]/int[@name='id'][.='3']", |
| "//result/doc[3]/int[@name='id'][.='2']", |
| "//result/doc[4]/int[@name='id'][.='1']" |
| ); |
| assertQ(req("fl", "*,score", "q", "*:*", "sort", "sum(x_td1, y_td1) asc"), |
| "//*[@numFound='4']", |
| "//float[@name='score']='1.0'", |
| "//result/doc[1]/int[@name='id'][.='1']", |
| "//result/doc[2]/int[@name='id'][.='2']", |
| "//result/doc[3]/int[@name='id'][.='3']", |
| "//result/doc[4]/int[@name='id'][.='4']" |
| ); |
| //the function is equal, w_td1 separates |
| assertQ(req("q", "*:*", "fl", "id", "sort", "sum(z_td1, y_td1) asc, w_td1 asc"), |
| "//*[@numFound='4']", |
| "//result/doc[1]/int[@name='id'][.='2']", |
| "//result/doc[2]/int[@name='id'][.='1']", |
| "//result/doc[3]/int[@name='id'][.='4']", |
| "//result/doc[4]/int[@name='id'][.='3']" |
| ); |
| } |
| |
| public void testSortJoinDocFreq() throws Exception |
| { |
| assertU(adoc("id", "4", "id_s", "D", "links_mfacet", "A", "links_mfacet", "B", "links_mfacet", "C" ) ); |
| assertU(adoc("id", "3", "id_s", "C", "links_mfacet", "A", "links_mfacet", "B" ) ); |
| assertU(commit()); // Make sure it uses two readers |
| assertU(adoc("id", "2", "id_s", "B", "links_mfacet", "A" ) ); |
| assertU(adoc("id", "1", "id_s", "A" ) ); |
| assertU(commit()); |
| |
| assertQ(req("q", "links_mfacet:B", "fl", "id", "sort", "id asc"), |
| "//*[@numFound='2']", |
| "//result/doc[1]/int[@name='id'][.='3']", |
| "//result/doc[2]/int[@name='id'][.='4']" |
| ); |
| |
| assertQ(req("q", "*:*", "fl", "id", "sort", "joindf(id_s, links_mfacet) desc"), |
| "//*[@numFound='4']", |
| "//result/doc[1]/int[@name='id'][.='1']", |
| "//result/doc[2]/int[@name='id'][.='2']", |
| "//result/doc[3]/int[@name='id'][.='3']", |
| "//result/doc[4]/int[@name='id'][.='4']" |
| ); |
| |
| assertQ(req("q", "*:*", "fl", "id", "sort", "joindf(id_s, links_mfacet) asc"), |
| "//*[@numFound='4']", |
| "//result/doc[1]/int[@name='id'][.='4']", |
| "//result/doc[2]/int[@name='id'][.='3']", |
| "//result/doc[3]/int[@name='id'][.='2']", |
| "//result/doc[4]/int[@name='id'][.='1']" |
| ); |
| } |
| } |