com.gemstone.gemfire.internal.cache.PartitionedRegionTestHelper Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of gemfire-junit Show documentation
Show all versions of gemfire-junit Show documentation
SnappyData store based off Pivotal GemFireXD
/*
* Copyright (c) 2010-2015 Pivotal Software, Inc. 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. See accompanying
* LICENSE file.
*/
package com.gemstone.gemfire.internal.cache;
import java.io.Serializable;
import java.util.Properties;
import junit.framework.Assert;
import com.gemstone.gemfire.LogWriter;
import com.gemstone.gemfire.cache.AttributesFactory;
import com.gemstone.gemfire.cache.Cache;
import com.gemstone.gemfire.cache.CacheExistsException;
import com.gemstone.gemfire.cache.CacheFactory;
import com.gemstone.gemfire.cache.DataPolicy;
import com.gemstone.gemfire.cache.EvictionAttributes;
import com.gemstone.gemfire.cache.PartitionAttributes;
import com.gemstone.gemfire.cache.PartitionAttributesFactory;
import com.gemstone.gemfire.cache.Region;
import com.gemstone.gemfire.cache.RegionAttributes;
import com.gemstone.gemfire.cache.RegionExistsException;
import com.gemstone.gemfire.cache.Scope;
import com.gemstone.gemfire.distributed.DistributedSystem;
import com.gemstone.gemfire.cache.query.SelectResults;
import com.gemstone.gemfire.cache.query.types.ObjectType;
/**
* This helper class is used by other test. This has functions to create region.
*
* @author tnegi, modified by tapshank, modified by gthombar
*
*/
public class PartitionedRegionTestHelper
{
static Cache cache = null;
/**
* This method creates a partitioned region with all the default values.
* The cache created is a loner, so this is only suitable for single VM tests.
*
* @param regionname
* @return region
* @throws RegionExistsException
*/
public static Region createPartionedRegion(String regionname)
throws RegionExistsException
{
AttributesFactory attribFactory = new AttributesFactory();
PartitionAttributesFactory paf = new PartitionAttributesFactory();
PartitionAttributes prattribs = paf.create();
attribFactory.setDataPolicy(DataPolicy.PARTITION);
attribFactory.setPartitionAttributes(prattribs);
RegionAttributes regionAttribs = attribFactory.create();
Region partitionedregion = createCache().createRegion(regionname, regionAttribs);
return partitionedregion;
}
/**
* This method creates a local region with all the default values.
* The cache created is a loner, so this is only suitable for single VM tests.
*
* @param regionname
* @return region
* @throws RegionExistsException
*/
public static Region createLocalRegion(String regionName)throws RegionExistsException{
AttributesFactory attr = new AttributesFactory();
attr.setScope(Scope.LOCAL);
Region localRegion = createCache().createRegion(regionName, attr.create());
return localRegion;
}
/**
* This method compares two selectResult Objects by
* 1. Size
* 2. SelectResults#CollectionType#ElementType()
*
* @param SelectResults
* @return void
* @throws
*/
public static String compareResultSets(SelectResults sr1 , SelectResults sr2){
ObjectType type1, type2;
String failureString = null;
type1 = sr1.getCollectionType().getElementType();
Assert.assertNotNull("PartitionedRegionTestHelper#compareResultSets: Type 1 is NULL " + type1, type1);
type2 = sr2.getCollectionType().getElementType();
Assert.assertNotNull(
"PartitionedRegionTestHelper#compareResultSets: Type 2 is NULL " + type2,
type2);
if ((type1.getClass().getName()).equals(type2.getClass().getName())) {
getLogger().info(
"PartitionedRegionTestHelper#compareResultSets: Both Search Results are of the same Type i.e.--> "
+ type1);
}
else {
getLogger()
.error(
"PartitionedRegionTestHelper#compareTwoQueryResults: Classes are : "
+ type1.getClass().getName() + " "
+ type2.getClass().getName());
failureString ="PartitionedRegionTestHelper#compareResultSets: FAILED:Search result Type is different in both the cases"+ type1.getClass().getName() + " "
+ type2.getClass().getName();
Assert.fail("PartitionedRegionTestHelper#compareResultSets: FAILED:Search result Type is different in both the cases");
return failureString;
}
if ((sr1.size()) == (sr2.size())) {
getLogger()
.info(
"PartitionedRegionTestHelper#compareResultSets: Both Search Results are non-zero and are of Same Size i.e. Size= "
+ sr1.size());
}
else {
getLogger()
.error(
"PartitionedRegionTestHelper#compareResultSets: FAILED:Search resultSet size are different in both the cases");
failureString="PartitionedRegionTestHelper#compareResultSets: FAILED:Search resultSet size are different in both the cases"+sr1.size()+" "+sr2.size();
Assert.fail("PartitionedRegionTestHelper#compareResultSets: FAILED:Search resultSet size are different in both the cases");
}
return failureString;
}
/**
* This is a function to create partitioned region with following paramaters:
*
* 1) name
*
* 2) local max memory
*
* 3) redundancy and scope.
*
* The cache created is a loner, so this is only suitable for single VM tests.
*
* @param regionName
* @param redundancy
* @param localMemory
* @return Region
*/
public static Region createPartitionedRegion(String regionName,
String localMaxMemory, int redundancy)
{
Region pr = null;
PartitionAttributes pa;
PartitionAttributesFactory paf = new PartitionAttributesFactory();
AttributesFactory af = new AttributesFactory();
RegionAttributes ra;
// setting property
// setting partition attributes to partitionAttributesFactory
int lmax;
try {
lmax = Integer.parseInt(localMaxMemory);
}
catch (NumberFormatException nfe) {
throw new IllegalArgumentException("localMaxMemory must be an integer (" + localMaxMemory + ")");
}
pa = paf.setLocalMaxMemory(lmax).setRedundantCopies(redundancy)
.create();
// setting attribute factor
af.setPartitionAttributes(pa);
// creating region attributes
ra = af.create();
cache = createCache();
try {
pr = cache.createRegion(regionName, ra);
}
catch (RegionExistsException rex) {
pr = cache.getRegion(regionName);
}
return pr;
}
/**
* This function is used to create serializable object for the partition
* region test.
*
* @param name
* @param id
* @return
*/
public static SerializableObject createPRSerializableObject(String name,
int id)
{
Object obj = new SerializableObject(name, id);
return (SerializableObject)obj;
}
/**
* This method creates cache. The cache created
* is a loner, so this is only suitable for single VM tests.
*
* @return
*/
public static synchronized Cache createCache()
{
if (cache == null) {
Properties dsp = new Properties();
dsp.setProperty("mcast-port", "0");
dsp.setProperty("locators", "");
DistributedSystem sys = DistributedSystem.connect(dsp);
try {
cache = CacheFactory.create(sys);
}
catch (CacheExistsException exp) {
cache = CacheFactory.getInstance(sys);
}
catch (RegionExistsException rex) {
cache = CacheFactory.getInstance(sys);
}
}
return cache;
}
/**
* This method closes the cache.
*
*
* @return
*/
public static synchronized void closeCache()
{
if(cache != null){
cache.close();
}
}
/**
* This method is used to return existing region.
*
* @param PRName
* @return
*/
public static Region getExistingRegion(String PRName)
{
createCache();
return cache.getRegion(PRName);
}
/**
* Gets the log writer for the
* The cache created is a loner, so this is only suitable for single VM tests.
*
* @return LogWriter
*/
public static LogWriter getLogger()
{
return createCache().getLogger();
}
public static RegionAttributes createRegionAttrsForPR(int red, int localMaxMem)
{
return createRegionAttrsForPR(red, localMaxMem,
PartitionAttributesFactory.RECOVERY_DELAY_DEFAULT);
}
/**
* This function creates Region attributes with provided scope,redundancy and
* localmaxMemory
*/
public static RegionAttributes createRegionAttrsForPR(int red, int localMaxMem, long recoveryDelay)
{
return createRegionAttrsForPR(red, localMaxMem, recoveryDelay, null);
}
/**
* This function creates Region attributes with provided scope,redundancy and
* localmaxMemory
*/
public static RegionAttributes createRegionAttrsForPR(int red, int localMaxMem, long recoveryDelay, EvictionAttributes evictionAttrs)
{
AttributesFactory attr = new AttributesFactory();
attr.setDataPolicy(DataPolicy.PARTITION);
PartitionAttributesFactory paf = new PartitionAttributesFactory();
PartitionAttributes prAttr = paf.setRedundantCopies(red)
.setLocalMaxMemory(localMaxMem).setRecoveryDelay(recoveryDelay).create();
attr.setPartitionAttributes(prAttr);
attr.setEvictionAttributes(evictionAttrs);
return attr.create();
}
}
/**
* class for creating serializable object which is used for LocalMaxMemory verification.
*/
class SerializableObject implements Serializable
{
String str;
int i;
public SerializableObject(String str, int i) {
this.str = str;
this.i = i;
}
public boolean equals(Object obj)
{
if (obj == null)
return false;
if (this.str.equals(((SerializableObject)obj).str) && this.i == ((SerializableObject)obj).i)
return true;
return false;
}
}