BaseTestHyperlink.java

/* ====================================================================
   Licensed to the Apache Software Foundation (ASF) under one or more
   contributor license agreements.  See the NOTICE file distributed with
   this work for additional information regarding copyright ownership.
   The ASF licenses this file to You under the Apache License, Version 2.0
   (the "License"); you may not use this file except in compliance with
   the License.  You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
==================================================================== */

package org.apache.poi.ss.usermodel;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotEquals;
import static org.junit.jupiter.api.Assertions.assertNotSame;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.List;

import org.apache.poi.common.usermodel.HyperlinkType;
import org.apache.poi.ss.ITestDataProvider;
import org.apache.poi.ss.SpreadsheetVersion;
import org.junit.jupiter.api.Test;

/**
 * Test different types of Excel hyperlinks
 */
public abstract class BaseTestHyperlink {

    private final ITestDataProvider _testDataProvider;

    protected BaseTestHyperlink(ITestDataProvider testDataProvider) {
        _testDataProvider = testDataProvider;
    }

    @Test
    public final void testBasicTypes() throws IOException {
        Workbook wb1 = _testDataProvider.createWorkbook();
        CreationHelper createHelper = wb1.getCreationHelper();

        Cell cell;
        Hyperlink link;
        Sheet sheet = wb1.createSheet("Hyperlinks");

        //URL
        cell = sheet.createRow(0).createCell((short) 0);
        cell.setCellValue("URL Link");
        link = createHelper.createHyperlink(HyperlinkType.URL);
        link.setAddress("https://poi.apache.org/");
        cell.setHyperlink(link);

        //link to a file in the current directory
        cell = sheet.createRow(1).createCell((short) 0);
        cell.setCellValue("File Link");
        link = createHelper.createHyperlink(HyperlinkType.FILE);
        link.setAddress("hyperinks-beta4-dump.txt");
        cell.setHyperlink(link);

        //e-mail link
        cell = sheet.createRow(2).createCell((short) 0);
        cell.setCellValue("Email Link");
        link = createHelper.createHyperlink(HyperlinkType.EMAIL);
        //note, if subject contains white spaces, make sure they are url-encoded
        link.setAddress("mailto:poi@apache.org?subject=Hyperlinks");
        cell.setHyperlink(link);

        //link to a place in this workbook

        //create a target sheet and cell
        Sheet sheet2 = wb1.createSheet("Target Sheet");
        sheet2.createRow(0).createCell((short) 0).setCellValue("Target Cell");

        cell = sheet.createRow(3).createCell((short) 0);
        cell.setCellValue("Worksheet Link");
        link = createHelper.createHyperlink(HyperlinkType.DOCUMENT);
        link.setAddress("'Target Sheet'!A1");
        cell.setHyperlink(link);

        Workbook wb2 = _testDataProvider.writeOutAndReadBack(wb1);
        wb1.close();

        sheet = wb2.getSheetAt(0);
        link = sheet.getRow(0).getCell(0).getHyperlink();

        assertEquals("https://poi.apache.org/", link.getAddress());
        link = sheet.getRow(1).getCell(0).getHyperlink();
        assertEquals("hyperinks-beta4-dump.txt", link.getAddress());
        link = sheet.getRow(2).getCell(0).getHyperlink();
        assertEquals("mailto:poi@apache.org?subject=Hyperlinks", link.getAddress());
        link = sheet.getRow(3).getCell(0).getHyperlink();
        assertEquals("'Target Sheet'!A1", link.getAddress());

        wb2.close();
    }

    // copy a hyperlink via the copy constructor
    @Test
    void testCopyHyperlink() throws IOException {
        final Workbook wb = _testDataProvider.createWorkbook();
        final CreationHelper createHelper = wb.getCreationHelper();

        final Sheet sheet = wb.createSheet("Hyperlinks");
        final Row row = sheet.createRow(0);
        final Cell cell1, cell2;
        final Hyperlink link1, link2;

        //URL
        cell1 = row.createCell(0);
        cell2 = row.createCell(1);
        cell1.setCellValue("URL Link");
        link1 = createHelper.createHyperlink(HyperlinkType.URL);
        link1.setAddress("https://poi.apache.org/");
        cell1.setHyperlink(link1);

        link2 = copyHyperlink(link1);

        // Change address (type is not changeable)
        link2.setAddress("http://apache.org/");
        cell2.setHyperlink(link2);

        // Make sure hyperlinks were deep-copied, and modifying one does not modify the other.
        assertNotSame(link1, link2);
        assertNotEquals(link1, link2);
        assertEquals("https://poi.apache.org/", link1.getAddress());
        assertEquals("http://apache.org/", link2.getAddress());
        assertEquals(link1, cell1.getHyperlink());
        assertEquals(link2, cell2.getHyperlink());

        // Make sure both hyperlinks were added to the sheet
        @SuppressWarnings("unchecked")
        final List<Hyperlink> actualHyperlinks = (List<Hyperlink>) sheet.getHyperlinkList();
        assertEquals(2, actualHyperlinks.size());
        assertEquals(link1, actualHyperlinks.get(0));
        assertEquals(link2, actualHyperlinks.get(1));

        wb.close();
    }

    @Test
    void testHyperlinkEmailType69265_https() throws IOException {
        boolean isHSSF = _testDataProvider.getSpreadsheetVersion() == SpreadsheetVersion.EXCEL97;

        try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
            try (Workbook workbook = _testDataProvider.createWorkbook()) {
                Sheet sheet = workbook.createSheet("Hyperlink Example");

                Row row = sheet.createRow(0);
                Cell cell = row.createCell(0);

                // Create a hyperlink
                CreationHelper createHelper = workbook.getCreationHelper();
                Hyperlink hyperlink = createHelper.createHyperlink(HyperlinkType.EMAIL);
                hyperlink.setLabel("mylabel");
                hyperlink.setAddress("https://www.example.com");

                // Set the label and the hyperlink
                cell.setCellValue("Click here");
                cell.setHyperlink(hyperlink);

                // Get the cell value and hyperlink address
                assertEquals("Click here", cell.getStringCellValue());
                Hyperlink cellHyperlink = cell.getHyperlink();
                assertEquals("https://www.example.com", cellHyperlink.getAddress());

                // HSSF does not support Email, thus falls back to URL, HSSF also uses a hardcoded "label"
                assertEquals(
                        isHSSF ? HyperlinkType.URL : HyperlinkType.EMAIL,
                        cellHyperlink.getType());
                assertEquals(
                        isHSSF ? "url" : "mylabel",
                        cellHyperlink.getLabel());

                workbook.write(out);
            }

            out.flush();

            try (Workbook wbBack = WorkbookFactory.create(new ByteArrayInputStream(out.toByteArray()))) {
                Sheet sheet = wbBack.getSheet("Hyperlink Example");
                Row row = sheet.getRow(0);
                Cell cell = row.getCell(0);

                Hyperlink hyperlink = cell.getHyperlink();

                // when not using "mailto:", it is reverted back to URL currently
                assertEquals(
                        HyperlinkType.URL,
                        hyperlink.getType());
                assertEquals(
                        isHSSF ? "url" : "mylabel",
                        hyperlink.getLabel());
            }
        }
    }

    @Test
    void testHyperlinkEmailType69265_mailto() throws IOException {
        boolean isHSSF = _testDataProvider.getSpreadsheetVersion() == SpreadsheetVersion.EXCEL97;

        try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
            try (Workbook workbook = _testDataProvider.createWorkbook()) {
                Sheet sheet = workbook.createSheet("Hyperlink Example");

                Row row = sheet.createRow(0);
                Cell cell = row.createCell(0);

                // Create a hyperlink
                CreationHelper createHelper = workbook.getCreationHelper();
                Hyperlink hyperlink = createHelper.createHyperlink(HyperlinkType.EMAIL);
                hyperlink.setLabel("mylabel");
                hyperlink.setAddress("mailto://www.example.com");

                // Set the label and the hyperlink
                cell.setCellValue("Click here");
                cell.setHyperlink(hyperlink);

                // Get the cell value and hyperlink address
                assertEquals("Click here", cell.getStringCellValue());
                Hyperlink cellHyperlink = cell.getHyperlink();
                assertEquals("mailto://www.example.com", cellHyperlink.getAddress());

                // "mailto:" is converted to type "EMAIL"
                assertEquals(
                        HyperlinkType.EMAIL,
                        cellHyperlink.getType());
                assertEquals(
                        isHSSF ? "url" : "mylabel",
                        cellHyperlink.getLabel());

                workbook.write(out);
            }

            out.flush();

            try (Workbook wbBack = WorkbookFactory.create(new ByteArrayInputStream(out.toByteArray()))) {
                Sheet sheet = wbBack.getSheet("Hyperlink Example");
                Row row = sheet.getRow(0);
                Cell cell = row.getCell(0);

                Hyperlink hyperlink = cell.getHyperlink();

                // "mailto:" is converted to type "EMAIL"
                assertEquals(
                        HyperlinkType.EMAIL,
                        hyperlink.getType());
                assertEquals(
                        isHSSF ? "url" : "mylabel",
                        hyperlink.getLabel());
            }
        }
    }

    public abstract Hyperlink copyHyperlink(Hyperlink link);
}