src/lib-mail/test-mail-html2text.c

	test-mail-html2text.c revision c9141125278100269eb3a907c911afe78c46717c
/* Copyright (c) 2015 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "str.h"
#include "istream.h"
#include "mail-html2text.h"
#include "test-common.h"

static struct {
    const char *input;
    const char *output;
} tests[] = {
    { "a&amp;&lt;&clubs;&gt;b",
      "a&<\xE2\x99\xA3>b" },
    { "&", "" },
    { "&amp", "" },

    { "a<style>stylesheet is ignored</style>b",
      "a b" },
    { "a<stylea>b</stylea>c",
      "a b c" },
    { "a<!--x <p foo=\"bar\">commented tags ignored also</p> y-->b",
      "ab" },
    { "a<script>javascript <p>foo</p> ignored</script>b",
      "a b" },
    { "a<scripta>b</scripta>c",
      "a b c" },
    { "a<blockquote><blockquote>second level</blockquote>ignored</blockquote>b",
      "a b" },
    { "a<![CDATA[<style>]] >b</style>]]>c",
      "a<style>]] >b</style>c" },

    { "a<foo", "a" },
    { "a<blockquote", "a" },
    { "a<blockquote>foo</blockquote", "a " },
    { "a<", "a" },
    { "a<![CDATA[b", "ab" },
    { "a<![CDATA[b]]", "ab" }
};

static const char *test_blockquote_input =
    "a<blockquote>b<blockquote><blockquote>c</blockquote>d</blockquote>e</blockquote>f";
static const char *test_blockquote_output = "a b c d e f";

static void test_mail_html2text(void)
{
    string_t *str = t_str_new(128);
    struct mail_html2text *ht;
    unsigned int i, j;

    test_begin("mail_html2text()");
    for (i = 0; i < N_ELEMENTS(tests); i++) {
        ht = mail_html2text_init(MAIL_HTML2TEXT_FLAG_SKIP_QUOTED);
        for (j = 0; tests[i].input[j] != '\0'; j++) {
            unsigned char c = tests[i].input[j];
            mail_html2text_more(ht, &c, 1, str);
        }
        test_assert_idx(strcmp(str_c(str), tests[i].output) == 0, i);
        mail_html2text_deinit(&ht);
        str_truncate(str, 0);
    }

    /* test without skipping quoted */
    ht = mail_html2text_init(0);
    mail_html2text_more(ht, (const void *)test_blockquote_input,
                strlen(test_blockquote_input), str);
    test_assert(strcmp(str_c(str), test_blockquote_output) == 0);
    mail_html2text_deinit(&ht);

    test_end();
}

int main(void)
{
    static void (*test_functions[])(void) = {
        test_mail_html2text,
        NULL
    };
    return test_run(test_functions);
}