org.clebi.subscribers.daos.SubscribersIntegTestHelper.java Source code

Introduction

Here is the source code for org.clebi.subscribers.daos.SubscribersIntegTestHelper.java

Source

// Copyright 2016 Clment Bizeau
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package org.clebi.subscribers.daos;

import org.apache.commons.io.FileUtils;
import org.apache.commons.lang.builder.EqualsBuilder;
import org.clebi.subscribers.daos.elasticsearch.SubscriberDaoImpl;
import org.clebi.subscribers.model.Email;
import org.clebi.subscribers.model.Subscriber;
import org.clebi.subscribers.model.serialize.JsonFactory;
import org.elasticsearch.action.admin.indices.delete.DeleteIndexRequest;
import org.elasticsearch.action.admin.indices.refresh.RefreshRequest;
import org.elasticsearch.client.Client;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.node.Node;
import org.elasticsearch.node.NodeBuilder;
import org.junit.Assert;
import org.junit.Before;
import org.junit.BeforeClass;

import java.io.File;
import java.io.IOException;
import java.time.ZoneOffset;
import java.time.ZonedDateTime;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

public class SubscribersIntegTestHelper {

    private static final String ES_DATA_PATH = "build/esdata";
    private static final String EXCLUDE_FROM_EQUALS = "fields";

    private static Settings settings = Settings.settingsBuilder().put("path.home", ES_DATA_PATH).build();
    private static Node node;

    @BeforeClass
    public static void setUp() throws IOException {
        FileUtils.deleteDirectory(new File(ES_DATA_PATH));
        node = NodeBuilder.nodeBuilder().settings(settings).local(true).node();
    }

    protected static Client getClient() {
        return node.client();
    }

    protected static void refreshIndices(String index) {
        node.client().admin().indices().refresh(new RefreshRequest(index)).actionGet();
    }

    /**
     * Initialize elasticsearch index for tests.
     */
    @Before
    public void before() {
        if (node.client().admin().indices().prepareExists().setIndices(SubscriberDaoImpl.INDEX_NAME).get()
                .isExists()) {
            DeleteIndexRequest indexRequest = new DeleteIndexRequest(SubscriberDaoImpl.INDEX_NAME);
            node.client().admin().indices().delete(indexRequest).actionGet();
        }
    }

    protected Map<String, Subscriber> indexTestSubscibers(String index) {
        final Map<String, Subscriber> subscribers = new HashMap<>();
        final String emailOptinActive = "optin_active_0@test.com";
        final String emailOptinNonActive = "optin_non-active_0@test.com";
        final String emailNonOptinActive = "non-optin-active_0@test.com";
        final String emailNonOptinNonActiv = "non-optin-non-active_0@test.com";
        subscribers.put(emailOptinActive, new Subscriber(true, true, new Email(emailOptinActive),
                ZonedDateTime.now(ZoneOffset.UTC), generateTestFields()));
        subscribers.put(emailOptinNonActive, new Subscriber(true, false, new Email(emailOptinNonActive),
                ZonedDateTime.now(ZoneOffset.UTC), generateTestFields()));
        subscribers.put(emailNonOptinActive, new Subscriber(false, true, new Email(emailNonOptinActive),
                ZonedDateTime.now(ZoneOffset.UTC), generateTestFields()));
        subscribers.put(emailNonOptinNonActiv, new Subscriber(false, false, new Email(emailNonOptinNonActiv),
                ZonedDateTime.now(ZoneOffset.UTC), generateTestFields()));
        for (Map.Entry<String, Subscriber> entry : subscribers.entrySet()) {
            indexSubsciber(index, entry.getValue());
        }
        return subscribers;
    }

    protected void indexSubsciber(String index, Subscriber subscriber) {
        node.client().prepareIndex(index, SubscriberDaoImpl.DOCUMENT_NAME).setId(subscriber.getEmail().toString())
                .setSource(JsonFactory.getGson().toJson(subscriber)).get();
    }

    protected Map<String, Object> generateTestFields() {
        Map<String, Object> fields = new HashMap<>();
        fields.put("field1", "test");
        fields.put("field2", 12.0);
        return fields;
    }

    protected void assertSubscriber(Subscriber expected, Subscriber actual) {
        Collection<String> excludes = new ArrayList<String>();
        excludes.add(EXCLUDE_FROM_EQUALS);
        Assert.assertTrue(EqualsBuilder.reflectionEquals(expected, actual, excludes));
        Assert.assertEquals(expected.getFields().size(), actual.getFields().size());
        for (Map.Entry<String, Object> entry : expected.getFields().entrySet()) {
            Assert.assertEquals(entry.getValue(), actual.getFields().get(entry.getKey()));
        }
    }

}