com.fferreira.example.hazelcast.mapstore.cassandra.HCCassandraWorkerTest.java Source code

Introduction

Here is the source code for com.fferreira.example.hazelcast.mapstore.cassandra.HCCassandraWorkerTest.java

Source

/**
 * 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.fferreira.example.hazelcast.mapstore.cassandra;

import com.datastax.driver.core.Session;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fferreira.example.hazelcast.Constants;
import com.fferreira.example.hazelcast.HazelcastWorker;
import com.fferreira.example.hazelcast.User;
import com.fferreira.example.hazelcast.mapstore.EntryEntity;
import com.fferreira.example.hazelcast.mapstore.HazelcastMapStore;
import java.util.Collection;
import java.util.UUID;
import static org.testng.AssertJUnit.assertEquals;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

@Test(suiteName = "Hazelcast Cassandra Worker Test", priority = 2)
public class HCCassandraWorkerTest {

    private static final String CREATE_GROUP = "create_group";
    private static final String RUD_GROUP = "rud_group";
    private static final String COLD_START_GROUP = "cold_start_group";

    private CassandraClient dao;

    private HazelcastWorker worker;
    private HazelcastMapStore mapStore;

    // data to be shared in test
    private String id;
    private User user;
    private int counter = 0;

    @BeforeClass
    public void setUpClass() throws Exception {

        // creating cassandra client with IP added to POM.
        dao = new CassandraClient();
        dao.initialize("localhost");

        // creating data keyspace and table
        final Session session = dao.connect();
        session.execute("CREATE KEYSPACE IF NOT EXISTS " + Constants.CASSANDRA_KEYSPACE + " WITH replication "
                + "= {'class':'SimpleStrategy', 'replication_factor':3};");
        session.execute("CREATE TABLE IF NOT EXISTS " + Constants.CASSANDRA_KEYSPACE_TABLE_NAME + " ("
                + "id text PRIMARY KEY," + "data text" + ");");

        mapStore = new HazelcastMapStore(User.class);
        mapStore.setDao(dao);

        worker = new HazelcastWorker(Constants.CASSANDRA_MAP_STORE);
    }

    @AfterClass
    public void tearDownClass() throws Exception {
        // droping data keyspace and table
        final Session session = dao.connect();
        session//
                .execute("DROP TABLE IF EXISTS " + Constants.CASSANDRA_KEYSPACE_TABLE_NAME + ";");
        session.execute("DROP KEYSPACE IF EXISTS " + Constants.CASSANDRA_KEYSPACE + ";");
        // stoping hazelcast instance and client
        worker.destroy();
    }

    @Test(groups = CREATE_GROUP, invocationCount = 3)
    public void test_add_user() throws InterruptedException {
        counter++;
        id = "user_cassamdra_" + counter + "_" + UUID.randomUUID();
        user = new User("Flvio" + counter, "Ferreira" + counter, "Portugal");
        worker.addUser(id, user);
        // just give time to it since it is async
        Thread.sleep(3000);
        assertEquals(dao.findAll().size(), counter);
    }

    @Test(dependsOnGroups = CREATE_GROUP, groups = RUD_GROUP)
    public void test_remove_subscriber() throws Exception {
        worker.removeUser(id);
        // just give time to it since it is async
        Thread.sleep(2000);
        counter--;

        assertEquals(dao.findAll().size(), counter);

        // getting valid values to id and user
        final EntryEntity res = dao.findAll().get(0);

        id = res.getId();
        user = new ObjectMapper().readValue(res.getMessage(), User.class);
    }

    @Test(dependsOnGroups = CREATE_GROUP, groups = RUD_GROUP)
    public void test_get_user() {
        assertEquals(worker.getUser(id), user);
    }

    @Test(dependsOnGroups = CREATE_GROUP, groups = RUD_GROUP)
    public void test_get_user_with_given_message() {
        final Collection<User> users = worker.getUsersByFirstName(user.getFirtName());
        assertEquals(users.size(), 1);
        assertEquals(users.toArray(new User[users.size()])[0].getFirtName(), user.getFirtName());
    }

}