From 7d6b72a2af93bc4f689cfce325be448ce3afb9f7 Mon Sep 17 00:00:00 2001
From: tschiman <tschiman@gmail.com>
Date: Mon, 19 Dec 2016 23:01:40 -0700
Subject: [PATCH] BAEL-315 Adding test class for demonstrating features of
 Spring Session and Spring Cloud

---
 .../gateway/GatewayApplicationTest.java       | 69 +++++++++++++++++++
 1 file changed, 69 insertions(+)
 create mode 100644 spring-cloud/spring-cloud-bootstrap/gateway/src/test/java/com/baeldung/spring/cloud/bootstrap/gateway/GatewayApplicationTest.java

diff --git a/spring-cloud/spring-cloud-bootstrap/gateway/src/test/java/com/baeldung/spring/cloud/bootstrap/gateway/GatewayApplicationTest.java b/spring-cloud/spring-cloud-bootstrap/gateway/src/test/java/com/baeldung/spring/cloud/bootstrap/gateway/GatewayApplicationTest.java
new file mode 100644
index 0000000000..d6ceac32f0
--- /dev/null
+++ b/spring-cloud/spring-cloud-bootstrap/gateway/src/test/java/com/baeldung/spring/cloud/bootstrap/gateway/GatewayApplicationTest.java
@@ -0,0 +1,69 @@
+package com.baeldung.spring.cloud.bootstrap.gateway;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.springframework.boot.test.web.client.TestRestTemplate;
+import org.springframework.http.*;
+import org.springframework.util.LinkedMultiValueMap;
+import org.springframework.util.MultiValueMap;
+
+public class GatewayApplicationTest {
+
+    @Test
+    public void testAccess() throws Exception {
+        TestRestTemplate testRestTemplate = new TestRestTemplate();
+        String testUrl = "http://localhost:8080";
+
+        ResponseEntity<String> response = testRestTemplate.getForEntity(testUrl + "/resource/hello/cloud", String.class);
+        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
+        Assert.assertEquals("hello cloud", response.getBody());
+
+        //try the protected resource and confirm the redirect to login
+        response = testRestTemplate.getForEntity(testUrl + "/resource/hello/user", String.class);
+        Assert.assertEquals(HttpStatus.FOUND, response.getStatusCode());
+        Assert.assertEquals("http://localhost:8080/login", response.getHeaders().get("Location").get(0));
+
+        //login as user/password
+        MultiValueMap<String, String> form = new LinkedMultiValueMap<>();
+        form.add("username", "user");
+        form.add("password", "password");
+        response = testRestTemplate.postForEntity(testUrl + "/login", form, String.class);
+
+        //extract the session from the cookie and propagate it to the next request
+        String sessionCookie = response.getHeaders().get("Set-Cookie").get(0).split(";")[0];
+        HttpHeaders headers = new HttpHeaders();
+        headers.add("Cookie", sessionCookie);
+        HttpEntity<String> httpEntity = new HttpEntity<>(headers);
+
+        //request the protected resource
+        response = testRestTemplate.exchange(testUrl + "/resource/hello/user", HttpMethod.GET, httpEntity, String.class);
+        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
+        Assert.assertEquals("hello cloud user", response.getBody());
+
+        //request the admin protected resource to determine it is still protected
+        response = testRestTemplate.exchange(testUrl + "/resource/hello/admin", HttpMethod.GET, httpEntity, String.class);
+        Assert.assertEquals(HttpStatus.FORBIDDEN, response.getStatusCode());
+
+        //login as the admin
+        form.clear();
+        form.add("username", "admin");
+        form.add("password", "admin");
+        response = testRestTemplate.postForEntity(testUrl + "/login", form, String.class);
+
+        //extract the session from the cookie and propagate it to the next request
+        sessionCookie = response.getHeaders().get("Set-Cookie").get(0).split(";")[0];
+        headers = new HttpHeaders();
+        headers.add("Cookie", sessionCookie);
+        httpEntity = new HttpEntity<>(headers);
+
+        //request the protected resource
+        response = testRestTemplate.exchange(testUrl + "/resource/hello/admin", HttpMethod.GET, httpEntity, String.class);
+        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
+        Assert.assertEquals("hello cloud admin", response.getBody());
+
+        //request the discovery resources as the admin
+        response = testRestTemplate.exchange(testUrl + "/discovery", HttpMethod.GET, httpEntity, String.class);
+        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
+    }
+
+}
\ No newline at end of file