| /*========================================================================= |
| * Copyright (c) 2010-2014 Pivotal Software, Inc. All Rights Reserved. |
| * This product is protected by U.S. and international copyright |
| * and intellectual property laws. Pivotal products are covered by |
| * one or more patents listed at http://www.pivotal.io/patents. |
| *========================================================================= |
| */ |
| package com.gemstone.gemfire.cache.query; |
| |
| import static org.junit.Assert.*; |
| |
| import java.util.Properties; |
| |
| import com.gemstone.gemfire.cache.AttributesFactory; |
| import com.gemstone.gemfire.cache.Cache; |
| import com.gemstone.gemfire.cache.CacheFactory; |
| import com.gemstone.gemfire.cache.Region; |
| import com.gemstone.gemfire.cache.RegionAttributes; |
| import com.gemstone.gemfire.distributed.DistributedSystem; |
| import com.gemstone.gemfire.cache.query.data.Address; |
| import com.gemstone.gemfire.cache.query.data.Portfolio; |
| import com.gemstone.gemfire.cache.query.data.Data; |
| import com.gemstone.gemfire.cache.query.data.Employee; |
| import com.gemstone.gemfire.cache.query.data.Manager; |
| import com.gemstone.gemfire.test.junit.categories.IntegrationTest; |
| |
| import java.util.HashSet; |
| import java.util.Set; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.experimental.categories.Category; |
| |
| /** |
| * Junit test for checking the value constraint region attributes. |
| * @author kdeshpan |
| */ |
| @Category(IntegrationTest.class) |
| public class Bug32947ValueConstraintJUnitTest { |
| |
| DistributedSystem distributedSystem = null; |
| Cache cache = null; |
| |
| @Before |
| public void setUp() throws java.lang.Exception { |
| Properties properties = new Properties(); |
| properties.setProperty("mcast-port", "0"); |
| distributedSystem = DistributedSystem.connect(properties); |
| cache = CacheFactory.create(distributedSystem); |
| } |
| |
| @After |
| public void tearDown() { |
| cache.close(); |
| distributedSystem.disconnect(); |
| } |
| |
| @Test |
| public void testBug32947ValueConstraints() throws Exception { |
| |
| boolean flag = false; |
| AttributesFactory factory = new AttributesFactory(); |
| factory.setValueConstraint(Portfolio.class); |
| RegionAttributes regionAttributes = factory.create(); |
| Region portolioRegion = cache.createRegion("portfolios", regionAttributes); |
| |
| portolioRegion.put("key1", new Portfolio(1)); |
| try { |
| portolioRegion.put("key2", new Data()); |
| } catch (ClassCastException e) { |
| flag = true; |
| } |
| |
| if (!flag) { |
| fail("Expected ClassCastException after put as valueConstraint is set to Portfolio.class"); |
| return; |
| } |
| |
| Set address1 = new HashSet(); |
| Set address2 = new HashSet(); |
| address1.add(new Address("Hp3 9yf", "Apsley")); |
| address1.add(new Address("Hp4 9yf", "Apsleyss")); |
| address2.add(new Address("Hp3 8DZ", "Hemel")); |
| address2.add(new Address("Hp4 8DZ", "Hemel")); |
| |
| // Note that Manager extends Employee |
| |
| Manager manager = new Manager("aaa", 27, 270, "QA", 1800, address1, 2701); |
| Employee employee = new Employee("bbb", 28, 280, "QA", 1900, address2); |
| |
| factory.setValueConstraint(Manager.class); |
| regionAttributes = factory.create(); |
| Region managerRegion = cache.createRegion("managers", regionAttributes); |
| |
| factory.setValueConstraint(Employee.class); |
| regionAttributes = factory.create(); |
| Region employeeRegion = cache.createRegion("employees", regionAttributes); |
| |
| employeeRegion.put("key1", manager); //This is perfectly valid, as Manager is Derived from Employee |
| |
| flag = false; |
| |
| try { |
| managerRegion.put("key1", employee); |
| } catch (ClassCastException e) { |
| flag = true; |
| } |
| |
| if (!flag) { |
| fail("Expected ClassCastException after put as valueConstraint is set to Manager.class"); |
| return; |
| } |
| |
| // Test passed successfully. |
| |
| } |
| } |