gernotbelger@9031: package test;
gernotbelger@9031: 
gernotbelger@9031: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9031:  * Software engineering by
gernotbelger@9031:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@9031:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@9031:  *
gernotbelger@9031:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@9031:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@9031:  * documentation coming with Dive4Elements River for details.
gernotbelger@9031:  */
gernotbelger@9031: 
gernotbelger@9031: import java.io.File;
gernotbelger@9031: import java.io.FileNotFoundException;
gernotbelger@9031: import java.io.FileOutputStream;
gernotbelger@9031: import java.io.IOException;
gernotbelger@9031: import java.io.InputStream;
gernotbelger@9031: import java.io.InputStreamReader;
gernotbelger@9031: import java.net.URL;
gernotbelger@9031: import java.util.ArrayList;
gernotbelger@9031: import java.util.List;
gernotbelger@9031: 
gernotbelger@9031: import org.apache.commons.io.FileUtils;
gernotbelger@9031: import org.apache.commons.io.IOUtils;
gernotbelger@9031: import org.dive4elements.artifacts.common.ArtifactNamespaceContext;
gernotbelger@9031: import org.dive4elements.artifacts.common.utils.ClientProtocolUtils;
gernotbelger@9031: import org.dive4elements.artifacts.httpclient.exceptions.ConnectionException;
gernotbelger@9031: import org.dive4elements.artifacts.httpclient.http.HttpClient;
gernotbelger@9031: import org.dive4elements.artifacts.httpclient.http.HttpClientImpl;
gernotbelger@9031: import org.dive4elements.artifacts.httpclient.http.response.DocumentResponseHandler;
gernotbelger@9031: import org.dive4elements.artifacts.httpclient.utils.XMLUtils;
gernotbelger@9031: import org.dive4elements.river.client.server.AdvanceServiceImpl;
gernotbelger@9031: import org.dive4elements.river.client.server.ArtifactHelper;
gernotbelger@9031: import org.dive4elements.river.client.server.CollectionHelper;
gernotbelger@9031: import org.dive4elements.river.client.server.CreateCollectionServiceImpl;
gernotbelger@9031: import org.dive4elements.river.client.server.FLYSArtifactCreator;
gernotbelger@9031: import org.dive4elements.river.client.server.FeedServiceImpl;
gernotbelger@9031: import org.dive4elements.river.client.server.auth.DefaultUser;
gernotbelger@9031: import org.dive4elements.river.client.server.auth.User;
gernotbelger@9031: import org.dive4elements.river.client.server.auth.UserClient;
gernotbelger@9031: import org.dive4elements.river.client.shared.exceptions.ServerException;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.Artifact;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.Collection;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.Data;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.DataItem;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.DefaultCollection;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.DefaultDataItem;
gernotbelger@9127: import org.dive4elements.river.client.shared.model.OutputMode;
gernotbelger@9031: import org.dive4elements.river.client.shared.model.StringOptionsData;
gernotbelger@9031: import org.junit.Assert;
gernotbelger@9031: import org.w3c.dom.Document;
gernotbelger@9031: import org.w3c.dom.Element;
gernotbelger@9031: 
gernotbelger@9127: import test.BerechnungsartenTester.CalcMode;
gernotbelger@9127: import test.BerechnungsartenTester.FilenameMapper;
gernotbelger@9127: import test.BerechnungsartenTester.River;
gernotbelger@9127: 
gernotbelger@9031: /**
gernotbelger@9031:  * @author Domenico Nardi Tironi
gernotbelger@9031:  *
gernotbelger@9031:  */
gernotbelger@9148: public abstract class AbstractModuleRunner {
gernotbelger@9127: 
gernotbelger@9127:     public enum Infotype {
gernotbelger@9127:         sinfo
gernotbelger@9127:     }
gernotbelger@9031: 
gernotbelger@9210:     private static final String lineSeparator = System.getProperty("line.separator");
gernotbelger@9210: 
gernotbelger@9031:     private final String serverUrl = "http://localhost:8181";
gernotbelger@9031:     private final String locale = "de";
gernotbelger@9031:     private final HttpClient client;
gernotbelger@9031: 
gernotbelger@9031:     private static final String exportFileDir = "D:" + File.separator;
gernotbelger@9127:     private static final String IGNORE_ERSTELLDATUM = "# Datum der Erstellung";
gernotbelger@9127:     private static final String IGNORE_FLYS_VERSION = "# FLYS-Version:";
gernotbelger@9127:     private static final String IGNORE_BEARBEITER = "# Bearbeiter:";
gernotbelger@9031: 
gernotbelger@9031:     private final String username;
gernotbelger@9031:     private final String password;
gernotbelger@9127:     private final Infotype infotype;
gernotbelger@9031:     private final String userUuid;
gernotbelger@9031:     private Collection collection;
gernotbelger@9031:     private Artifact artifact;
gernotbelger@9127:     private final FilenameMapper fileName;
gernotbelger@9127: 
gernotbelger@9127:     // common attributes
gernotbelger@9127:     private final CalcMode calcMode;
gernotbelger@9127:     private final double from;
gernotbelger@9127:     private final double to;
gernotbelger@9127:     private final River river;
gernotbelger@9031: 
gernotbelger@9148:     public AbstractModuleRunner(final String username, final String password, final Infotype infotype, final CalcMode sinfoCalcFlowDepth,
gernotbelger@9127:             final FilenameMapper helloWorldFile, final double from, final double to, final River beispielfluss) throws ConnectionException, ServerException {
gernotbelger@9127:         // common attributes (evtl. doch in subklassen, evtl. Zwischenhierarchiestufe einführen
gernotbelger@9127: 
gernotbelger@9127:         this.calcMode = sinfoCalcFlowDepth;
gernotbelger@9127:         this.river = beispielfluss;
gernotbelger@9127:         this.from = from;
gernotbelger@9127:         this.to = to;
gernotbelger@9127: 
gernotbelger@9031:         this.username = username;
gernotbelger@9031:         this.password = password;
gernotbelger@9031:         this.infotype = infotype;
gernotbelger@9127:         this.fileName = helloWorldFile;
gernotbelger@9031: 
gernotbelger@9031:         // init
gernotbelger@9031:         this.client = new HttpClientImpl(this.serverUrl, this.locale);
gernotbelger@9031:         this.userUuid = makeUserUuid();
gernotbelger@9031:         this.collection = getCollection();
gernotbelger@9031:         this.artifact = getArtifact();
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9031:     private String makeUserUuid() throws ConnectionException {
gernotbelger@9031:         final User user = new DefaultUser(this.username, this.password, null, false, new ArrayList<String>(), new ArrayList<String>());
gernotbelger@9031:         final UserClient userClient = new UserClient(this.serverUrl);
gernotbelger@9031:         Element userElement;
gernotbelger@9031: 
gernotbelger@9031:         userElement = userClient.findUser(user);
gernotbelger@9031:         return userElement.getAttributeNS(ArtifactNamespaceContext.NAMESPACE_URI, "uuid");
gernotbelger@9031: 
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9031:     protected final Artifact getArtifact() throws ServerException, ConnectionException {
gernotbelger@9031:         if (this.artifact == null) {
gernotbelger@9127:             this.artifact = ArtifactHelper.createArtifact(this.serverUrl, this.locale, this.infotype.name(), null);
gernotbelger@9031:             setCollection(CollectionHelper.addArtifact(getCollection(), this.artifact, this.serverUrl, this.locale)); // wichtig; sorgt für Persistenz
gernotbelger@9031:         }
gernotbelger@9031:         return this.artifact;
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9148:     protected Collection getCollection() throws ConnectionException {
gernotbelger@9031: 
gernotbelger@9031:         if (this.collection == null) {
gernotbelger@9031:             // lazy-Loading
gernotbelger@9031:             final Document create = ClientProtocolUtils.newCreateCollectionDocument(null);
gernotbelger@9031:             final Document doc = (Document) this.client.createCollection(create, this.userUuid, new DocumentResponseHandler());
gernotbelger@9031:             final String uuid = XMLUtils.xpathString(doc, CreateCollectionServiceImpl.XPATH_COLLECTION_UUID, ArtifactNamespaceContext.INSTANCE);
gernotbelger@9031:             final String ttlStr = XMLUtils.xpathString(doc, CreateCollectionServiceImpl.XPATH_COLLECTION_TTL, ArtifactNamespaceContext.INSTANCE);
gernotbelger@9031:             this.collection = new DefaultCollection(uuid, Long.valueOf(ttlStr), uuid);
gernotbelger@9031:         }
gernotbelger@9031:         return this.collection;
gernotbelger@9031: 
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     private final void setCollection(final Collection collection) {
gernotbelger@9031:         this.collection = collection;
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     private final void setArtifact(final Artifact artifact) {
gernotbelger@9031:         this.artifact = artifact;
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9031:     public abstract void runTest(final boolean exportToFile) throws ConnectionException, ServerException, IOException;
gernotbelger@9031: 
gernotbelger@9031:     protected final void describeCollection() throws ConnectionException {
gernotbelger@9031: 
gernotbelger@9031:         final String uuid = getCollection().identifier();
gernotbelger@9031:         final Document describe = ClientProtocolUtils.newDescribeCollectionDocument(uuid);
gernotbelger@9031:         final Document response = (Document) this.client.doCollectionAction(describe, uuid, new DocumentResponseHandler());
gernotbelger@9031:         final Collection c = CollectionHelper.parseCollection(response);
gernotbelger@9031:         setCollection(c);
gernotbelger@9031: 
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9031:     protected final void feedAndGo(final Data[] data, final int reachableStateIndex) throws ConnectionException, ServerException {
gernotbelger@9031:         feed(data);
gernotbelger@9127:         advance(getReachableStateByIndex(getArtifact(), reachableStateIndex)); // reachablestate könnte auch String sein.
gernotbelger@9127: 
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     private final String getReachableStateByIndex(final Artifact artifact, final int index) {
gernotbelger@9031: 
gernotbelger@9031:         final String[] states = artifact.getArtifactDescription().getReachableStates();
gernotbelger@9031:         if (states != null) {
gernotbelger@9210: 
gernotbelger@9210:             if (states.length > index)
gernotbelger@9031:                 return states[index];
gernotbelger@9210: 
gernotbelger@9210:             return states[0];
gernotbelger@9031:         }
gernotbelger@9210: 
gernotbelger@9210:         return "";
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     private final void feed(final Data[] data) throws ServerException, ConnectionException {
gernotbelger@9031:         final Document feed = ClientProtocolUtils.newFeedDocument(getArtifact().getUuid(), getArtifact().getHash(), createKVP(data));
gernotbelger@9031:         final Document description = (Document) this.client.feed(
gernotbelger@9031:                 new org.dive4elements.artifacts.httpclient.objects.Artifact(getArtifact().getUuid(), getArtifact().getHash()), feed,
gernotbelger@9031:                 new DocumentResponseHandler());
gernotbelger@9031: 
gernotbelger@9031:         final String result = XMLUtils.xpathString(description, FeedServiceImpl.XPATH_RESULT, ArtifactNamespaceContext.INSTANCE);
gernotbelger@9031: 
gernotbelger@9031:         if (result == null || !result.equals(FeedServiceImpl.OPERATION_FAILURE)) {
gernotbelger@9031:             setArtifact((Artifact) new FLYSArtifactCreator().create(description));
gernotbelger@9031:         } else if (result != null && result.equals(FeedServiceImpl.OPERATION_FAILURE)) {
gernotbelger@9031:             final String msg = XMLUtils.xpathString(description, FeedServiceImpl.XPATH_RESULT_MSG, ArtifactNamespaceContext.INSTANCE);
gernotbelger@9031:             throw new ServerException(msg);
gernotbelger@9031:         }
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9148:     protected final Data[] extractPairData(final List<String> pairIds, final String dataName) {
gernotbelger@9148:         final Data[] data = new Data[pairIds.size()];
gernotbelger@9148:         int i = 0;
gernotbelger@9148:         for (final String pairId : pairIds) {
gernotbelger@9148:             final Data pair = new StringOptionsData(dataName, dataName, new DataItem[] { new DefaultDataItem(pairId, pairId, pairId) });
gernotbelger@9148:             data[i] = pair;
gernotbelger@9148:             i++;
gernotbelger@9148:         }
gernotbelger@9148:         return data;
gernotbelger@9148:     }
gernotbelger@9148: 
gernotbelger@9127:     private final String[][] createKVP(final Data[] data) {
gernotbelger@9031:         if (data != null) {
gernotbelger@9031:             final String[][] kvp = new String[data.length][];
gernotbelger@9031: 
gernotbelger@9031:             int i = 0;
gernotbelger@9031: 
gernotbelger@9031:             for (final Data d : data) {
gernotbelger@9031:                 final String key = d.getLabel();
gernotbelger@9031:                 final String value = d.getStringValue();
gernotbelger@9031: 
gernotbelger@9031:                 kvp[i++] = new String[] { key, value };
gernotbelger@9031:             }
gernotbelger@9031: 
gernotbelger@9031:             return kvp;
gernotbelger@9031:         }
gernotbelger@9031:         return null;
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     private final void advance(final String target) throws ConnectionException, ServerException {
gernotbelger@9031:         final Document advance = ClientProtocolUtils.newAdvanceDocument(getArtifact().getUuid(), getArtifact().getHash(), target);
gernotbelger@9031:         final Document description = (Document) this.client.advance(
gernotbelger@9031:                 new org.dive4elements.artifacts.httpclient.objects.Artifact(getArtifact().getUuid(), getArtifact().getHash()), advance,
gernotbelger@9031:                 new DocumentResponseHandler());
gernotbelger@9031: 
gernotbelger@9031:         if (description == null) {
gernotbelger@9031:             throw new ServerException(AdvanceServiceImpl.ERROR_ADVANCE_ARTIFACT);
gernotbelger@9031:         }
gernotbelger@9031: 
gernotbelger@9031:         final String result = XMLUtils.xpathString(description, AdvanceServiceImpl.XPATH_RESULT, ArtifactNamespaceContext.INSTANCE);
gernotbelger@9031: 
gernotbelger@9031:         if (result == null || !result.equals(AdvanceServiceImpl.OPERATION_FAILURE)) {
gernotbelger@9031:             setArtifact((Artifact) new FLYSArtifactCreator().create(description));
gernotbelger@9031:         }
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9031:     /// ExportServiceImpl
gernotbelger@9148:     protected final void assertAndWriteToFile(final String mode, final boolean exportToFile) throws IOException {
gernotbelger@9031: 
gernotbelger@9031:         final String type = "csv";
gernotbelger@9031: 
gernotbelger@9031:         final String enc = "windows-1252";// req.getParameter("encoding");
gernotbelger@9031: 
gernotbelger@9127:         final URL expectedResource = getClass().getResource(this.fileName.getFilename());
gernotbelger@9031:         final Document attr = null;
gernotbelger@9031:         final Document request = ClientProtocolUtils.newOutCollectionDocument(getCollection().identifier(), mode, type, attr);
gernotbelger@9031: 
gernotbelger@9031:         final InputStream response = this.client.collectionOut(request, getCollection().identifier(), mode);
gernotbelger@9031: 
gernotbelger@9031:         final String actual = deleteErstelldatum(IOUtils.toString(response, "UTF-8"));
gernotbelger@9031: 
gernotbelger@9031:         final String expected = deleteErstelldatum(FileUtils.readFileToString(new File(expectedResource.getFile()), enc));
gernotbelger@9031: 
gernotbelger@9031:         // if (!actual.equals(expected)) {
gernotbelger@9031:         if (exportToFile) {
gernotbelger@9031:             doGetWriteToDisk(mode); // TODO: WENN der Test negativ ausfällt, Datei abspeichern -> Diskussion
gernotbelger@9031:         }
gernotbelger@9031: 
gernotbelger@9133:         Assert.assertEquals(expected, actual);
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     private final String deleteErstelldatum(final String input) {
gernotbelger@9031:         String result = "";
gernotbelger@9210:         final String[] lines = input.split(lineSeparator);
gernotbelger@9031:         for (final String line : lines) {
gernotbelger@9148:             if (!line.contains(AbstractModuleRunner.IGNORE_ERSTELLDATUM) && !line.contains(AbstractModuleRunner.IGNORE_FLYS_VERSION)
gernotbelger@9148:                     && !line.contains(AbstractModuleRunner.IGNORE_BEARBEITER)) {
gernotbelger@9210:                 result = result + line + lineSeparator;
gernotbelger@9031:             }
gernotbelger@9031:         }
gernotbelger@9031:         return result;
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     public final void doGetWriteToDisk(final String mode) throws FileNotFoundException, IOException {
gernotbelger@9031: 
gernotbelger@9031:         final String name = mode;
gernotbelger@9031:         final String type = "csv";
gernotbelger@9031: 
gernotbelger@9127:         final String fn = name + System.currentTimeMillis() + "." + type;
gernotbelger@9031:         final String enc = "windows-1252";
gernotbelger@9031: 
gernotbelger@9031:         final String filepath = exportFileDir + fn;
gernotbelger@9031: 
gernotbelger@9031:         final Document attr = null;
gernotbelger@9031:         final Document request = ClientProtocolUtils.newOutCollectionDocument(getCollection().identifier(), mode, type, attr);
gernotbelger@9031: 
gernotbelger@9031:         final InputStream response = this.client.collectionOut(request, getCollection().identifier(), mode);
gernotbelger@9031:         final InputStreamReader in = new InputStreamReader(response, "UTF-8");
gernotbelger@9031: 
gernotbelger@9031:         IOUtils.copy(in, new FileOutputStream(filepath), enc);
gernotbelger@9031: 
gernotbelger@9031:     }
gernotbelger@9031: 
gernotbelger@9127:     protected final void selectCalcMode() throws ConnectionException, ServerException {
gernotbelger@9127: 
gernotbelger@9127:         /* Select CalcMode */
gernotbelger@9127:         final String calcmodeStr = this.calcMode.name();
gernotbelger@9127:         final Data dataCalcMode = new StringOptionsData("calculation_mode", "calculation_mode",
gernotbelger@9127:                 new DataItem[] { new DefaultDataItem(calcmodeStr, calcmodeStr, calcmodeStr) });
gernotbelger@9127:         feedAndGo(new Data[] { dataCalcMode }, 0);
gernotbelger@9127:     }
gernotbelger@9127: 
gernotbelger@9127:     protected final void selectRange() throws ConnectionException, ServerException {
gernotbelger@9127:         final String fromStr = String.valueOf(this.from);
gernotbelger@9127:         final String toStr = String.valueOf(this.to);
gernotbelger@9127:         final Data dataFrom = new StringOptionsData("ld_from", "ld_from", new DataItem[] { new DefaultDataItem(fromStr, fromStr, fromStr) });
gernotbelger@9127:         final Data dataTo = new StringOptionsData("ld_to", "ld_to", new DataItem[] { new DefaultDataItem(toStr, toStr, toStr) });
gernotbelger@9127:         final Data[] rangeFromToDetermined = new Data[] { dataFrom, dataTo };
gernotbelger@9127: 
gernotbelger@9127:         feedAndGo(rangeFromToDetermined, 0);
gernotbelger@9127:     }
gernotbelger@9127: 
gernotbelger@9127:     protected final void selectRiver() throws ConnectionException, ServerException {
gernotbelger@9127:         final String riverStr = this.river.name();
gernotbelger@9127:         final Data data = new StringOptionsData("river", "river", new DataItem[] { new DefaultDataItem(riverStr, riverStr, riverStr) });
gernotbelger@9127:         feedAndGo(new Data[] { data }, 0);
gernotbelger@9127:     }
gernotbelger@9127: 
gernotbelger@9127:     protected final void export(final boolean exportToFile) throws IOException, ServerException {
gernotbelger@9148:         final OutputMode[] modes = getArtifact().getArtifactDescription().getOutputModes();
gernotbelger@9127:         if (modes != null) {
gernotbelger@9148:             for (final OutputMode mode : modes) {
gernotbelger@9148:                 if (mode.getDescription().contains("_export"))
gernotbelger@9148:                     assertAndWriteToFile(mode.getName(), exportToFile);
gernotbelger@9148:             }
gernotbelger@9127:         }
gernotbelger@9127:     }
gernotbelger@9127: 
gernotbelger@9148:     protected final String getRecommendationPairString(final SimpleRecommendation rec1, final SimpleRecommendation rec2)
gernotbelger@9148:             throws ConnectionException, ServerException {
gernotbelger@9148: 
gernotbelger@9148:         return rec1.getRecommendationPairString(rec2, getCollection(), this.serverUrl, this.locale);
gernotbelger@9148:     }
gernotbelger@9210: }