common/src/test/java/com/android/utils/HtmlBuilderTest.java

/*
 * Copyright (C) 2013 The Android Open Source Project
 *
 * Licensed 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 com.android.utils;

import junit.framework.TestCase;

import java.io.File;
import java.io.IOException;

public class HtmlBuilderTest extends TestCase {

    public void testAddLink() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.add("Plain.");
        builder.addLink(" (link) ", "runnable:0");
        builder.add("Plain.");
        // Check that the spaces surrounding the link text are not included in the link range
        assertEquals("Plain. <A HREF=\"runnable:0\">(link)</A>Plain.", builder.getHtml());
    }

    public void testAddBold() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.addBold("This is bold");
        assertEquals("<B>This is bold</B>", builder.getHtml());
    }

    public void testAddItalic() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.addItalic("This is italic");
        assertEquals("<I>This is italic</I>", builder.getHtml());
    }

    public void testNestLinkInBold() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.add("Plain. ");
        builder.beginBold();
        builder.add("Bold. ");
        builder.addLink("mylink", "foo://bar:123");
        builder.endBold();
        assertEquals("Plain. <B>Bold. <A HREF=\"foo://bar:123\">mylink</A></B>",
                     builder.getHtml());
    }

    public void testAddList() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.add("Plain").newline();
        builder.beginList().listItem().add("item 1").listItem().add("item 2").endList();

        assertEquals("Plain<BR/>\n" +
                     "<DL>\n" +
                     "<DD>-&NBSP;item 1\n" +
                     "<DD>-&NBSP;item 2\n" +
                     "</DL>", builder.getHtml());
    }

    public void testAddLinkWithBeforeAndAfterText() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.addLink("This is the ", "linked text", "!", "foo://bar");
        assertEquals("This is the <A HREF=\"foo://bar\">linked text</A>!", builder.getHtml());
    }

    public void testAddTable() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.beginTable().addTableRow(true, "Header1", "Header2")
          .addTableRow("Data1", "Data2")
          .endTable();
        assertEquals(
          "<table><tr><th>Header1</th><th>Header2</th></tr><tr><td>Data1</td><td>Data2</td></tr></table>",
          builder.getHtml());
    }

    public void testAddTableWithAlignment() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.beginTable("valign=\"top\"").addTableRow("Data1", "Data2").endTable();
        assertEquals(
          "<table><tr><td valign=\"top\">Data1</td><td valign=\"top\">Data2</td></tr></table>",
          builder.getHtml());
    }

    public void testAddDiv() {
        HtmlBuilder builder = new HtmlBuilder();
        assertEquals("<div>Hello</div>", builder.beginDiv().add("Hello").endDiv().getHtml());
    }

    public void testAddDivWithPadding() {
        HtmlBuilder builder = new HtmlBuilder();
        assertEquals("<div style=\"padding: 10px; text-color: gray\">Hello</div>",
                     builder.beginDiv("padding: 10px; text-color: gray").add("Hello").endDiv()
                       .getHtml());
    }

    public void testAddImage() throws IOException {
        File f = File.createTempFile("img", "png");
        f.deleteOnExit();

        String actual = new HtmlBuilder().addImage(SdkUtils.fileToUrl(f), "preview").getHtml();
        String path = f.getAbsolutePath();

        if (!path.startsWith("/")) {
            path = '/' + path;
        }
        String expected = String.format("<img src='file:%1$s' alt=\"preview\" />", path);
        if (File.separatorChar != '/') {
            // SdkUtil.fileToUrl always returns / as a separator so adjust
            // Windows path accordingly.
            expected = expected.replace(File.separatorChar, '/');
        }
        assertEquals(expected, actual);
    }

    public void testNewlineIfNecessary() {
        HtmlBuilder builder = new HtmlBuilder();
        builder.newlineIfNecessary();
        assertEquals("<BR/>\n", builder.getHtml());
        builder.newlineIfNecessary();
        assertEquals("<BR/>\n", builder.getHtml());
        builder.add("a");
        builder.newlineIfNecessary();
        assertEquals("<BR/>\na<BR/>\n", builder.getHtml());
        builder.newline();
        builder.newlineIfNecessary();
        builder.newlineIfNecessary();
        builder.newlineIfNecessary();
        assertEquals("<BR/>\na<BR/>\n<BR/>\n", builder.getHtml());
    }
}