org.arquillian.ape.nosql.couchdb.CouchDbDeclarativeSupport Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of arquillian-ape-nosql-couchdb Show documentation

The newest version!

package org.arquillian.ape.nosql.couchdb;

import java.lang.annotation.Annotation;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import org.arquillian.ape.api.MetadataExtractor;
import org.arquillian.ape.nosql.NoSqlDeclarativeSupport;
import org.arquillian.ape.nosql.NoSqlPopulator;
import org.jboss.arquillian.core.api.annotation.Observes;
import org.jboss.arquillian.test.spi.TestClass;
import org.jboss.arquillian.test.spi.event.suite.After;
import org.jboss.arquillian.test.spi.event.suite.Before;
import org.jboss.arquillian.test.spi.event.suite.BeforeClass;

public class CouchDbDeclarativeSupport extends NoSqlDeclarativeSupport {

    protected CouchDbDeclarativeSupport() {
        super(new NoSqlPopulator(new CouchDbPopulatorService()));
    }

    void connect(@Observes BeforeClass beforeClass) {
        this.configure(beforeClass.getTestClass());
    }

    void populate(@Observes Before before) {
        populate(before.getTestClass(), before.getTestMethod());
    }

    // Clean should be executed before insert
    void cleanBefore(@Observes(precedence = 10) Before before) {
        this.clean(before.getTestClass(), before.getTestMethod(), false);
    }

    void cleanAfter(@Observes After after) {
        this.clean(after.getTestClass(), after.getTestMethod(), true);
    }

    @Override
    protected Map getConfiguration(TestClass testClass) {
        final MetadataExtractor metadataExtractor = new MetadataExtractor(testClass);
        final Optional redisConfigurationOptional =
            Optional.ofNullable(metadataExtractor.using(CouchDbConfiguration.class).getAnnotationOnClassLevel());

        final Map options = new HashMap<>();
        redisConfigurationOptional.ifPresent(redisConfiguration -> options.putAll(CouchDbOptions.from(redisConfiguration)));

        return options;
    }

    @Override
    protected Class getPopulatorAnnotation() {
        return CouchDb.class;
    }
}