Test under highly concurrent access conditions

coreutils-fiber/src/test/java/org/grycap/coreutils/fiber/test/HighlyConcurrencyTest.java

@@ -0,0 +1,92 @@
+/*
+ * Core Utils - Fiber-enabled clients.
+ * Copyright 2015-2016 GRyCAP (Universitat Politecnica de Valencia)
+ * 
+ * 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.
+ * 
+ * This product combines work with different licenses. See the "NOTICE" text
+ * file for details on the various modules and licenses.
+ * 
+ * The "NOTICE" text file is part of the distribution. Any derivative works
+ * that you distribute must include a readable copy of the "NOTICE" text file.
+ */
+
+package org.grycap.coreutils.fiber.test;
+
+import static com.google.common.collect.ImmutableList.of;
+import static org.grycap.coreutils.fiber.http.Http2Client.getHttp2Client;
+import static org.grycap.coreutils.fiber.test.mockserver.FiberExpectationInitializer.MOCK_SERVER_BASE_URL;
+import static org.hamcrest.CoreMatchers.allOf;
+import static org.hamcrest.CoreMatchers.equalTo;
+import static org.hamcrest.CoreMatchers.notNullValue;
+import static org.hamcrest.MatcherAssert.assertThat;
+
+import java.io.IOException;
+import java.util.concurrent.TimeUnit;
+
+import org.grycap.coreutils.fiber.http.Http2Client;
+import org.grycap.coreutils.test.category.IntegrationTests;
+import org.grycap.coreutils.test.rules.TestPrinter;
+import org.grycap.coreutils.test.rules.TestWatcher2;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.experimental.categories.Category;
+import org.junit.rules.TestRule;
+
+import com.squareup.okhttp.Callback;
+import com.squareup.okhttp.Request;
+import com.squareup.okhttp.Response;
+
+import net.jodah.concurrentunit.Waiter;
+
+/**
+ * Tests the {@link Http2Client} under highly concurrency conditions.
+ * @author Erik Torres <etserrano@gmail.com>
+ * @since 0.1.0
+ */
+@Category(IntegrationTests.class)
+public class HighlyConcurrencyTest {
+
+	@Rule
+	public TestPrinter pw = new TestPrinter();
+
+	@Rule
+	public TestRule watchman = new TestWatcher2(pw);
+
+	@Test
+	public void test() throws Exception {
+		// create the client
+		final Http2Client client = getHttp2Client();
+		assertThat("HTTP+SPDY client was created", client, notNullValue());
+
+		// prepare the test
+		final Waiter waiter = new Waiter();
+		for (int i = 0; i < 100; i++) {
+			final int id = i;
+			client.asyncGet(new StringBuilder(MOCK_SERVER_BASE_URL).append("/test/concurrent/").append(i).toString(), of("application/json"), true, new Callback() {
+				@Override
+				public void onResponse(final Response response) throws IOException {
+					waiter.assertTrue(response.isSuccessful());
+					final String payload = response.body().source().readUtf8();
+					waiter.assertThat(payload, allOf(notNullValue(), equalTo(new StringBuilder("Response-to-").append(id).toString())));
+					waiter.resume();
+				}
+				@Override
+				public void onFailure(final Request request, final IOException throwable) {
+					waiter.fail(throwable);			
+				}
+			});
+		}
+		waiter.await(30l, TimeUnit.SECONDS, 100);
+	}	
+}

coreutils-fiber/src/test/java/org/grycap/coreutils/fiber/test/mockserver/FiberExpectationInitializer.java

@@ -118,6 +118,14 @@ public class FiberExpectationInitializer implements ExpectationInitializer {
 					.withHeaders(new Header("Cache-Control", "public, max-age=86400"))
 					.withDelay(new Delay(TimeUnit.MILLISECONDS, r.getDelay())));
 		});
+		// highly concurrent
+		for (int i = 0; i < 100; i++) {
+			client.when(request().withMethod("GET").withPath(String.format("/test/concurrent/%d", i)))
+			.respond(response().withStatusCode(200)
+					.withHeaders(new Header("Content-Type", "text/plain; charset=utf-8"),
+							new Header("Cache-Control", "public, max-age=86400"))
+					.withBody(new StringBuilder("Response-to-").append(i).toString()));
+		}
 	}
 
 }