private Assessment createAssessment(Group g, int questionCount, int reinforcementCount) { Assessment a = new Assessment(); a.setGroup(g); List<AssessmentMessage> messages = new ArrayList<AssessmentMessage>(); for (int i = 0; i < Math.max(questionCount, reinforcementCount); ++i) { if (i < questionCount) { Question q = new Question(); q.setAnswers(new String[3]); questionDao.save(q); messages.add(new AssessmentMessage(q)); } if (i < reinforcementCount) { Reinforcement r = new Reinforcement(); reinforcementDao.save(r); messages.add(new AssessmentMessage(r)); } } a.setMessages(messages); assessmentDao.save(a); return a; }