clustercontroller-utils/src/test/java/com/yahoo/vespa/clustercontroller/utils/communication/http/writer/HttpWriterTest.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.clustercontroller.utils.communication.http.writer;

import org.junit.jupiter.api.Test;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

public class HttpWriterTest {

    private static String defaultTitle = "My Title";
    private static String defaultHeader = "<html>\n"
                                        + "  <head>\n"
                                        + "    <title>My Title</title>\n"
                                        + "  </head>\n"
                                        + "  <body>\n"
                                        + "    <h1>My Title</h1>\n";
    private static String defaultFooter = "  </body>\n"
                                        + "</html>\n";

    @Test
    void testStructure() {
        HttpWriter writer = new HttpWriter();
        String header = defaultHeader.replace(defaultTitle, "Untitled page");
        assertEquals(header + defaultFooter, writer.toString());
    }

    @Test
    void testTitle() {
        HttpWriter writer = new HttpWriter().addTitle(defaultTitle);
        assertEquals(defaultHeader + defaultFooter, writer.toString());
    }

    @Test
    void testParagraph() {
        String paragraph = "This is a paragraph";
        String paragraph2 = "More text";
        HttpWriter writer = new HttpWriter().addTitle(defaultTitle).write(paragraph).write(paragraph2);
        String content = "    <p>\n"
                + "      " + paragraph + "\n"
                + "    </p>\n"
                + "    <p>\n"
                + "      " + paragraph2 + "\n"
                + "    </p>\n";
        assertEquals(defaultHeader + content + defaultFooter, writer.toString());
    }

    @Test
    void testLink() {
        String name = "My link";
        String link = "/foo/bar?hmm";
        HttpWriter writer = new HttpWriter().addTitle(defaultTitle).writeLink(name, link);
        String content = "    <a href=\"" + link + "\">" + name + "</a>\n";
        assertEquals(defaultHeader + content + defaultFooter, writer.toString());
    }

    @Test
    void testErrors() {
        try {
            HttpWriter writer = new HttpWriter().addTitle(defaultTitle);
            writer.toString();
            writer.write("foo");
            assertTrue(false);
        } catch (IllegalStateException e) {
        }
        try {
            new HttpWriter().write("foo").addTitle("bar");
            assertTrue(false);
        } catch (IllegalStateException e) {
        }
    }

}