| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more contributor license |
| * agreements. See the NOTICE file distributed with this work for additional information regarding |
| * copyright ownership. The ASF licenses this file to You 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 org.apache.geode.management; |
| |
| import static org.apache.geode.cache.Region.SEPARATOR; |
| import static org.assertj.core.api.Assertions.assertThat; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.lang.reflect.Constructor; |
| import java.net.URI; |
| import java.net.URISyntaxException; |
| import java.net.URL; |
| import java.nio.file.Files; |
| import java.util.List; |
| |
| import org.jetbrains.annotations.NotNull; |
| import org.junit.BeforeClass; |
| import org.junit.ClassRule; |
| import org.junit.Test; |
| import org.junit.experimental.categories.Category; |
| |
| import org.apache.geode.cache.Region; |
| import org.apache.geode.cache.client.ClientRegionFactory; |
| import org.apache.geode.cache.client.ClientRegionShortcut; |
| import org.apache.geode.pdx.ReflectionBasedAutoSerializer; |
| import org.apache.geode.pdx.internal.AutoSerializableManager; |
| import org.apache.geode.test.compiler.InMemoryClassFile; |
| import org.apache.geode.test.compiler.InMemoryClassFileLoader; |
| import org.apache.geode.test.compiler.InMemoryJavaCompiler; |
| import org.apache.geode.test.dunit.Invoke; |
| import org.apache.geode.test.dunit.rules.ClientVM; |
| import org.apache.geode.test.dunit.rules.ClusterStartupRule; |
| import org.apache.geode.test.dunit.rules.MemberVM; |
| import org.apache.geode.test.junit.categories.GfshTest; |
| import org.apache.geode.test.junit.rules.GfshCommandRule; |
| |
| @Category({GfshTest.class}) |
| public class QueryPdxDataDUnitTest { |
| |
| private static MemberVM locator; |
| private static MemberVM server1; |
| private static MemberVM server2; |
| private static ClientVM client; |
| |
| @ClassRule |
| public static ClusterStartupRule cluster = new ClusterStartupRule(); |
| |
| @ClassRule |
| public static GfshCommandRule gfsh = new GfshCommandRule(); |
| |
| @BeforeClass |
| public static void beforeClass() throws Exception { |
| Invoke.invokeInEveryVM( |
| () -> System.setProperty(AutoSerializableManager.NO_HARDCODED_EXCLUDES_PARAM, "true")); |
| |
| locator = cluster.startLocatorVM(0); |
| gfsh.connectAndVerify(locator); |
| gfsh.executeAndAssertThat( |
| "configure pdx --read-serialized=true --auto-serializable-classes=org.apache.geode" |
| + ".management.*") |
| .statusIsSuccess(); |
| |
| server1 = cluster.startServerVM(1, locator.getPort()); |
| server2 = cluster.startServerVM(2, locator.getPort()); |
| |
| gfsh.executeAndAssertThat("create region --name=BOZ --type=PARTITION"); |
| |
| int port = locator.getPort(); |
| client = cluster.startClientVM(3, cf -> { |
| cf.withLocatorConnection(port); |
| cf.withCacheSetup(c -> c |
| .setPdxSerializer(new ReflectionBasedAutoSerializer("org.apache.geode.management.*"))); |
| }); |
| |
| client.invoke(() -> { |
| ClientRegionFactory factory = |
| ClusterStartupRule.getClientCache().createClientRegionFactory(ClientRegionShortcut.PROXY); |
| Region<Object, Object> region = factory.create("BOZ"); |
| |
| InMemoryJavaCompiler compiler = new InMemoryJavaCompiler(); |
| List<InMemoryClassFile> classFiles = compiler.compile( |
| contentOfResource("/org/apache/geode/management/Address.java"), |
| contentOfResource("/org/apache/geode/management/Customer.java")); |
| InMemoryClassFileLoader classLoader = new InMemoryClassFileLoader(); |
| List<Class<?>> classes = classLoader.defineClasses(classFiles); |
| Class<?> customerClass = classes.get(1); |
| Constructor<?> constructor = |
| customerClass.getConstructor(String.class, String.class, String.class); |
| for (int i = 0; i < 100; i++) { |
| Object customer = constructor.newInstance("name_" + i, "Main " + i, "City " + i); |
| region.put(i + "", customer); |
| } |
| }); |
| } |
| |
| @NotNull |
| private static String contentOfResource(String javaResourceName) |
| throws URISyntaxException, IOException { |
| URL resourceFileURL = QueryPdxDataDUnitTest.class.getResource(javaResourceName); |
| assertThat(resourceFileURL).isNotNull(); |
| URI resourceUri = resourceFileURL.toURI(); |
| return new String(Files.readAllBytes(new File(resourceUri).toPath())); |
| } |
| |
| @Test |
| public void queryForPdxBackedEntriesShouldSucceed() { |
| gfsh.executeAndAssertThat("query --query=\"select * from " + SEPARATOR + "BOZ.values\"") |
| .statusIsSuccess(); |
| } |
| } |