Java tutorial
/** * This file is part of lavagna. * * lavagna 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, either version 3 of the License, or * (at your option) any later version. * * lavagna 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 lavagna. If not, see <http://www.gnu.org/licenses/>. */ package io.lavagna.service; import io.lavagna.config.PersistenceAndServiceConfig; import io.lavagna.model.Board; import io.lavagna.model.BoardColumn; import io.lavagna.model.BoardColumnDefinition; import io.lavagna.model.Card; import io.lavagna.model.CardLabel; import io.lavagna.model.CardLabelValue; import io.lavagna.model.LabelListValue; import io.lavagna.model.LabelListValueWithMetadata; import io.lavagna.model.Permission; import io.lavagna.model.Project; import io.lavagna.model.Role; import io.lavagna.model.User; import io.lavagna.model.UserWithPermission; import io.lavagna.service.config.TestServiceConfig; import java.io.IOException; import java.util.Collections; import java.util.Date; import java.util.EnumSet; import java.util.List; import java.util.Set; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.junit.Assert; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import org.springframework.transaction.annotation.Transactional; @RunWith(SpringJUnit4ClassRunner.class) @ContextConfiguration(classes = { TestServiceConfig.class, PersistenceAndServiceConfig.class }) @Transactional public class ExcelExportServiceTest { @Autowired private ExcelExportService excelExportService; @Autowired private BoardRepository boardRepository; @Autowired private UserRepository userRepository; @Autowired private BoardColumnRepository boardColumnRepository; @Autowired private CardService cardService; @Autowired private PermissionService permissionService; @Autowired private ProjectService projectService; @Autowired private CardLabelRepository cardLabelRepository; Project project; UserWithPermission user; @Before public void prepare() { Helper.createUser(userRepository, "test", "test-user"); User u = userRepository.findUserByName("test", "test-user"); Role r = new Role("TEST"); permissionService.createRole(r); permissionService.updatePermissionsToRole(r, EnumSet.of(Permission.READ)); permissionService.assignRolesToUsers(Collections.singletonMap(r, Collections.singleton(u.getId()))); user = new UserWithPermission(u, permissionService.findBasePermissionByUserId(u.getId()), Collections.<String, Set<Permission>>emptyMap(), Collections.<Integer, Set<Permission>>emptyMap()); project = projectService.create("test", "TEST", "desc"); CardLabel lnull = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.NULL, CardLabel.LabelDomain.USER, "null", 0); CardLabel lstring = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.STRING, CardLabel.LabelDomain.USER, "string", 0); CardLabel ltimestamp = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.TIMESTAMP, CardLabel.LabelDomain.USER, "date", 0); CardLabel luser = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.USER, CardLabel.LabelDomain.USER, "user", 0); CardLabel lcard = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.CARD, CardLabel.LabelDomain.USER, "card", 0); CardLabel lint = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.INT, CardLabel.LabelDomain.USER, "int", 0); CardLabel llist = cardLabelRepository.addLabel(project.getId(), false, CardLabel.LabelType.LIST, CardLabel.LabelDomain.USER, "list", 0); LabelListValue llistVal = cardLabelRepository.addLabelListValue(llist.getId(), "abcd"); // Init new card Board board = boardRepository.createNewBoard("test-label", "LABEL", "label", projectService.findByShortName("TEST").getId()); List<BoardColumnDefinition> definitions = projectService.findColumnDefinitionsByProjectId(project.getId()); BoardColumn column = boardColumnRepository.addColumnToBoard("label-column", definitions.get(0).getId(), BoardColumn.BoardColumnLocation.BOARD, board.getId()); Card card = cardService.createCard("card", column.getId(), new Date(), user); // Init milestone CardLabel l = cardLabelRepository.findLabelByName(project.getId(), "MILESTONE", CardLabel.LabelDomain.SYSTEM); LabelListValue mlv = cardLabelRepository.addLabelListValue(l.getId(), "1.0"); // Add labels cardLabelRepository.addLabelValueToCard(l, card.getId(), new CardLabelValue.LabelValue(null, null, null, null, null, mlv.getId())); cardLabelRepository.addLabelValueToCard(lnull, card.getId(), new CardLabelValue.LabelValue(null, null, null, null, null, null)); cardLabelRepository.addLabelValueToCard(lstring, card.getId(), new CardLabelValue.LabelValue("ABC")); cardLabelRepository.addLabelValueToCard(ltimestamp, card.getId(), new CardLabelValue.LabelValue(new Date())); cardLabelRepository.addLabelValueToCard(luser, card.getId(), new CardLabelValue.LabelValue(null, null, null, null, u.getId(), null)); cardLabelRepository.addLabelValueToCard(lcard, card.getId(), new CardLabelValue.LabelValue(null, null, null, card.getId(), null, null)); cardLabelRepository.addLabelValueToCard(lint, card.getId(), new CardLabelValue.LabelValue(null, null, 999, null, null, null)); cardLabelRepository.addLabelValueToCard(llist, card.getId(), new CardLabelValue.LabelValue(null, null, null, null, null, llistVal.getId())); } @Test public void testGetWrongMilestone() { LabelListValueWithMetadata m = excelExportService.getMilestone(project.getId(), "AAAA"); Assert.assertNull(m); } @Test public void testGetMilestone() { LabelListValueWithMetadata m = excelExportService.getMilestone(project.getId(), "1.0"); Assert.assertNotNull(m); } @Test(expected = IllegalArgumentException.class) public void testExportMilestoneToExcelWrongValue() throws IOException { excelExportService.exportMilestoneToExcel(project.getShortName(), "AAA", user); } @Test public void testExportMilestoneToExcel() throws IOException { HSSFWorkbook w = excelExportService.exportMilestoneToExcel(project.getShortName(), "1.0", user); Assert.assertNotNull(w); Assert.assertEquals(1, w.getSheet("1.0").getLastRowNum()); // 0 based -> 1 means 2 rows (header + 1 card) } @Test public void testExportProjectToExcel() throws IOException { HSSFWorkbook w = excelExportService.exportProjectToExcel(project.getShortName(), user); Assert.assertNotNull(w); Assert.assertEquals(1, w.getSheet(project.getName()).getLastRowNum()); // 0 based -> 1 means 2 rows (header + 1 card) } }