for (long i = beginNumber; i <= endNumber; i++) {
final BlankPolicy blankPolicy = new BlankPolicy(new PolicyId(policySeries, i));
blankPolicyDAO.persistPolicy(blankPolicy);
}
final Employee firstEmployee = new Employee(new Passport("АВ", "468454"), "Иванов", "Иван");
final Account firstAgent = new Account(firstEmployee);
final Set<UserRoles> firstAgentUserRoles = new TreeSet<UserRoles>();
firstAgentUserRoles.add(UserRoles.ROLE_AGENT);
firstAgent.setAssociatedRoles(firstAgentUserRoles);
userDAO.persist(firstAgent);
final Employee secondEmployee = new Employee(new Passport("АУ", "468454"), "Иванов", "Иван");
final Account secondAgent = new Account(secondEmployee);
final Set<UserRoles> secondAgentUserRoles = new TreeSet<UserRoles>();
secondAgentUserRoles.add(UserRoles.ROLE_AGENT);
secondAgent.setAssociatedRoles(secondAgentUserRoles);
userDAO.persist(secondAgent);
final List<BlankPolicy> policiesOne = blankPolicyDAO.findBlankPolicies(policySeries, 1, 5);
final List<BlankPolicy> policiesTwo = blankPolicyDAO.findBlankPolicies(policySeries, 6, 10);
for (BlankPolicy policy : policiesOne) {
policy.setAgent(firstEmployee);
policy.setIssueToAgentDate(new Date());
policy.setIssuedToAgent(true);
blankPolicyDAO.updateBlankPolicy(policy);
}
for (BlankPolicy policy : policiesTwo) {
policy.setAgent(secondEmployee);
policy.setIssueToAgentDate(new Date());
policy.setIssuedToAgent(true);
blankPolicyDAO.updateBlankPolicy(policy);
}
final List<Employee> policiesAgents = blankPolicyDAO.findAgentsOfPolicies(policySeries, 1, 10);
assertNotNull(policiesAgents);
assertEquals(2, policiesAgents.size());
assertEquals(firstEmployee.getFirstName(), policiesAgents.get(0).getFirstName());
assertEquals(secondEmployee.getFirstName(), policiesAgents.get(1).getFirstName());
}