Add Integration Tests for Spring Geode Sample on Async Inline Caching configured with Queue Batch Time Interval.

spring-geode-samples/caching/inline-async/src/test/java/example/app/caching/inline/async/queue_batch_time_interval/AsyncInlineCachingUsingQueueBatchTimeIntervalIntegrationTests.java

@@ -0,0 +1,193 @@
+/*
+ * Copyright 2020 the original author or authors.
+ *
+ * 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
+ *
+ *     https://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 example.app.caching.inline.async.queue_batch_time_interval;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.springframework.data.gemfire.util.RuntimeExceptionFactory.newIllegalStateException;
+
+import java.time.Duration;
+import java.util.Optional;
+
+import javax.annotation.Resource;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import org.apache.geode.cache.Cache;
+import org.apache.geode.cache.DataPolicy;
+import org.apache.geode.cache.Region;
+import org.apache.geode.cache.RegionShortcut;
+import org.apache.geode.cache.asyncqueue.AsyncEventListener;
+import org.apache.geode.cache.asyncqueue.AsyncEventQueue;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
+import org.springframework.boot.autoconfigure.domain.EntityScan;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Import;
+import org.springframework.data.gemfire.GemfireTemplate;
+import org.springframework.data.gemfire.config.annotation.EnableEntityDefinedRegions;
+import org.springframework.data.gemfire.config.annotation.PeerCacheApplication;
+import org.springframework.data.gemfire.tests.util.ReflectionUtils;
+import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
+import org.springframework.geode.cache.RepositoryAsyncEventListener;
+import org.springframework.test.context.ActiveProfiles;
+import org.springframework.test.context.junit4.SpringRunner;
+
+import org.awaitility.Awaitility;
+
+import example.app.caching.inline.async.client.model.Golfer;
+import example.app.caching.inline.async.client.repo.GolferRepository;
+import example.app.caching.inline.async.client.service.GolferService;
+import example.app.caching.inline.async.config.AsyncInlineCachingConfiguration;
+
+/**
+ * Integration Tests for Spring Boot configured Async Inline Caching with Apache Geode using Queue Batch Time Interval.
+ *
+ * @author John Blum
+ * @see java.time.Duration
+ * @see org.junit.Test
+ * @see org.apache.geode.cache.Cache
+ * @see org.apache.geode.cache.Region
+ * @see org.apache.geode.cache.asyncqueue.AsyncEventListener
+ * @see org.apache.geode.cache.asyncqueue.AsyncEventQueue
+ * @see org.springframework.boot.autoconfigure.EnableAutoConfiguration
+ * @see org.springframework.boot.autoconfigure.domain.EntityScan
+ * @see org.springframework.boot.test.context.SpringBootTest
+ * @see org.springframework.context.annotation.Bean
+ * @see org.springframework.context.annotation.Import
+ * @see org.springframework.data.gemfire.config.annotation.PeerCacheApplication
+ * @see org.springframework.data.jpa.repository.config.EnableJpaRepositories
+ * @see org.springframework.geode.cache.RepositoryAsyncEventListener
+ * @see org.springframework.test.context.ActiveProfiles
+ * @see org.springframework.test.context.junit4.SpringRunner
+ * @see example.app.caching.inline.async.config.AsyncInlineCachingConfiguration
+ * @since 1.4.0
+ */
+@RunWith(SpringRunner.class)
+@ActiveProfiles("queue-batch-time-interval")
+@SpringBootTest(
+	properties = {
+		//"spring.jpa.show-sql=true",
+		"spring.jpa.hibernate.ddl-auto=none",
+		"spring.geode.sample.async-inline-caching.queue.batch-time-interval-ms=2000"
+	},
+	webEnvironment = SpringBootTest.WebEnvironment.NONE
+)
+@SuppressWarnings("unused")
+public class AsyncInlineCachingUsingQueueBatchTimeIntervalIntegrationTests {
+
+	private volatile AsyncEventQueue golfersEventQueue;
+
+	@Autowired
+	private GolferService golferService;
+
+	@Resource(name = "Golfers")
+	private Region<String, Golfer> golfers;
+
+	private volatile RepositoryAsyncEventListener<Golfer, String> golfersEventQueueListener;
+
+	@Before
+	public void assertGolfApplicationState() {
+
+		assertThat(this.golferService).isNotNull();
+		assertThat(this.golferService.getAllGolfersFromCache()).isEmpty();
+		assertThat(this.golferService.getAllGolfersFromDatabase()).isEmpty();
+	}
+
+	@Before
+	@SuppressWarnings("unchecked")
+	public void assertRegionAndAsyncEventQueueSetup() throws Exception {
+
+		assertThat(this.golfers).isNotNull();
+		assertThat(this.golfers.getName()).isEqualTo("Golfers");
+		assertThat(this.golfers.getAttributes()).isNotNull();
+		assertThat(this.golfers.getAttributes().getDataPolicy()).isEqualTo(DataPolicy.REPLICATE);
+		assertThat(this.golfers.getAttributes().getAsyncEventQueueIds()).hasSize(1);
+
+		String asyncEventQueueId = this.golfers.getAttributes().getAsyncEventQueueIds().iterator().next();
+
+		assertThat(asyncEventQueueId).isNotEmpty();
+		assertThat(asyncEventQueueId).startsWith("Golfers-AEQ-");
+
+		AsyncEventQueue golfersEventQueue = Optional.of(this.golfers)
+			.map(Region::getRegionService)
+			.filter(Cache.class::isInstance)
+			.map(Cache.class::cast)
+			.map(cache -> cache.getAsyncEventQueue(asyncEventQueueId))
+			.orElseThrow(() -> newIllegalStateException("AEQ with ID [%s] not found", asyncEventQueueId));
+
+		assertThat(golfersEventQueue).isNotNull();
+		assertThat(golfersEventQueue.getBatchSize()).isEqualTo(1000000);
+		assertThat(golfersEventQueue.getBatchTimeInterval()).isEqualTo(Duration.ofSeconds(2).toMillis());
+		assertThat(golfersEventQueue.getDispatcherThreads()).isEqualTo(1);
+		assertThat(golfersEventQueue.isParallel()).isFalse();
+
+		this.golfersEventQueue = golfersEventQueue;
+
+		AsyncEventListener listener = golfersEventQueue.getAsyncEventListener();
+
+		assertThat(listener).isInstanceOf(RepositoryAsyncEventListener.class);
+
+		RepositoryAsyncEventListener<Golfer, String> repositoryListener =
+			(RepositoryAsyncEventListener<Golfer, String>) listener;
+
+		assertThat(ReflectionUtils.<Object>getFieldValue(repositoryListener, "repository"))
+			.isInstanceOf(GolferRepository.class);
+
+		this.golfersEventQueueListener = repositoryListener;
+	}
+
+	@Test
+	public void databaseIsOnlyUpdatedOnCacheRegionQueueBatchTimeIntervals() {
+
+		assertThat(this.golfersEventQueue).isNotNull();
+		assertThat(this.golfersEventQueueListener).isNotNull();
+
+		Golfer tigerWoods = Golfer.newGolfer("Tiger Woods");
+
+		this.golferService.update(tigerWoods);
+
+		assertThat(this.golferService.getAllGolfersFromCache()).containsExactly(tigerWoods);
+		assertThat(this.golferService.getAllGolfersFromDatabase()).isEmpty();
+
+		Awaitility.with().pollInSameThread().pollDelay(Duration.ofSeconds(1))
+			.await().atMost(Duration.ofSeconds(15))
+			.until(this.golfersEventQueueListener::hasFired);
+
+		assertThat(this.golferService.getAllGolfersFromCache()).containsExactly(tigerWoods);
+		assertThat(this.golferService.getAllGolfersFromDatabase()).containsExactly(tigerWoods);
+	}
+
+	@PeerCacheApplication
+	@EnableAutoConfiguration
+	@EnableEntityDefinedRegions(basePackageClasses = Golfer.class, serverRegionShortcut = RegionShortcut.REPLICATE)
+	@EnableJpaRepositories(basePackageClasses = GolferRepository.class)
+	@EntityScan(basePackageClasses = Golfer.class)
+	@Import(AsyncInlineCachingConfiguration.class)
+	static class TestConfiguration {
+
+		@Bean
+		GolferService golferService(@Qualifier("golfersTemplate") GemfireTemplate golfersTemplate,
+			GolferRepository golferRepository) {
+
+			return new GolferService(golfersTemplate, golferRepository);
+		}
+	}
+}