Convert TestCSVLoader to SolrTestCaseJ4

git-svn-id: https://svn.apache.org/repos/asf/lucene/solr/branches/newtrunk@925614 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Robert Muir 2010-03-20 14:00:37 +00:00
parent 4a75462bb8
commit 3d295fb99a
1 changed files with 18 additions and 3 deletions

View File

@ -17,29 +17,40 @@
package org.apache.solr.handler; package org.apache.solr.handler;
import org.apache.solr.SolrTestCaseJ4;
import org.apache.solr.util.AbstractSolrTestCase; import org.apache.solr.util.AbstractSolrTestCase;
import org.apache.solr.request.LocalSolrQueryRequest; import org.apache.solr.request.LocalSolrQueryRequest;
import org.apache.solr.common.util.ContentStream; import org.apache.solr.common.util.ContentStream;
import org.apache.solr.common.util.ContentStreamBase; import org.apache.solr.common.util.ContentStreamBase;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import java.io.*; import java.io.*;
import java.util.List; import java.util.List;
import java.util.ArrayList; import java.util.ArrayList;
public class TestCSVLoader extends AbstractSolrTestCase { public class TestCSVLoader extends SolrTestCaseJ4 {
public String getSchemaFile() { return "schema12.xml"; } @BeforeClass
public String getSolrConfigFile() { return "solrconfig.xml"; } public static void beforeClass() throws Exception {
initCore("solrconfig.xml","schema12.xml");
}
String filename = "solr_tmp.csv"; String filename = "solr_tmp.csv";
String def_charset = "UTF-8"; String def_charset = "UTF-8";
File file = new File(filename); File file = new File(filename);
@Before
public void setUp() throws Exception { public void setUp() throws Exception {
// if you override setUp or tearDown, you better call // if you override setUp or tearDown, you better call
// the super classes version // the super classes version
super.setUp(); super.setUp();
cleanup();
} }
@After
public void tearDown() throws Exception { public void tearDown() throws Exception {
// if you override setUp or tearDown, you better call // if you override setUp or tearDown, you better call
// the super classes version // the super classes version
@ -81,6 +92,7 @@ public class TestCSVLoader extends AbstractSolrTestCase {
h.query("/update/csv",req); h.query("/update/csv",req);
} }
@Test
public void testCSVLoad() throws Exception { public void testCSVLoad() throws Exception {
makeFile("id\n100\n101\n102"); makeFile("id\n100\n101\n102");
loadLocal("stream.file",filename); loadLocal("stream.file",filename);
@ -90,6 +102,7 @@ public class TestCSVLoader extends AbstractSolrTestCase {
assertQ(req("id:[100 TO 110]"),"//*[@numFound='3']"); assertQ(req("id:[100 TO 110]"),"//*[@numFound='3']");
} }
@Test
public void testCommitFalse() throws Exception { public void testCommitFalse() throws Exception {
makeFile("id\n100\n101\n102"); makeFile("id\n100\n101\n102");
loadLocal("stream.file",filename,"commit","false"); loadLocal("stream.file",filename,"commit","false");
@ -98,12 +111,14 @@ public class TestCSVLoader extends AbstractSolrTestCase {
assertQ(req("id:[100 TO 110]"),"//*[@numFound='3']"); assertQ(req("id:[100 TO 110]"),"//*[@numFound='3']");
} }
@Test
public void testCommitTrue() throws Exception { public void testCommitTrue() throws Exception {
makeFile("id\n100\n101\n102"); makeFile("id\n100\n101\n102");
loadLocal("stream.file",filename,"commit","true"); loadLocal("stream.file",filename,"commit","true");
assertQ(req("id:[100 TO 110]"),"//*[@numFound='3']"); assertQ(req("id:[100 TO 110]"),"//*[@numFound='3']");
} }
@Test
public void testCSV() throws Exception { public void testCSV() throws Exception {
lrf.args.put("version","2.0"); lrf.args.put("version","2.0");