inicjalizacja

src/main/java/application/utils/Initializer.java

@@ -1,70 +1,45 @@
 package application.utils;
 
-import application.model.Role;
-import application.model.User;
+import application.model.*;
+import application.repositories.DoorRepository;
 import application.repositories.RoleRepository;
 import application.repositories.UserRepository;
-import org.springframework.beans.factory.annotation.Autowired;
 
-import java.util.ArrayList;
+import java.util.HashSet;
+import java.util.Set;
 
 /**
  * Created by lukasgol on 30.04.14.
  */
 public class Initializer {
-    @Autowired
-    private UserRepository userRepository;
-    @Autowired
-    private RoleRepository roleRepository;
 
+    RoleRepository roleRepository;
+    UserRepository userRepository;
+    DoorRepository doorRepository;
+
+    public Initializer(RoleRepository roleRepository, UserRepository useRoleRepository) {
+        this.roleRepository = roleRepository;
+        this.userRepository = useRoleRepository;
+    }
+
+    public Initializer(RoleRepository roleRepository, UserRepository userRepository, DoorRepository doorRepository) {
+        this.roleRepository = roleRepository;
+        this.userRepository = userRepository;
+        this.doorRepository = doorRepository;
+    }
 
     public void init(){
-        Role ceo = new Role("CEO","101010");
-        Role worker = new Role("Worker","12345");
-        roleRepository.save(new Role("Cleaner","54321"));
-        ceo = roleRepository.save(ceo);
 
+        User marcin = new User("Marcin", "Michalek", "m", "p");
+        Set<Role> roles = new HashSet<Role>();
+        roles.add(new Role("CLEANER", "shityJob"));
+        roles.add(new Role("BOSS", "fuckYea"));
+        marcin.setRoles(roles);
+        marcin.addMessage(new Message("hello"));
+        marcin.addMessage(new Message("You're fired"));
+        userRepository.save(marcin);
 
-        User user1 = new User("Łukasz", "Goliński", "boss@nfc.pl", "pass");
-        user1.addRole(worker);
-        user1.addRole(ceo);
-        userRepository.save(user1);
-
-
-        ArrayList<Role> roles = (ArrayList<Role>) roleRepository.findAll();
-
-        for(Role r: roles){
-            System.out.println(r.getRoleName());
-        }
-
-        Role r = roles.get(0);
-
-        User user3 = new User("Monika", "Bigas", "boss_monika@nfc.pl", "pass");
-        user3.addRole(worker);
-        user3.addRole(ceo);
-        userRepository.save(user3);
-        //entityManager.merge(user3);
-
-        User user2 = new User("Marcin", "Michałek", "boss_marcin@nfc.pl", "pass");
-        user2.addRole(ceo);
-        userRepository.save(user2);
-
-
-
-
-        ArrayList<User> users = (ArrayList<User>) userRepository.findAll();
-        for(User u: users){
-            System.out.println(u.getFirstName()+" "+u.getLastName());
-        }
-
-        for (User user : userRepository.findAll()) {
-            System.out.println("USER: " + user.getFirstName() + " " + user.getLastName() + " " + user.getEmail());
-            System.out.println("roles: ");
-            for (Role role : user.getRoles()) {
-                System.out.println("    -" + role.getRoleName());
-            }
-            System.out.println("enduser");
-            System.out.println("");
-        }
+        DoorMessage message=new DoorMessage("Be Careful");
+        doorRepository.save(new Door(message));
     }
 }