Datainitklasser

git-svn-id: svn://svn.dsv.su.se/scipro/scipro/trunk@95 73ecded7-942e-4092-bab0-0e58ef0ee984

src/main/java/se/su/dsv/scipro/datainitializers/EventDataInitializer.java

@@ -0,0 +1,218 @@
+package se.su.dsv.scipro.datainitializers;
+
+import java.util.Date;
+import java.util.Set;
+import java.util.TreeSet;
+import se.su.dsv.scipro.data.dao.interfaces.EventDao;
+import se.su.dsv.scipro.data.dao.interfaces.GroupEventDao;
+import se.su.dsv.scipro.data.dao.interfaces.ProjectEventDao;
+import se.su.dsv.scipro.data.dao.interfaces.ProjectDao;
+import se.su.dsv.scipro.data.dao.interfaces.ProjectScheduleDao;
+import se.su.dsv.scipro.data.dao.interfaces.RoleDao;
+import se.su.dsv.scipro.data.dao.interfaces.UserDao;
+import se.su.dsv.scipro.data.dataobjects.Employee;
+import se.su.dsv.scipro.data.dataobjects.GroupEvent;
+import se.su.dsv.scipro.data.dataobjects.ProjectEvent;
+import se.su.dsv.scipro.data.dataobjects.Project;
+import se.su.dsv.scipro.data.dataobjects.ProjectSchedule;
+import se.su.dsv.scipro.data.dataobjects.Student;
+import se.su.dsv.scipro.data.dataobjects.User;
+
+/**
+ * @author Richard Wilkinson - richard.wilkinson@jweekend.com
+ *
+ */
+public class EventDataInitializer {
+	
+	private EventDao eventDao;
+	private ProjectEventDao projectEventDao;	
+	private GroupEventDao groupEventDao;	
+	private ProjectScheduleDao projectScheduleDao;		
+	private ProjectDao projectDao;	
+	private RoleDao roleDao;	
+	private UserDao userDao;
+	
+	
+	private User indEventUser;
+	private User groupEventCreatorUser;
+	
+	private User projectUser1;
+	private User projectUser2;
+	private Project project1;
+	private Project project2;
+
+	protected ProjectEvent indEvent;
+	protected ProjectEvent deletedIndEvent;
+	
+	private GroupEvent groupEvent;
+	private GroupEvent deletedGroupEvent;
+
+	
+	public void dataInit() {
+		indEventUser = new User();
+		indEventUser.setFirstName("indEventUser");
+		indEventUser = userDao.save(indEventUser);
+		
+		Student indEventUserRole = new Student();
+		indEventUserRole.setUser(indEventUser);
+		indEventUserRole = (Student) roleDao.save(indEventUserRole);
+		
+		groupEventCreatorUser = new User();
+		groupEventCreatorUser.setFirstName("eventCreatorUser");
+		groupEventCreatorUser = userDao.save(groupEventCreatorUser);
+		
+		Employee groupEventCreatorUserRole = new Employee();
+		groupEventCreatorUserRole.setUser(groupEventCreatorUser);
+		groupEventCreatorUserRole = (Employee) roleDao.save(groupEventCreatorUserRole);
+		
+		projectUser1 = new User();
+		projectUser1.setFirstName("projectUser1");
+		projectUser1 = userDao.save(projectUser1);
+		
+		Student projectUser1Role = new Student();
+		projectUser1Role.setUser(projectUser1);
+		projectUser1Role = (Student) roleDao.save(projectUser1Role);
+		
+		projectUser2 = new User();
+		projectUser2.setFirstName("projectUser2");
+		projectUser2 = userDao.save(projectUser2);
+		
+		Student projectUser2Role = new Student();
+		projectUser2Role.setUser(projectUser2);
+		projectUser2Role = (Student) roleDao.save(projectUser2Role);
+		
+		project1 = new Project();
+		project1.setTitle("Project 1");
+		project1.addProjectParticipant(indEventUserRole);
+		project1.addProjectParticipant(projectUser1Role);
+		project1 = projectDao.save(project1);
+		
+		ProjectSchedule schedule1 = new ProjectSchedule();
+		schedule1.setProject(project1);
+		schedule1 = projectScheduleDao.save(schedule1);
+		
+		project2 = new Project();
+		project2.setTitle("Project 2");
+		project2.addProjectParticipant(indEventUserRole);
+		project2.addProjectParticipant(projectUser2Role);	
+		project2 = projectDao.save(project2);
+		
+		ProjectSchedule schedule2 = new ProjectSchedule();
+		schedule2.setProject(project2);
+		schedule2 = projectScheduleDao.save(schedule2);
+		
+		indEvent = new ProjectEvent();
+		indEvent.setTitle("Individual event title");
+		indEvent.setDescription("Individual Event Description");
+		indEvent.setDueDate(new Date(System.currentTimeMillis() + 100000));
+		Set<Student> indEventStudents = new TreeSet<Student>();
+		indEventStudents.add(indEventUserRole);
+		indEvent.setParticipants(indEventStudents);
+		indEvent.setProjectSchedule(schedule1);
+		indEvent.setCreator(groupEventCreatorUserRole);
+		indEvent = projectEventDao.save(indEvent);
+		
+		deletedIndEvent = new ProjectEvent();
+		deletedIndEvent.setTitle("Deleted Individual Event title");
+		deletedIndEvent.setDescription("Deleted Individual Event description");
+		deletedIndEvent.setDueDate(new Date(System.currentTimeMillis() + 100000));
+		deletedIndEvent.setDeleted(true);
+		Set<Student> indDeletedEventStudents = new TreeSet<Student>();
+		indDeletedEventStudents.add(indEventUserRole);
+		indEvent.setParticipants(indDeletedEventStudents);
+		deletedIndEvent.setProjectSchedule(schedule1);
+		deletedIndEvent.setCreator(groupEventCreatorUserRole);
+		deletedIndEvent = projectEventDao.save(deletedIndEvent);
+		
+		groupEvent = new GroupEvent();
+		groupEvent.setTitle("Group event title");
+		groupEvent.setDescription("Group Event Description");
+		groupEvent.setDueDate(new Date(System.currentTimeMillis() + 5000000));
+		groupEvent.addProjectSchedule(schedule1);
+		groupEvent.addProjectSchedule(schedule2);
+		groupEvent.setCreator(groupEventCreatorUserRole);
+		groupEvent = groupEventDao.save(groupEvent);
+		
+		deletedGroupEvent = new GroupEvent();
+		deletedGroupEvent.setTitle("Deleted group Event title");
+		deletedGroupEvent.setDescription("Deleted group Event description");
+		deletedGroupEvent.setDueDate(new Date(System.currentTimeMillis() + 100000));
+		deletedGroupEvent.setDeleted(true);
+		deletedGroupEvent.addProjectSchedule(schedule1);
+		deletedGroupEvent.addProjectSchedule(schedule2);
+		deletedGroupEvent.setCreator(groupEventCreatorUserRole);
+		deletedGroupEvent = groupEventDao.save(deletedGroupEvent);
+	}
+
+
+	public EventDao getEventDao() {
+		return eventDao;
+	}
+
+
+	public void setEventDao(EventDao eventDao) {
+		this.eventDao = eventDao;
+	}
+
+
+	public GroupEventDao getGroupEventDao() {
+		return groupEventDao;
+	}
+
+
+	public void setGroupEventDao(GroupEventDao groupEventDao) {
+		this.groupEventDao = groupEventDao;
+	}
+
+
+	public ProjectScheduleDao getProjectScheduleDao() {
+		return projectScheduleDao;
+	}
+
+
+	public void setProjectScheduleDao(ProjectScheduleDao projectScheduleDao) {
+		this.projectScheduleDao = projectScheduleDao;
+	}
+
+
+	public ProjectDao getProjectDao() {
+		return projectDao;
+	}
+
+
+	public void setProjectDao(ProjectDao projectDao) {
+		this.projectDao = projectDao;
+	}
+
+
+	public RoleDao getRoleDao() {
+		return roleDao;
+	}
+
+
+	public void setRoleDao(RoleDao roleDao) {
+		this.roleDao = roleDao;
+	}
+
+
+	public UserDao getUserDao() {
+		return userDao;
+	}
+
+
+	public void setUserDao(UserDao userDao) {
+		this.userDao = userDao;
+	}
+
+
+	public ProjectEventDao getProjectEventDao() {
+		return projectEventDao;
+	}
+
+
+	public void setProjectEventDao(ProjectEventDao projectEventDao) {
+		this.projectEventDao = projectEventDao;
+	}
+
+}
+