1 files changed, 34 insertions, 18 deletions
diff --git a/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java b/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java
index a1ab349..f96e129 100644
--- a/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java
+++ b/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java
@@ -11,8 +11,6 @@ import org.springframework.context.annotation.Import;
 import org.springframework.security.test.context.support.WithMockUser;
 import org.springframework.test.web.servlet.MockMvc;
 
-import java.util.Date;
-
 import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.csrf;
 import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.user;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
@@ -60,15 +58,9 @@ class BuilddbAccountControllerTest {
     @Test
     void shouldReturnProfileView() throws Exception {
 
-        AppUser appUser = new AppUser();
-        appUser.setFirstName("John");
-        appUser.setLastName("Doe");
-        appUser.setEmail("test@berzerkula.org");
-        appUser.setPhone("555-555-5555");
-        appUser.setAddress("612 Wolf Avenue");
-        appUser.setRole("client");
-        appUser.setPassword("password");
-        appUser.setCreatedAt(new Date());
+        appUserRepository.deleteAll();
+        AppUser appUser = new AppUser("John", "Doe", "test@berzerkula.org",
+                "555-555-5555", "612 Wolf Avenue", "password", "client");
         appUserRepository.save(appUser);
 
         this.mockMvc.perform(get("/profile")
@@ -120,15 +112,39 @@ class BuilddbAccountControllerTest {
 
     @Test
     void shouldRegisterSuccess() throws Exception {
+        appUserRepository.deleteAll();
+
+        mockMvc.perform(post("/register")
+                        .with(csrf())
+                        .formField("firstName", "John")
+                        .formField("lastName", "Doe")
+                        .formField("email", "test@berzerkula.org")
+                        .formField("phone", "555-555-5555")
+                        .formField("address", "612 Wolf Avenue")
+                        .formField("password", "password")
+                        .formField("confirmPassword", "password"))
+                .andExpect(status().is3xxRedirection())
+                .andExpect(view().name("redirect:/login?registerSuccess"))
+                .andDo(print());
+    }
+
+    @Test
+    void shouldRegisterEmailAlreadyUsed() throws Exception {
+
+        appUserRepository.deleteAll();
+        AppUser appUser = new AppUser("John", "Doe", "test@berzerkula.org",
+                "555-555-5555", "612 Wolf Avenue", "password", "client");
+        appUserRepository.save(appUser);
+
         mockMvc.perform(post("/register")
                         .with(csrf())
-                .formField("firstName", "John")
-                .formField("lastName", "Doe")
-                .formField("email", "test@berzerkula.org")
-                .formField("phone", "555-555-5555")
-                .formField("address", "612 Wolf Avenue")
-                .formField("password", "password")
-                .formField("confirmPassword", "password"))
+                        .formField("firstName", "John")
+                        .formField("lastName", "Doe")
+                        .formField("email", "test@berzerkula.org")
+                        .formField("phone", "555-555-5555")
+                        .formField("address", "612 Wolf Avenue")
+                        .formField("password", "password")
+                        .formField("confirmPassword", "password"))
                 .andExpect(status().isOk())
                 .andExpect(view().name("register"))
                 .andDo(print());