Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

LUCENE-4588 (cont): (EnwikiContentSource fixes) avoid using the forbidden StringBufferInputStream.. 

git-svn-id: https://svn.apache.org/repos/asf/lucene/dev/trunk@1417871 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Doron Cohen 2012-12-06 13:33:34 +00:00
parent e7cc2239fd
commit 72d71de227
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lucene/benchmark/src/test/org/apache/lucene/benchmark/byTask/feeds/EnwikiContentSourceTest.java
View File

@ -17,14 +17,17 @@ package org.apache.lucene.benchmark.byTask.feeds;
* limitations under the License.
*/
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.text.ParseException;
import java.util.Properties;
import org.apache.lucene.benchmark.byTask.utils.Config;
import org.apache.lucene.util.IOUtils;
import org.apache.lucene.util.LuceneTestCase;
import org.junit.Ignore;
import org.junit.Test;
public class EnwikiContentSourceTest extends LuceneTestCase {
@ -38,10 +41,16 @@ public class EnwikiContentSourceTest extends LuceneTestCase {
this.docs = docs;
}
@SuppressWarnings("deprecation") // fine for the characters used in this test
@Override
protected InputStream openInputStream() throws IOException {
return new java.io.StringBufferInputStream(docs);
// StringBufferInputStream would have been handy, but it is forbidden
ByteArrayOutputStream baos = new ByteArrayOutputStream();
OutputStreamWriter w = new OutputStreamWriter(baos, IOUtils.CHARSET_UTF_8);
w.write(docs);
w.close();
byte[] byteArray = baos.toByteArray();
baos.close();
return new ByteArrayInputStream(byteArray);
}
}