TimesheetBookingTest.java

/////////////////////////////////////////////////////////////////////////////
//
// Project ProjectForge Community Edition
//         www.projectforge.org
//
// Copyright (C) 2001-2014 Kai Reinhard (k.reinhard@micromata.de)
//
// ProjectForge is dual-licensed.
//
// This community edition is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License as published
// by the Free Software Foundation; version 3 of the License.
//
// This community edition is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
// Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, see http://www.gnu.org/licenses/.
//
/////////////////////////////////////////////////////////////////////////////

package org.projectforge.task;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import java.util.Calendar;

import org.junit.Test;
import org.projectforge.access.AccessDao;
import org.projectforge.access.AccessEntryDO;
import org.projectforge.access.AccessException;
import org.projectforge.access.AccessType;
import org.projectforge.access.GroupTaskAccessDO;
import org.projectforge.common.DateHolder;
import org.projectforge.common.DatePrecision;
import org.projectforge.fibu.AuftragDO;
import org.projectforge.fibu.AuftragDao;
import org.projectforge.fibu.AuftragsPositionDO;
import org.projectforge.test.TestBase;
import org.projectforge.timesheet.TimesheetDO;
import org.projectforge.timesheet.TimesheetDao;
import org.projectforge.user.PFUserDO;


public class TimesheetBookingTest extends TestBase
{
  private TaskDao taskDao;

  private AccessDao accessDao;

  private TimesheetDao timesheetDao;

  private AuftragDao auftragDao;

  static final DateHolder date = new DateHolder(DatePrecision.MINUTE_15);

  static boolean initialized = false;

  public void setAccessDao(final AccessDao accessDao)
  {
    this.accessDao = accessDao;
  }

  public void setTaskDao(final TaskDao taskDao)
  {
    this.taskDao = taskDao;
  }

  public void setTimesheetDao(final TimesheetDao timesheetDao)
  {
    this.timesheetDao = timesheetDao;
  }

  public void setAuftragDao(final AuftragDao auftragDao)
  {
    this.auftragDao = auftragDao;
  }

  private synchronized void initialize() // @BeforeClass not possible because DAOs are needed.
  {
    if (initialized == true) {
      return;
    }
    initialized = true;
    logon(getUser(TEST_ADMIN_USER));
    TaskDO task;
    task = initTestDB.addTask("TimesheetBookingTest", "root");
    final GroupTaskAccessDO access = new GroupTaskAccessDO().setGroup(initTestDB.addGroup("TBT", TEST_USER)).addAccessEntry(
        new AccessEntryDO(AccessType.OWN_TIMESHEETS, true, true, true, true)).setTask(task);
    accessDao.save(access);
    logon(getUser(TEST_FINANCE_USER));
    taskDao.update(initTestDB.addTask("TBT-1", "TimesheetBookingTest"));
    taskDao.update(initTestDB.addTask("TBT-1.1", "TBT-1").setStatus(TaskStatus.C));
    taskDao.markAsDeleted(initTestDB.addTask("TBT-1.2", "TBT-1"));
    taskDao.update(initTestDB.addTask("TBT-1.2.1", "TBT-1.2"));
    taskDao.update(initTestDB.addTask("TBT-2", "TimesheetBookingTest").setTimesheetBookingStatus(TimesheetBookingStatus.TREE_CLOSED));
    taskDao.update(initTestDB.addTask("TBT-2.1", "TBT-2").setTimesheetBookingStatus(TimesheetBookingStatus.OPENED));
    taskDao.update(initTestDB.addTask("TBT-3", "TimesheetBookingTest").setTimesheetBookingStatus(TimesheetBookingStatus.ONLY_LEAFS));
    initTestDB.addTask("TBT-3.1", "TBT-3");
    initTestDB.addTask("TBT-3.1.1", "TBT-3.1");
    initTestDB.addTask("TBT-3.1.2", "TBT-3.1");
    initTestDB.addTask("TBT-3.2", "TBT-3");
    taskDao.update(initTestDB.addTask("TBT-4", "TimesheetBookingTest").setTimesheetBookingStatus(TimesheetBookingStatus.NO_BOOKING));
    taskDao.update(initTestDB.addTask("TBT-4.1", "TBT-4").setTimesheetBookingStatus(TimesheetBookingStatus.INHERIT));
    taskDao.update(initTestDB.addTask("TBT-4.1.1", "TBT-4.1").setTimesheetBookingStatus(TimesheetBookingStatus.OPENED));
    initTestDB.addTask("TBT-5", "TimesheetBookingTest");
    initTestDB.addTask("TBT-5.1", "TBT-5");
    initTestDB.addTask("TBT-5.1.1", "TBT-5.1");
    initTestDB.addTask("TBT-5.1.2", "TBT-5.1");
    initTestDB.addTask("TBT-5.2", "TBT-5");
    initTestDB.addTask("TBT-5.2.1", "TBT-5.2");
    initTestDB.addTask("TBT-5.2.1.1", "TBT-5.2.1");
  }

  @Test
  public void testTimesheetBookingStatus()
  {
    initialize();
    logon(getUser(TEST_USER));
    TimesheetDO sheet = createNewSheet().setTask(getTask("TBT-2"));
    save(sheet, "timesheet.error.taskNotBookable.treeClosedForBooking");
    save(sheet.setTask(getTask("TBT-2.1")), "timesheet.error.taskNotBookable.treeClosedForBooking");
    save(sheet.setTask(getTask("TBT-3")), "timesheet.error.taskNotBookable.onlyLeafsAllowedForBooking");
    save(sheet.setTask(getTask("TBT-3.1")), "timesheet.error.taskNotBookable.onlyLeafsAllowedForBooking");
    timesheetDao.save(sheet.setTask(getTask("TBT-3.2"))); // Leaf task node.
    timesheetDao.save(createNewSheet().setTask(getTask("TBT-3.1.1"))); // Leaf task node.
    timesheetDao.save(createNewSheet().setTask(getTask("TBT-3.1.2"))); // Leaf task node.
    sheet = createNewSheet();
    save(sheet.setTask(getTask("TBT-4")), "timesheet.error.taskNotBookable.taskClosedForBooking");
    save(sheet.setTask(getTask("TBT-4.1")), "timesheet.error.taskNotBookable.taskClosedForBooking"); // Inherited (not opened)
    timesheetDao.save(sheet.setTask(getTask("TBT-4.1.1"))); // Opened for booking.
  }

  @Test
  public void testOrderPositions()
  {
    initialize();
    logon(getUser(TEST_FINANCE_USER));
    final AuftragDO auftrag = new AuftragDO().addPosition(new AuftragsPositionDO().setTask(getTask("TBT-5.1")).setTitel("Pos 1"))
        .addPosition(new AuftragsPositionDO().setTask(getTask("TBT-5.2.1.1")).setTitel("Pos 2"));
    auftragDao.save(auftrag.setNummer(auftragDao.getNextNumber(auftrag)));
    logon(getUser(TEST_USER));
    TimesheetDO sheet = createNewSheet();
    save(sheet.setTask(getTask("TBT-5")), "timesheet.error.taskNotBookable.orderPositionsFoundInSubTasks");
    timesheetDao.save(sheet.setTask(getTask("TBT-5.1")));
    timesheetDao.save(createNewSheet().setTask(getTask("TBT-5.1.1")));
    timesheetDao.save(createNewSheet().setTask(getTask("TBT-5.1.2")));
    sheet = createNewSheet();
    save(sheet.setTask(getTask("TBT-5.2")), "timesheet.error.taskNotBookable.orderPositionsFoundInSubTasks");
    save(sheet.setTask(getTask("TBT-5.2.1")), "timesheet.error.taskNotBookable.orderPositionsFoundInSubTasks");
    timesheetDao.save(sheet.setTask(getTask("TBT-5.2.1.1")));
  }

  @Test
  public void testTaskStatus()
  {
    initialize();
    final PFUserDO user = getUser(TEST_USER);
    logon(user);
    TimesheetDO sheet = createNewSheet().setTask(getTask("TBT-1"));
    timesheetDao.save(sheet);
    sheet = createNewSheet().setTask(getTask("TBT-1.1"));
    save(sheet, "timesheet.error.taskNotBookable.taskNotOpened");
    save(sheet.setTask(getTask("TBT-1.2")), "timesheet.error.taskNotBookable.taskDeleted");
    save(sheet.setTask(getTask("TBT-1.2.1")), "timesheet.error.taskNotBookable.taskDeleted");
  }

  private TimesheetDO createNewSheet()
  {
    return new TimesheetDO().setUser(getUser(TEST_USER)).setStartDate(date.getDate()).setStopTime(
        date.add(Calendar.MINUTE, 15).getTimestamp());
  }

  private void save(final TimesheetDO sheet, final String expectedErrorMsgKey)
  {
    try {
      timesheetDao.save(sheet);
      fail("AccessException expected: " + expectedErrorMsgKey);
    } catch (final AccessException ex) {
      assertEquals(expectedErrorMsgKey, ex.getI18nKey());
    }
  }
}