models, repos, test (#1225)

* models, repos, test * update models and test * updated test location * update test location
2017-02-26 21:30:47 +02:00
parent 7f20437cb0
commit 1a067bc945
9 changed files with 394 additions and 0 deletions
@@ -0,0 +1,103 @@
+package com.baeldung.relationships;
+
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
+import org.springframework.boot.test.web.client.TestRestTemplate;
+import org.springframework.http.HttpEntity;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpMethod;
+import org.springframework.http.ResponseEntity;
+import org.springframework.test.context.junit4.SpringRunner;
+
+import com.baeldung.SpringDataRestApplication;
+import com.baeldung.models.Address;
+import com.baeldung.models.Author;
+import com.baeldung.models.Book;
+import com.baeldung.models.Library;
+
+import org.junit.Test;
+import static org.junit.Assert.*;
+
+import org.json.JSONArray;
+import org.json.JSONObject;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(classes = SpringDataRestApplication.class, webEnvironment = WebEnvironment.DEFINED_PORT)
+public class SpringDataRelationshipsTest {
+
+    @Autowired
+    private TestRestTemplate template;
+
+    private static final String BOOK_ENDPOINT = "http://localhost:8080/books/";
+    private static final String AUTHOR_ENDPOINT = "http://localhost:8080/authors/";
+    private static final String ADDRESS_ENDPOINT = "http://localhost:8080/addresses/";
+    private static final String LIBRARY_ENDPOINT = "http://localhost:8080/libraries/";
+
+    private static final String LIBRARY_NAME = "My Library";
+    private static final String AUTHOR_NAME = "George Orwell";
+
+    @Test
+    public void whenSaveOneToOneRelationship_thenCorrect() {
+        Library library = new Library(LIBRARY_NAME);
+        template.postForEntity(LIBRARY_ENDPOINT, library, Library.class);
+
+        Address address = new Address("Main street, nr 1");
+        template.postForEntity(ADDRESS_ENDPOINT, address, Address.class);
+
+        HttpHeaders requestHeaders = new HttpHeaders();
+        requestHeaders.add("Content-type", "text/uri-list");
+        HttpEntity<String> httpEntity = new HttpEntity<String>(ADDRESS_ENDPOINT + "/1", requestHeaders);
+        template.exchange(LIBRARY_ENDPOINT + "/1/libraryAddress", HttpMethod.PUT, httpEntity, String.class);
+
+        ResponseEntity<Library> libraryGetResponse = template.getForEntity(ADDRESS_ENDPOINT + "/1/library", Library.class);
+        assertEquals("library is incorrect", libraryGetResponse.getBody()
+            .getName(), LIBRARY_NAME);
+    }
+
+    @Test
+    public void whenSaveOneToManyRelationship_thenCorrect() {
+        Library library = new Library(LIBRARY_NAME);
+        template.postForEntity(LIBRARY_ENDPOINT, library, Library.class);
+
+        Book book1 = new Book("Dune");
+        template.postForEntity(BOOK_ENDPOINT, book1, Book.class);
+
+        Book book2 = new Book("1984");
+        template.postForEntity(BOOK_ENDPOINT, book2, Book.class);
+
+        HttpHeaders requestHeaders = new HttpHeaders();
+        requestHeaders.add("Content-type", "text/uri-list");
+        HttpEntity<String> bookHttpEntity = new HttpEntity<String>(LIBRARY_ENDPOINT + "/1", requestHeaders);
+        template.exchange(BOOK_ENDPOINT + "/1/library", HttpMethod.PUT, bookHttpEntity, String.class);
+        template.exchange(BOOK_ENDPOINT + "/2/library", HttpMethod.PUT, bookHttpEntity, String.class);
+
+        ResponseEntity<Library> libraryGetResponse = template.getForEntity(BOOK_ENDPOINT + "/1/library", Library.class);
+        assertEquals("library is incorrect", libraryGetResponse.getBody()
+            .getName(), LIBRARY_NAME);
+    }
+
+    @Test
+    public void whenSaveManyToManyRelationship_thenCorrect() {
+        Author author1 = new Author(AUTHOR_NAME);
+        template.postForEntity(AUTHOR_ENDPOINT, author1, Author.class);
+
+        Book book1 = new Book("Animal Farm");
+        template.postForEntity(BOOK_ENDPOINT, book1, Book.class);
+
+        Book book2 = new Book("1984");
+        template.postForEntity(BOOK_ENDPOINT, book2, Book.class);
+
+        HttpHeaders requestHeaders = new HttpHeaders();
+        requestHeaders.add("Content-type", "text/uri-list");
+        HttpEntity<String> httpEntity = new HttpEntity<String>(BOOK_ENDPOINT + "/1\n" + BOOK_ENDPOINT + "/2", requestHeaders);
+        template.exchange(AUTHOR_ENDPOINT + "/1/books", HttpMethod.PUT, httpEntity, String.class);
+
+        String jsonResponse = template.getForObject(BOOK_ENDPOINT + "/1/authors", String.class);
+        JSONObject jsonObj = new JSONObject(jsonResponse).getJSONObject("_embedded");
+        JSONArray jsonArray = jsonObj.getJSONArray("authors");
+        assertEquals("author is incorrect", jsonArray.getJSONObject(0)
+            .getString("name"), AUTHOR_NAME);
+    }
+}