Add temp solution to have more test data

Author: dvelopp

src/main/java/springAngularApp/system/ws/InitDBController.java

@@ -1,5 +1,6 @@
 package springAngularApp.system.ws;
 
+import org.hibernate.SessionFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.ResponseEntity;
 import org.springframework.security.crypto.password.PasswordEncoder;
@@ -13,27 +14,36 @@
 import springAngularApp.users.domain.repositories.UserGroupRepository;
 import springAngularApp.users.domain.repositories.UserRepository;
 
+import java.io.File;
+import java.io.IOException;
+
+import static com.google.common.collect.Lists.newArrayList;
+import static com.google.common.io.Files.readLines;
+import static java.nio.charset.Charset.defaultCharset;
 import static org.springframework.http.HttpStatus.OK;
 import static springAngularApp.users.domain.entities.UserAuthorities.*;
-import static springAngularApp.users.domain.entities.UserAuthorities.ROLE_USER_GROUP_VIEW;
 
 /**
  * TODO Delete me
  */
 @RestController
+@SuppressWarnings("all")
 public class InitDBController {
 
     @Autowired private UserAuthorityRepository userAuthorityRepository;
     @Autowired private UserGroupRepository userGroupRepository;
     @Autowired private UserRepository userRepository;
     @Autowired private PasswordEncoder passwordEncoder;
+    @Autowired private SessionFactory sessionFactory;
 
     @Transactional
     @RequestMapping(value = "/initDB")
-    public ResponseEntity init() {
-        userRepository.deleteAll();
-        userGroupRepository.deleteAll();
-        userAuthorityRepository.deleteAll();
+    public ResponseEntity init() throws IOException {
+        sessionFactory.getCurrentSession().createQuery("DELETE FROM User").executeUpdate();
+        sessionFactory.getCurrentSession().createQuery("DELETE FROM UserGroup").executeUpdate();
+        sessionFactory.getCurrentSession().createQuery("DELETE FROM UserAuthority").executeUpdate();
+        sessionFactory.getCurrentSession().flush();
+        sessionFactory.getCurrentSession().clear();
 
         UserGroup adminGroup = new UserGroup("Administrator");
         adminGroup.setSuperUserGroup(true);
@@ -65,23 +75,32 @@ public ResponseEntity init() {
         admin.setFirstName("Bob");
         admin.setLastName("Smith");
 
-        User user = new User("user", password, defaultUserGroup);
-        admin.setFirstName("Adam");
-        admin.setLastName("Hansen");
-
-        userAuthorityRepository.save(roleDefault);
-        userAuthorityRepository.save(roleUserEdit);
-        userAuthorityRepository.save(roleUserView);
-        userAuthorityRepository.save(roleUserDelete);
-        userAuthorityRepository.save(roleConfigurationView);
-        userAuthorityRepository.save(roleUserGroupEdit);
-        userAuthorityRepository.save(roleUserGroupView);
-        userAuthorityRepository.save(roleUserGroupDelete);
-        userGroupRepository.save(adminGroup);
-        userGroupRepository.save(defaultUserGroup);
+        userAuthorityRepository.save(
+                newArrayList(
+                        roleDefault,
+                        roleUserEdit,
+                        roleUserView,
+                        roleUserDelete,
+                        roleConfigurationView,
+                        roleUserGroupEdit,
+                        roleUserGroupView,
+                        roleUserGroupDelete
+                ));
+        userGroupRepository.save(newArrayList(adminGroup, defaultUserGroup));
         userRepository.save(admin);
-        userRepository.save(user);
+        generateAdditionalUsers(adminGroup, password);
         return new ResponseEntity(OK);
     }
 
+    private void generateAdditionalUsers(UserGroup adminGroup, String password) throws IOException {
+        File fileWithNames = new File(getClass().getClassLoader().getResource("utils/names.txt").getFile());
+        readLines(fileWithNames, defaultCharset()).forEach(name -> {
+            String[] nameParts = name.split(" ");
+            User user = new User(nameParts[0] + nameParts[1], password, adminGroup);
+            user.setFirstName(nameParts[0]);
+            user.setLastName(nameParts[1]);
+            userRepository.save(user);
+        });
+    }
+
 }

src/main/java/springAngularApp/users/domain/entities/UserGroup.java

@@ -54,8 +54,9 @@ public boolean isSuperUserGroup() {
         return superUserGroup;
     }
 
-    public void setSuperUserGroup(boolean superUserGroup) {
+    public UserGroup setSuperUserGroup(boolean superUserGroup) {
         this.superUserGroup = superUserGroup;
+        return this;
     }
 
     public UserGroup(String name, List<UserAuthority> authorities) {

src/main/resources/utils/names.txt

@@ -0,0 +1,20 @@
+Gaylene Furlong
+Georgina Ruggerio
+Delmy Hamburger
+Tam Vanduzer
+Savanna Rector
+Jacalyn Binford
+Madie Gilbert
+Kip Buckholz
+Tressa Raley
+Aisha Petties
+Magali Breeze
+Sabina Bronson
+Mitchel Candy
+Arianne Culbreath
+Frida Fordyce
+Erica Anson
+Livia Raposa
+Hermine Arnot
+Kenton Janelle
+Lamonica Mcraney

src/test/java/springAngularApp/system/ws/InitDBControllerTest.java

@@ -1,5 +1,6 @@
 package springAngularApp.system.ws;
 
+import org.hibernate.SessionFactory;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -14,6 +15,7 @@
 import springAngularApp.users.domain.repositories.UserRepository;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.Answers.RETURNS_DEEP_STUBS;
 import static org.mockito.Matchers.anyString;
 import static org.mockito.Mockito.when;
 import static org.springframework.http.HttpStatus.OK;
@@ -27,6 +29,7 @@ public class InitDBControllerTest {
     @Mock private UserGroupRepository userGroupRepository;
     @Mock private UserRepository userRepository;
     @Mock private PasswordEncoder passwordEncoder;
+    @Mock(answer = RETURNS_DEEP_STUBS) private SessionFactory sessionFactory;
 
     private MockMvc mockMvc;