Java tutorial
/* * #%L * Alfresco System Build Test * %% * Copyright (C) 2005 - 2016 Alfresco Software Limited * %% * This file is part of the Alfresco software. * If the software was purchased under a paid Alfresco license, the terms of * the paid license agreement will prevail. Otherwise, the software is * provided under the following open source license terms: * * Alfresco 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 3 of the License, or * (at your option) any later version. * * Alfresco 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 Alfresco. If not, see <http://www.gnu.org/licenses/>. * #L% */ package org.alfresco.repo.solr; import java.io.BufferedReader; import java.io.FileNotFoundException; import java.io.IOException; import java.io.InputStream; import java.io.InputStreamReader; import java.io.Reader; import java.io.UnsupportedEncodingException; import java.util.HashMap; import java.util.Iterator; import java.util.Properties; import javax.servlet.http.HttpServletResponse; import junit.framework.TestCase; import org.alfresco.encryption.KeyResourceLoader; import org.alfresco.encryption.KeyStoreParameters; import org.alfresco.encryption.ssl.SSLEncryptionParameters; import org.alfresco.httpclient.HttpClientFactory; import org.alfresco.httpclient.HttpClientFactory.SecureCommsType; import org.alfresco.repo.search.impl.lucene.LuceneQueryParserException; import org.apache.commons.codec.net.URLCodec; import org.apache.commons.httpclient.Header; import org.apache.commons.httpclient.HttpClient; import org.apache.commons.httpclient.HttpException; import org.apache.commons.httpclient.HttpStatus; import org.apache.commons.httpclient.URI; import org.apache.commons.httpclient.UsernamePasswordCredentials; import org.apache.commons.httpclient.auth.AuthScope; import org.apache.commons.httpclient.methods.GetMethod; import org.apache.commons.httpclient.params.HttpClientParams; import org.json.JSONArray; import org.json.JSONException; import org.json.JSONObject; import org.json.JSONTokener; /** * @author Andy * */ public class EmbeddedSolrTest extends TestCase { private String baseUrl; private HttpClient httpClient; @Override public void setUp() throws Exception { KeyStoreParameters keyStoreParameters = new KeyStoreParameters("SSL Key Store", "JCEKS", null, "ssl-keystore-passwords.properties", "ssl.keystore"); KeyStoreParameters trustStoreParameters = new KeyStoreParameters("SSL Trust Store", "JCEKS", null, "ssl-truststore-passwords.properties", "ssl.truststore"); SSLEncryptionParameters sslEncryptionParameters = new SSLEncryptionParameters(keyStoreParameters, trustStoreParameters); ClasspathKeyResourceLoader keyResourceLoader = new ClasspathKeyResourceLoader(); HttpClientFactory httpClientFactory = new HttpClientFactory(SecureCommsType.getType("https"), sslEncryptionParameters, keyResourceLoader, null, null, "localhost", 8080, 8443, 40, 40, 0); StringBuilder sb = new StringBuilder(); sb.append("/solr/admin/cores"); this.baseUrl = sb.toString(); httpClient = httpClientFactory.getHttpClient(); HttpClientParams params = httpClient.getParams(); params.setBooleanParameter(HttpClientParams.PREEMPTIVE_AUTHENTICATION, true); httpClient.getState().setCredentials(new AuthScope(AuthScope.ANY_HOST, AuthScope.ANY_PORT), new UsernamePasswordCredentials("admin", "admin")); } public void testEmbeddedAFTS() throws JSONException { HashMap<String, String> args = new HashMap<String, String>(); args.put("action", "TEST"); args.put("wt", "json"); JSONObject json = execute(args); Counter errors = new Counter(); Counter failures = new Counter(); countErrorsAndFailures(json, errors, failures); assertEquals("Errors", 0, errors.value()); assertEquals("Failures", 0, failures.value()); } public void testEmbeddedCmis() throws JSONException { HashMap<String, String> args = new HashMap<String, String>(); args.put("action", "CMISTEST"); args.put("wt", "json"); JSONObject json = execute(args); Counter errors = new Counter(); Counter failures = new Counter(); countErrorsAndFailures(json, errors, failures); assertEquals("Errors", 0, errors.value()); assertEquals("Failures", 0, failures.value()); } private void countErrorsAndFailures(JSONObject object, Counter errors, Counter failures) throws JSONException { for (Iterator<?> it = object.keys(); it.hasNext(); /**/) { String key = (String) it.next(); Object current = object.get(key); if (current instanceof JSONObject) { countErrorsAndFailures((JSONObject) current, errors, failures); } else if (current instanceof JSONArray) { } else { String value = object.getString(key); if (value.contains("ERROR")) { errors.increment(); } if (value.contains("FAILED")) { failures.increment(); } } } } private class ClasspathKeyResourceLoader implements KeyResourceLoader { @Override public InputStream getKeyStore(String location) throws FileNotFoundException { return getClass().getClassLoader().getResourceAsStream(location); } @Override public Properties loadKeyMetaData(String location) throws IOException { Properties p = new Properties(); p.load(getClass().getClassLoader().getResourceAsStream(location)); return p; } } public JSONObject execute(HashMap<String, String> args) { try { URLCodec encoder = new URLCodec(); StringBuilder url = new StringBuilder(); for (String key : args.keySet()) { String value = args.get(key); if (url.length() == 0) { url.append(baseUrl); url.append("?"); url.append(encoder.encode(key, "UTF-8")); url.append("="); url.append(encoder.encode(value, "UTF-8")); } else { url.append("&"); url.append(encoder.encode(key, "UTF-8")); url.append("="); url.append(encoder.encode(value, "UTF-8")); } } GetMethod get = new GetMethod(url.toString()); try { httpClient.executeMethod(get); if (get.getStatusCode() == HttpStatus.SC_MOVED_PERMANENTLY || get.getStatusCode() == HttpStatus.SC_MOVED_TEMPORARILY) { Header locationHeader = get.getResponseHeader("location"); if (locationHeader != null) { String redirectLocation = locationHeader.getValue(); get.setURI(new URI(redirectLocation, true)); httpClient.executeMethod(get); } } if (get.getStatusCode() != HttpServletResponse.SC_OK) { throw new LuceneQueryParserException( "Request failed " + get.getStatusCode() + " " + url.toString()); } Reader reader = new BufferedReader(new InputStreamReader(get.getResponseBodyAsStream())); // TODO - replace with streaming-based solution e.g. SimpleJSON ContentHandler JSONObject json = new JSONObject(new JSONTokener(reader)); return json; } finally { get.releaseConnection(); } } catch (UnsupportedEncodingException e) { throw new LuceneQueryParserException("", e); } catch (HttpException e) { throw new LuceneQueryParserException("", e); } catch (IOException e) { throw new LuceneQueryParserException("", e); } catch (JSONException e) { throw new LuceneQueryParserException("", e); } } private static class Counter { int value = 0; Counter() { } void increment() { value++; } int value() { return value; } } }