1 files changed, 4 insertions, 3 deletions
diff --git a/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java b/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java
index f96e129..7631980 100644
--- a/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java
+++ b/src/test/java/org/berzerkula/builddb/controllers/BuilddbAccountControllerTest.java
@@ -1,5 +1,6 @@
 package org.berzerkula.builddb.controllers;
 
+import org.berzerkula.builddb.BuilddbConstants;
 import org.berzerkula.builddb.config.SecurityConfig;
 import org.berzerkula.builddb.models.AppUser;
 import org.berzerkula.builddb.repositories.AppUserRepository;
@@ -54,13 +55,13 @@ class BuilddbAccountControllerTest {
                 .andDo(print());
     }
 
-    @WithMockUser(roles="client")
+    @WithMockUser(roles=BuilddbConstants.ROLE_CLIENT)
     @Test
     void shouldReturnProfileView() throws Exception {
 
         appUserRepository.deleteAll();
         AppUser appUser = new AppUser("John", "Doe", "test@berzerkula.org",
-                "555-555-5555", "612 Wolf Avenue", "password", "client");
+                "555-555-5555", "612 Wolf Avenue", "password", BuilddbConstants.ROLE_CLIENT);
         appUserRepository.save(appUser);
 
         this.mockMvc.perform(get("/profile")
@@ -133,7 +134,7 @@ class BuilddbAccountControllerTest {
 
         appUserRepository.deleteAll();
         AppUser appUser = new AppUser("John", "Doe", "test@berzerkula.org",
-                "555-555-5555", "612 Wolf Avenue", "password", "client");
+                "555-555-5555", "612 Wolf Avenue", "password", BuilddbConstants.ROLE_CLIENT);
         appUserRepository.save(appUser);
 
         mockMvc.perform(post("/register")