test.java.com.ibm.cloudant.kafka.connect.CloudantSourceConnectorTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kafka-connect-cloudant Show documentation
Apache Kafka Connect API connector for Cloudant
The newest version!
/*
 * Copyright © 2016, 2018 IBM Corp. All rights reserved.
 *
 * 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 com.ibm.cloudant.kafka.connect;

import com.ibm.cloudant.kafka.common.InterfaceConst;
import com.ibm.cloudant.kafka.connect.utils.ConnectorUtils;

import junit.framework.TestCase;

import org.apache.kafka.connect.connector.ConnectorContext;
import org.junit.Assert;
import org.junit.Rule;
import org.junit.rules.ExpectedException;
import org.powermock.api.easymock.PowerMock;

import java.util.List;
import java.util.Map;

/**
 * @author holger
 */
public class CloudantSourceConnectorTest extends TestCase {

    @Rule
    public ExpectedException thrown = ExpectedException.none();

    private CloudantSourceConnector connector;
    private Map sourceProperties;


    /* (non-Javadoc)
     * @see junit.framework.TestCase#setUp()
     */
    protected void setUp() throws Exception {
        super.setUp();

        sourceProperties = ConnectorUtils.getSourceProperties();

        connector = new CloudantSourceConnector();
        ConnectorContext context = PowerMock.createMock(ConnectorContext.class);
        connector.initialize(context);
    }

    /**
     * Test method for {@link com.ibm.cloudant.kafka.connect.CloudantSourceConnector#stop()}.
     */
    public void testStop() {
        // fail("Not yet implemented");
    }


    /**
     * Test method for
     * {@link com.ibm.cloudant.kafka.connect.CloudantSourceConnector#start(java.util.Map)}.
     */
    public void testStartMapOfStringString() {
        PowerMock.replayAll();
        connector.start(sourceProperties);
        PowerMock.verifyAll();
    }

    /**
     * Test method for
     * {@link com.ibm.cloudant.kafka.connect.CloudantSourceConnector#taskConfigs(int)}.
     */
    public void testTaskConfigs() {
        PowerMock.replayAll();
        connector.start(sourceProperties);

        int tasks_max = Integer.parseInt(sourceProperties.get(InterfaceConst.TASKS_MAX));
        List> taskConfigs = connector.taskConfigs(tasks_max);

        Assert.assertEquals(1, taskConfigs.size());

        Assert.assertEquals(sourceProperties.get(InterfaceConst.URL), taskConfigs.get(0).get
                (InterfaceConst.URL));
        Assert.assertEquals(sourceProperties.get(InterfaceConst.USER_NAME), taskConfigs.get(0)
                .get(InterfaceConst.USER_NAME));
        Assert.assertEquals(sourceProperties.get(InterfaceConst.PASSWORD), taskConfigs.get(0).get
                (InterfaceConst.PASSWORD));
        Assert.assertEquals(sourceProperties.get(InterfaceConst.LAST_CHANGE_SEQ), taskConfigs.get
                (0).get(InterfaceConst.LAST_CHANGE_SEQ));

        Assert.assertEquals(sourceProperties.get(InterfaceConst.TOPIC), taskConfigs.get(0).get
                (InterfaceConst.TOPIC));

        Assert.assertEquals(sourceProperties.get(InterfaceConst.TASKS_MAX), taskConfigs.get(0)
                .get(InterfaceConst.TASKS_MAX));
        Assert.assertEquals(sourceProperties.get(InterfaceConst.BATCH_SIZE), taskConfigs.get(0)
                .get(InterfaceConst.BATCH_SIZE));

        Assert.assertEquals(sourceProperties.get(InterfaceConst.OMIT_DESIGN_DOCS),
                taskConfigs.get(0).get(InterfaceConst.OMIT_DESIGN_DOCS));

        Assert.assertEquals(sourceProperties.get(InterfaceConst.USE_VALUE_SCHEMA_STRUCT),
                taskConfigs.get(0).get(InterfaceConst.USE_VALUE_SCHEMA_STRUCT));
        Assert.assertEquals(sourceProperties.get(InterfaceConst.FLATTEN_VALUE_SCHEMA_STRUCT),
                taskConfigs.get(0).get(InterfaceConst.FLATTEN_VALUE_SCHEMA_STRUCT));

        PowerMock.verifyAll();
    }
}