/* LanguageTool, a natural language style checker
* Copyright (C) 2014 Daniel Naber (http://www.danielnaber.de)
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
* USA
*/
package org.languagetool.server;
import org.junit.Ignore;
import org.junit.Test;
import org.languagetool.Language;
import org.languagetool.tools.StringTools;
import org.xml.sax.SAXException;
import javax.xml.parsers.ParserConfigurationException;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Test HTTP server access from multiple threads with multiple languages.
*/
@Ignore("for interactive use; requires local Tatoeba data")
public class HTTPServerMultiLangLoadTest extends HTTPServerLoadTest {
private static final int MAX_TEXT_LENGTH = 5000;
private static final int MAX_SLEEP_MILLIS = 10;
private final Map<Language, String> langCodeToText = new HashMap<>();
private final Random random = new Random(1234);
private final AtomicInteger counter = new AtomicInteger();
@Test
@Override
public void testHTTPServer() throws Exception {
File dir = new File("/media/Data/tatoeba/");
for (Language language : Language.REAL_LANGUAGES) {
File file = new File(dir, "tatoeba-" + language.getShortName() + ".txt");
if (!file.exists()) {
System.err.println("No data found for " + language + ", language will not be tested");
} else {
String content = StringTools.readerToString(new FileReader(file));
langCodeToText.put(language, content);
System.err.println("Using " + content.length() + " bytes of data for " + language);
}
}
super.testHTTPServer();
}
@Override
protected int getThreadCount() {
return 4;
}
@Override
protected int getRepeatCount() {
return Integer.MAX_VALUE;
}
@Override
void runTests() throws IOException, SAXException, ParserConfigurationException {
Language language = getRandomLanguage();
String text = langCodeToText.get(language);
int fromPos = random.nextInt(text.length());
int toPos = fromPos + random.nextInt(MAX_TEXT_LENGTH);
String textSubstring = text.substring(fromPos, Math.min(toPos, text.length()));
long sleepTime = random.nextInt(MAX_SLEEP_MILLIS);
try {
Thread.sleep(sleepTime);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
long startTime = System.currentTimeMillis();
counter.incrementAndGet();
checkByPOST(language, textSubstring);
System.out.println(counter.get() + ". Sleep: " + sleepTime + "ms, Lang: " + language.getShortNameWithCountryAndVariant()
+ ", Time: " + (System.currentTimeMillis()-startTime) + "ms");
}
private Language getRandomLanguage() {
Random rnd = new Random();
int randomNumber = rnd.nextInt(langCodeToText.size());
int i = 0;
for (Language lang : langCodeToText.keySet()) {
if (i++ == randomNumber) {
return lang;
}
}
throw new RuntimeException("Could not find a random language (" + i + ")");
}
}