org.opengis.cite.iso19142.transaction.TransactionFixture Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ets-wfs20 Show documentation
Checks WFS 2.0 implementations for conformance to ISO 19142:2010.
There is a newer version: 2.0-r18
package org.opengis.cite.iso19142.transaction;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.xml.namespace.QName;

import org.apache.xerces.xs.XSModel;
import org.opengis.cite.iso19142.BaseFixture;
import org.opengis.cite.iso19142.FeatureTypeInfo;
import org.opengis.cite.iso19142.ProtocolBinding;
import org.opengis.cite.iso19142.SuiteAttribute;
import org.opengis.cite.iso19142.WFS2;
import org.opengis.cite.iso19142.util.DataSampler;
import org.opengis.cite.iso19142.util.ServiceMetadataUtils;
import org.opengis.cite.iso19142.util.WFSMessage;
import org.testng.ISuite;
import org.testng.ITestContext;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.DataProvider;
import org.w3c.dom.Document;

/**
 * Provides configuration methods that facilitate testing of transaction
 * capabilities.
 */
public class TransactionFixture extends BaseFixture {

    protected DataSampler dataSampler;
    /**
     * An XSModel object representing the application schema supported by the
     * SUT.
     */
    protected XSModel model;

    public TransactionFixture() {
        super();
    }

    /**
     * This is intended only to facilitate unit testing.
     * 
     * @param model
     *            A representation of an application schema.
     */
    void setModel(XSModel model) {
        this.model = model;
    }

    /**
     * Obtains a DataSampler object from the test run context (the value of the
     * {@link SuiteAttribute#SAMPLER SuiteAttribute.SAMPLER attribute}). A
     * schema model (XSModel) is also obtained from the test context by
     * accessing the {@link org.opengis.cite.iso19136.SuiteAttribute#XSMODEL
     * xsmodel} attribute.
     * 
     * @param testContext
     *            The test run context.
     */
    @BeforeClass(alwaysRun = true)
    public void initTransactionFixture(ITestContext testContext) {
        ISuite suite = testContext.getSuite();
        this.dataSampler = (DataSampler) suite.getAttribute(SuiteAttribute.SAMPLER.getName());
        this.model = (XSModel) suite.getAttribute(org.opengis.cite.iso19136.SuiteAttribute.XSMODEL.getName());
    }

    /**
     * Builds a DOM Document representing a Transaction request entity.
     */
    @BeforeMethod
    public void buildTransactionRequest() {
        this.reqEntity = WFSMessage.createRequestEntity(WFS2.TRANSACTION, this.wfsVersion);
    }

    /**
     * A DataProvider that supplies a collection of parameter tuples where each
     * tuple has two elements:
     * 
     * ProtocolBinding - a supported transaction request binding
     * QName - the name of a feature type for which data are available
     * 
     * 
     * @param testContext
     *            The ITestContext object for the test run.
     * @return {@literal Iterator} An iterator over a collection of
     *         parameter tuples.
     */
    @DataProvider(name = "binding+availFeatureType")
    public Iterator trxTestParameters(ITestContext testContext) {
        ISuite suite = testContext.getSuite();
        Document wfsMetadata = (Document) suite.getAttribute(SuiteAttribute.TEST_SUBJECT.getName());
        if (null == wfsMetadata) {
            throw new NullPointerException("Service description not found in ITestContext");
        }
        Set trxBindings = ServiceMetadataUtils.getOperationBindings(wfsMetadata, WFS2.TRANSACTION);
        DataSampler sampler = (DataSampler) suite.getAttribute(SuiteAttribute.SAMPLER.getName());
        Map featureInfo = sampler.getFeatureTypeInfo();
        List paramList = new ArrayList();
        for (ProtocolBinding binding : trxBindings) {
            for (FeatureTypeInfo typeInfo : featureInfo.values()) {
                if (typeInfo.isInstantiated()) {
                    Object[] tuple = { binding, typeInfo.getTypeName() };
                    paramList.add(tuple);
                }
            }
        }
        return paramList.iterator();
    }
}