[BAEL-14089] - Moved code for Entity to DTO article

spring-boot-rest/src/test/java/com/baeldung/SpringContextIntegrationTest.java

@@ -6,7 +6,7 @@ import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.test.context.junit4.SpringRunner;
 
 @RunWith(SpringRunner.class)
-@SpringBootTest
+@SpringBootTest(classes = {SpringBootRestApplication.class})
 public class SpringContextIntegrationTest {
 
     @Test

spring-boot-rest/src/test/java/com/baeldung/modelmapper/PostDtoUnitTest.java

@@ -0,0 +1,40 @@
+package com.baeldung.modelmapper;
+
+import static org.junit.Assert.assertEquals;
+import static org.apache.commons.lang3.RandomStringUtils.randomAlphabetic;
+import org.junit.Test;
+import org.modelmapper.ModelMapper;
+
+import com.baeldung.modelmapper.dto.PostDto;
+import com.baeldung.modelmapper.model.Post;
+
+public class PostDtoUnitTest {
+    
+    private ModelMapper modelMapper = new ModelMapper();
+    
+    @Test
+    public void whenConvertPostEntityToPostDto_thenCorrect() {
+        Post post = new Post();
+        post.setId(Long.valueOf(1));
+        post.setTitle(randomAlphabetic(6));
+        post.setUrl("www.test.com");
+ 
+        PostDto postDto = modelMapper.map(post, PostDto.class);
+        assertEquals(post.getId(), postDto.getId());
+        assertEquals(post.getTitle(), postDto.getTitle());
+        assertEquals(post.getUrl(), postDto.getUrl());
+    }
+ 
+    @Test
+    public void whenConvertPostDtoToPostEntity_thenCorrect() {
+        PostDto postDto = new PostDto();
+        postDto.setId(Long.valueOf(1));
+        postDto.setTitle(randomAlphabetic(6));
+        postDto.setUrl("www.test.com");
+ 
+        Post post = modelMapper.map(postDto, Post.class);
+        assertEquals(postDto.getId(), post.getId());
+        assertEquals(postDto.getTitle(), post.getTitle());
+        assertEquals(postDto.getUrl(), post.getUrl());
+    }
+}