package com.impetus.client.crud.datatypes;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.Query;
import junit.framework.Assert;
import org.apache.cassandra.thrift.CfDef;
import org.apache.cassandra.thrift.ColumnDef;
import org.apache.cassandra.thrift.IndexType;
import org.apache.cassandra.thrift.KsDef;
import org.apache.cassandra.thrift.NotFoundException;
import org.apache.thrift.TException;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import com.impetus.client.crud.datatypes.entities.StudentShortWrapper;
import com.impetus.kundera.client.cassandra.persistence.CassandraCli;
public class StudentCassandraShortWrapperTest extends CassandraBase
{
private static final String keyspace = "KunderaCassandraDataType";
@Before
public void setUp() throws Exception
{
super.setUp();
}
@After
public void tearDown() throws Exception
{
super.tearDown();
}
@Test
public void testExecuteUseSameEm()
{
testPersist(true);
testFindById(true);
testMerge(true);
testFindByQuery(true);
testNamedQueryUseSameEm(true);
testDelete(true);
}
@Test
public void testExecute()
{
testPersist(false);
testFindById(false);
testMerge(false);
testFindByQuery(false);
testNamedQuery(false);
testDelete(false);
}
public void testPersist(boolean useSameEm)
{
EntityManager em = emf.createEntityManager();
// Insert max value of Short
StudentShortWrapper studentMax = new StudentShortWrapper();
studentMax.setAge((Short) getMaxValue(short.class));
studentMax.setId((Short) getMaxValue(Short.class));
studentMax.setName((String) getMaxValue(String.class));
em.persist(studentMax);
// Insert min value of Short
StudentShortWrapper studentMin = new StudentShortWrapper();
studentMin.setAge((Short) getMinValue(short.class));
studentMin.setId((Short) getMinValue(Short.class));
studentMin.setName((String) getMinValue(String.class));
em.persist(studentMin);
// Insert random value of Short
StudentShortWrapper student = new StudentShortWrapper();
student.setAge((Short) getRandomValue(short.class));
student.setId((Short) getRandomValue(Short.class));
student.setName((String) getRandomValue(String.class));
em.persist(student);
em.close();
}
public void testFindById(boolean useSameEm)
{
EntityManager em = emf.createEntityManager();
StudentShortWrapper studentMax = em.find(StudentShortWrapper.class, getMaxValue(Short.class));
Assert.assertNotNull(studentMax);
Assert.assertEquals(getMaxValue(short.class), studentMax.getAge());
Assert.assertEquals(getMaxValue(String.class), studentMax.getName());
if (!useSameEm)
{
em.close();
em = emf.createEntityManager();
}
StudentShortWrapper studentMin = em.find(StudentShortWrapper.class, getMinValue(Short.class));
Assert.assertNotNull(studentMin);
Assert.assertEquals(getMinValue(short.class), studentMin.getAge());
Assert.assertEquals(getMinValue(String.class), studentMin.getName());
if (!useSameEm)
{
em.close();
em = emf.createEntityManager();
}
StudentShortWrapper student = em.find(StudentShortWrapper.class, getRandomValue(Short.class));
Assert.assertNotNull(student);
Assert.assertEquals(getRandomValue(short.class), student.getAge());
Assert.assertEquals(getRandomValue(String.class), student.getName());
em.close();
}
public void testMerge(boolean useSameEm)
{
EntityManager em = emf.createEntityManager();
StudentShortWrapper student = em.find(StudentShortWrapper.class, getMaxValue(Short.class));
Assert.assertNotNull(student);
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals(getMaxValue(String.class), student.getName());
student.setName("Kuldeep");
em.merge(student);
if (!useSameEm)
{
em.close();
em = emf.createEntityManager();
}
StudentShortWrapper newStudent = em.find(StudentShortWrapper.class, getMaxValue(Short.class));
Assert.assertNotNull(newStudent);
Assert.assertEquals(getMaxValue(short.class), newStudent.getAge());
Assert.assertEquals("Kuldeep", newStudent.getName());
}
public void testFindByQuery(boolean useSameEm)
{
findAllQuery();
findByName();
findByAge();
findByNameAndAgeGTAndLT();
findByNameAndAgeGTEQAndLTEQ();
findByNameAndAgeGTAndLTEQ();
findByNameAndAgeWithOrClause();
findByAgeAndNameGTAndLT();
findByNameAndAGEBetween();
// findByRange();
}
private void findByAgeAndNameGTAndLT()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.age = " + getMinValue(short.class)
+ " and s.name > Amresh and s.name <= " + getMaxValue(String.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(1, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
Assert.assertEquals(getMinValue(Short.class), student.getId());
Assert.assertEquals(getMinValue(short.class), student.getAge());
Assert.assertEquals(getMinValue(String.class), student.getName());
count++;
}
Assert.assertEquals(1, count);
em.close();
}
private void findByRange()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.id between " + getMinValue(Short.class) + " and "
+ getMaxValue(Short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(3, students.size());
int count = 0;
for (StudentShortWrapper student : students)
{
if (student.getId().equals(getMaxValue(Short.class)))
{
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals("Kuldeep", student.getName());
count++;
}
else if (student.getId().equals(getMinValue(Short.class)))
{
Assert.assertEquals(getMinValue(short.class), student.getAge());
Assert.assertEquals(getMinValue(String.class), student.getName());
count++;
}
else
{
Assert.assertEquals(getRandomValue(Short.class), student.getId());
Assert.assertEquals(getRandomValue(short.class), student.getAge());
Assert.assertEquals(getRandomValue(String.class), student.getName());
count++;
}
}
Assert.assertEquals(3, count);
em.close();
}
private void findByNameAndAgeWithOrClause()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.name = Kuldeep and s.age > " + getMinValue(short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(1, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
Assert.assertEquals(getMaxValue(Short.class), student.getId());
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals("Kuldeep", student.getName());
count++;
}
Assert.assertEquals(1, count);
em.close();
}
private void findByNameAndAgeGTAndLTEQ()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.name = Kuldeep and s.age > " + getMinValue(short.class)
+ " and s.age <= " + getMaxValue(short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(1, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
Assert.assertEquals(getMaxValue(Short.class), student.getId());
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals("Kuldeep", student.getName());
count++;
}
Assert.assertEquals(1, count);
em.close();
}
public void testNamedQueryUseSameEm(boolean useSameEm)
{
updateNamed(true);
deleteNamed(true);
}
public void testNamedQuery(boolean useSameEm)
{
updateNamed(false);
deleteNamed(false);
}
public void testDelete(boolean useSameEm)
{
EntityManager em = emf.createEntityManager();
StudentShortWrapper studentMax = em.find(StudentShortWrapper.class, getMaxValue(Short.class));
Assert.assertNotNull(studentMax);
Assert.assertEquals(getMaxValue(short.class), studentMax.getAge());
Assert.assertEquals("Kuldeep", studentMax.getName());
em.remove(studentMax);
if (!useSameEm)
{
em.close();
em = emf.createEntityManager();
}
studentMax = em.find(StudentShortWrapper.class, getMaxValue(Short.class));
Assert.assertNull(studentMax);
em.close();
}
/**
*
*/
private void deleteNamed(boolean useSameEm)
{
String deleteQuery = "Delete From StudentShortWrapper s where s.name=Vivek";
EntityManager em = emf.createEntityManager();
Query q = em.createQuery(deleteQuery);
q.executeUpdate();
if (!useSameEm)
{
em.close();
em = emf.createEntityManager();
}
StudentShortWrapper newStudent = em.find(StudentShortWrapper.class, getRandomValue(Short.class));
Assert.assertNull(newStudent);
em.close();
}
/**
* @return
*/
private void updateNamed(boolean useSameEm)
{
EntityManager em = emf.createEntityManager();
String updateQuery = "Update StudentShortWrapper s SET s.name=Vivek where s.name=Amresh";
Query q = em.createQuery(updateQuery);
q.executeUpdate();
if (!useSameEm)
{
em.close();
em = emf.createEntityManager();
}
StudentShortWrapper newStudent = em.find(StudentShortWrapper.class, getRandomValue(Short.class));
Assert.assertNotNull(newStudent);
Assert.assertEquals(getRandomValue(short.class), newStudent.getAge());
Assert.assertEquals("Vivek", newStudent.getName());
em.close();
}
private void findByNameAndAGEBetween()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.name = Amresh and s.age between "
+ getMinValue(short.class) + " and " + getMaxValue(short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(1, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
Assert.assertEquals(getRandomValue(Short.class), student.getId());
Assert.assertEquals(getRandomValue(short.class), student.getAge());
Assert.assertEquals(getRandomValue(String.class), student.getName());
count++;
}
Assert.assertEquals(1, count);
em.close();
}
private void findByNameAndAgeGTAndLT()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.name = Amresh and s.age > " + getMinValue(short.class)
+ " and s.age < " + getMaxValue(short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(1, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
Assert.assertEquals(getRandomValue(Short.class), student.getId());
Assert.assertEquals(getRandomValue(short.class), student.getAge());
Assert.assertEquals(getRandomValue(String.class), student.getName());
count++;
}
Assert.assertEquals(1, count);
em.close();
}
private void findByNameAndAgeGTEQAndLTEQ()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.name = Kuldeep and s.age >= " + getMinValue(short.class)
+ " and s.age <= " + getMaxValue(short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(2, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
if (student.getId().equals(getMaxValue(Short.class)))
{
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals("Kuldeep", student.getName());
count++;
}
else
{
Assert.assertEquals(getMinValue(Short.class), student.getId());
Assert.assertEquals(getMinValue(short.class), student.getAge());
Assert.assertEquals(getMinValue(String.class), student.getName());
count++;
}
}
Assert.assertEquals(2, count);
em.close();
}
private void findByAge()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.age = " + getRandomValue(short.class);
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(1, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
Assert.assertEquals(getRandomValue(Short.class), student.getId());
Assert.assertEquals(getRandomValue(short.class), student.getAge());
Assert.assertEquals(getRandomValue(String.class), student.getName());
count++;
}
Assert.assertEquals(1, count);
em.close();
}
/**
*
*/
private void findByName()
{
EntityManager em;
String query;
Query q;
List<StudentShortWrapper> students;
int count;
em = emf.createEntityManager();
query = "Select s From StudentShortWrapper s where s.name = Kuldeep";
q = em.createQuery(query);
students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(2, students.size());
count = 0;
for (StudentShortWrapper student : students)
{
if (student.getId().equals(getMaxValue(Short.class)))
{
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals("Kuldeep", student.getName());
count++;
}
else
{
Assert.assertEquals(getMinValue(Short.class), student.getId());
Assert.assertEquals(getMinValue(short.class), student.getAge());
Assert.assertEquals(getMinValue(String.class), student.getName());
count++;
}
}
Assert.assertEquals(2, count);
em.close();
}
/**
*
*/
private void findAllQuery()
{
EntityManager em = emf.createEntityManager();
// Selet all query.
String query = "Select s From StudentShortWrapper s ";
Query q = em.createQuery(query);
List<StudentShortWrapper> students = q.getResultList();
Assert.assertNotNull(students);
Assert.assertEquals(3, students.size());
int count = 0;
for (StudentShortWrapper student : students)
{
if (student.getId().equals(getMaxValue(Short.class)))
{
Assert.assertEquals(getMaxValue(short.class), student.getAge());
Assert.assertEquals("Kuldeep", student.getName());
count++;
}
else if (student.getId().equals(getMinValue(Short.class)))
{
Assert.assertEquals(getMinValue(short.class), student.getAge());
Assert.assertEquals(getMinValue(String.class), student.getName());
count++;
}
else
{
Assert.assertEquals(getRandomValue(Short.class), student.getId());
Assert.assertEquals(getRandomValue(short.class), student.getAge());
Assert.assertEquals(getRandomValue(String.class), student.getName());
count++;
}
}
Assert.assertEquals(3, count);
em.close();
}
public void startCluster()
{
try
{
CassandraCli.cassandraSetUp();
}
catch (IOException e)
{
}
catch (TException e)
{
}
}
public void stopCluster()
{
// TODO Auto-generated method stub
}
public void createSchema()
{
try
{
KsDef ksDef = null;
CfDef cfDef = new CfDef();
cfDef.name = "StudentShortWrapper";
cfDef.keyspace = keyspace;
cfDef.setKey_validation_class("Int32Type");
cfDef.setComparator_type("UTF8Type");
ColumnDef name = new ColumnDef(ByteBuffer.wrap("NAME".getBytes()), "UTF8Type");
name.index_type = IndexType.KEYS;
cfDef.addToColumn_metadata(name);
ColumnDef age = new ColumnDef(ByteBuffer.wrap("AGE".getBytes()), "Int32Type");
age.index_type = IndexType.KEYS;
cfDef.addToColumn_metadata(age);
List<CfDef> cfDefs = new ArrayList<CfDef>();
cfDefs.add(cfDef);
try
{
CassandraCli.initClient();
ksDef = CassandraCli.client.describe_keyspace(keyspace);
CassandraCli.client.set_keyspace(keyspace);
List<CfDef> cfDefn = ksDef.getCf_defs();
for (CfDef cfDef1 : cfDefn)
{
if (cfDef1.getName().equalsIgnoreCase("StudentShortWrapper"))
{
CassandraCli.client.system_drop_column_family("StudentShortWrapper");
}
}
CassandraCli.client.system_add_column_family(cfDef);
}
catch (NotFoundException e)
{
ksDef = new KsDef(keyspace, "org.apache.cassandra.locator.SimpleStrategy", cfDefs);
// Set replication factor
if (ksDef.strategy_options == null)
{
ksDef.strategy_options = new LinkedHashMap<String, String>();
}
// Set replication factor, the value MUST be an Short
ksDef.strategy_options.put("replication_factor", "1");
CassandraCli.client.system_add_keyspace(ksDef);
}
CassandraCli.client.set_keyspace(keyspace);
}
catch (TException e)
{
}
}
public void dropSchema()
{
CassandraCli.executeCqlQuery("TRUNCATE \"StudentShortWrapper\"", keyspace);
}
}