- Add REST Integration tests.

wise-webapp/src/test/java/com/wisemapping/test/rest/RestAdminITCase.java

@@ -1,17 +1,98 @@
 package com.wisemapping.test.rest;
 
 
+import com.wisemapping.rest.model.RestUser;
+import org.jetbrains.annotations.NonNls;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.http.*;
+import org.springframework.http.client.SimpleClientHttpRequestFactory;
+import org.springframework.security.crypto.codec.Base64;
+import org.springframework.web.client.RestTemplate;
+import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
+import static org.testng.Assert.assertEquals;
+
+import java.io.IOException;
+import java.net.HttpURLConnection;
+import java.net.URI;
+import java.util.ArrayList;
+import java.util.List;
+
 
 @Test
 public class RestAdminITCase {
 
-    @Test
-    public void createNewUser() {
-        String uri = "http://localhost:9000/service/admin/user";
+    @NonNls
+    private static final String HOST_PORT = "http://localhost:8080/";
+    private static final String BASE_REST_URL = HOST_PORT + "service";
 
-//        RestTemplate template = new RestTemplate();
-//        location = template.postForLocation(uri);
+    @Test(dataProvider = "ContentType-Provider-Function")
+    public void findUser(final @NotNull MediaType mediaType) {    // Configure media types ...
+        final HttpHeaders requestHeaders = createHeaders(mediaType);
+        final RestTemplate templateRest = createTemplate();
+
+        HttpEntity<RestUser> findUserEntity = new HttpEntity<RestUser>(requestHeaders);
+        final ResponseEntity<RestUser> result = templateRest.exchange(BASE_REST_URL + "/admin/users/2", HttpMethod.GET, findUserEntity, RestUser.class);
+    }
+
+    @Test(dataProvider = "ContentType-Provider-Function")
+    public void createNewUser(final @NotNull MediaType mediaType) {
+
+        // Configure media types ...
+        final HttpHeaders requestHeaders = createHeaders(mediaType);
+
+        // Configure media ...
+        final RestTemplate templateRest = createTemplate();
+
+        // Fill user data ...
+        final RestUser restUser = new RestUser();
+
+        restUser.setEmail("foo" + System.nanoTime() + "@example.org");
+        restUser.setUsername("foo");
+        restUser.setFirstname("foo first name");
+        restUser.setLastname("foo last name");
+        restUser.setPassword("foo password");
+
+        // Post request ...
+        HttpEntity<RestUser> createUserEntity = new HttpEntity<RestUser>(restUser, requestHeaders);
+        URI location = templateRest.postForLocation(BASE_REST_URL + "/admin/users", createUserEntity);
+        System.out.println("location:" + location);
+
+        // Check that the user has been created ...
+        HttpEntity<RestUser> findUserEntity = new HttpEntity<RestUser>(requestHeaders);
+        final String url = "http://localhost:8080" + location;
+        final ResponseEntity<RestUser> result = templateRest.exchange(url, HttpMethod.GET, findUserEntity, RestUser.class);
+        assertEquals(result.getBody(), restUser, "Returned object object seems not be the same.");
+    }
+
+    private HttpHeaders createHeaders(MediaType mediaType) {
+        List<MediaType> acceptableMediaTypes = new ArrayList<MediaType>();
+        acceptableMediaTypes.add(mediaType);
+        final HttpHeaders requestHeaders = new HttpHeaders();
+        requestHeaders.setAccept(acceptableMediaTypes);
+        requestHeaders.setContentType(mediaType);
+        return requestHeaders;
+    }
+
+    private RestTemplate createTemplate() {
+        SimpleClientHttpRequestFactory s = new SimpleClientHttpRequestFactory() {
+            @Override
+            protected void prepareConnection(HttpURLConnection connection, String httpMethod) throws IOException {
+                super.prepareConnection(connection, httpMethod);
+
+                //Basic Authentication for Police API
+                String authorisation = "admin@wisemapping.org" + ":" + "admin";
+                byte[] encodedAuthorisation = Base64.encode(authorisation.getBytes());
+                connection.setRequestProperty("Authorization", "Basic " + new String(encodedAuthorisation));
+            }
+
+        };
+        return new RestTemplate(s);
+    }
+
+    @DataProvider(name = "ContentType-Provider-Function")
+    public Object[][] contentTypes() {
+        return new Object[][]{{MediaType.APPLICATION_XML}, {MediaType.APPLICATION_JSON}};
     }
 }