|
|
#include "unity/unity.h" |
|
|
#include <libxml/HTMLtree.h> |
|
|
|
|
|
#include <libxml/tree.h> |
|
|
#include <libxml/xmlIO.h> |
|
|
#include <libxml/parser.h> |
|
|
#include <string.h> |
|
|
#include <stdlib.h> |
|
|
|
|
|
|
|
|
extern void test_htmlDtdDumpOutput(xmlOutputBufferPtr buf, xmlDocPtr doc, const char *encoding); |
|
|
|
|
|
static char* run_dump_and_get_string(xmlDocPtr doc) { |
|
|
xmlBufferPtr xbuf = xmlBufferCreate(); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(xbuf, "Failed to create xmlBuffer"); |
|
|
|
|
|
xmlOutputBufferPtr out = xmlOutputBufferCreateBuffer(xbuf, NULL); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(out, "Failed to create xmlOutputBuffer"); |
|
|
|
|
|
test_htmlDtdDumpOutput(out, doc, "utf-8"); |
|
|
|
|
|
|
|
|
xmlOutputBufferFlush(out); |
|
|
|
|
|
const xmlChar *content = xmlBufferContent(xbuf); |
|
|
int len = xmlBufferLength(xbuf); |
|
|
if (len < 0) len = 0; |
|
|
|
|
|
char *ret = (char *)malloc((size_t)len + 1); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(ret, "malloc failed"); |
|
|
if (len > 0 && content != NULL) { |
|
|
memcpy(ret, content, (size_t)len); |
|
|
} |
|
|
ret[len] = '\0'; |
|
|
|
|
|
xmlOutputBufferClose(out); |
|
|
xmlBufferFree(xbuf); |
|
|
|
|
|
return ret; |
|
|
} |
|
|
|
|
|
void setUp(void) { |
|
|
xmlInitParser(); |
|
|
} |
|
|
|
|
|
void tearDown(void) { |
|
|
xmlCleanupParser(); |
|
|
} |
|
|
|
|
|
static xmlDocPtr make_doc(void) { |
|
|
xmlDocPtr doc = xmlNewDoc(BAD_CAST "1.0"); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(doc, "xmlNewDoc failed"); |
|
|
return doc; |
|
|
} |
|
|
|
|
|
void test_htmlDtdDumpOutput_no_subset_outputs_nothing(void) { |
|
|
xmlDocPtr doc = make_doc(); |
|
|
|
|
|
char *out = run_dump_and_get_string(doc); |
|
|
TEST_ASSERT_NOT_NULL(out); |
|
|
TEST_ASSERT_EQUAL_STRING("", out); |
|
|
|
|
|
free(out); |
|
|
xmlFreeDoc(doc); |
|
|
} |
|
|
|
|
|
void test_htmlDtdDumpOutput_name_only(void) { |
|
|
xmlDocPtr doc = make_doc(); |
|
|
xmlDtdPtr dtd = xmlCreateIntSubset(doc, BAD_CAST "html", NULL, NULL); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(dtd, "xmlCreateIntSubset failed"); |
|
|
|
|
|
char *out = run_dump_and_get_string(doc); |
|
|
TEST_ASSERT_EQUAL_STRING("<!DOCTYPE html>\n", out); |
|
|
|
|
|
free(out); |
|
|
xmlFreeDoc(doc); |
|
|
} |
|
|
|
|
|
void test_htmlDtdDumpOutput_public_only(void) { |
|
|
xmlDocPtr doc = make_doc(); |
|
|
xmlDtdPtr dtd = xmlCreateIntSubset(doc, BAD_CAST "html", |
|
|
BAD_CAST "PUBLIC_ID", |
|
|
NULL); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(dtd, "xmlCreateIntSubset failed"); |
|
|
|
|
|
char *out = run_dump_and_get_string(doc); |
|
|
TEST_ASSERT_EQUAL_STRING("<!DOCTYPE html PUBLIC \"PUBLIC_ID\">\n", out); |
|
|
|
|
|
free(out); |
|
|
xmlFreeDoc(doc); |
|
|
} |
|
|
|
|
|
void test_htmlDtdDumpOutput_system_only_non_legacy(void) { |
|
|
xmlDocPtr doc = make_doc(); |
|
|
xmlDtdPtr dtd = xmlCreateIntSubset(doc, BAD_CAST "html", |
|
|
NULL, |
|
|
BAD_CAST "http://example.com/sys.dtd"); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(dtd, "xmlCreateIntSubset failed"); |
|
|
|
|
|
char *out = run_dump_and_get_string(doc); |
|
|
TEST_ASSERT_EQUAL_STRING("<!DOCTYPE html SYSTEM \"http://example.com/sys.dtd\">\n", out); |
|
|
|
|
|
free(out); |
|
|
xmlFreeDoc(doc); |
|
|
} |
|
|
|
|
|
void test_htmlDtdDumpOutput_system_only_legacy_compat_suppressed(void) { |
|
|
xmlDocPtr doc = make_doc(); |
|
|
xmlDtdPtr dtd = xmlCreateIntSubset(doc, BAD_CAST "html", |
|
|
NULL, |
|
|
BAD_CAST "about:legacy-compat"); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(dtd, "xmlCreateIntSubset failed"); |
|
|
|
|
|
char *out = run_dump_and_get_string(doc); |
|
|
|
|
|
TEST_ASSERT_EQUAL_STRING("<!DOCTYPE html>\n", out); |
|
|
|
|
|
free(out); |
|
|
xmlFreeDoc(doc); |
|
|
} |
|
|
|
|
|
void test_htmlDtdDumpOutput_public_and_system(void) { |
|
|
xmlDocPtr doc = make_doc(); |
|
|
xmlDtdPtr dtd = xmlCreateIntSubset(doc, BAD_CAST "html", |
|
|
BAD_CAST "-//W3C//DTD HTML 4.01//EN", |
|
|
BAD_CAST "http://www.w3.org/TR/html4/strict.dtd"); |
|
|
TEST_ASSERT_NOT_NULL_MESSAGE(dtd, "xmlCreateIntSubset failed"); |
|
|
|
|
|
char *out = run_dump_and_get_string(doc); |
|
|
TEST_ASSERT_EQUAL_STRING("<!DOCTYPE html PUBLIC \"-//W3C//DTD HTML 4.01//EN\" \"http://www.w3.org/TR/html4/strict.dtd\">\n", out); |
|
|
|
|
|
free(out); |
|
|
xmlFreeDoc(doc); |
|
|
} |
|
|
|
|
|
int main(void) { |
|
|
UNITY_BEGIN(); |
|
|
|
|
|
RUN_TEST(test_htmlDtdDumpOutput_no_subset_outputs_nothing); |
|
|
RUN_TEST(test_htmlDtdDumpOutput_name_only); |
|
|
RUN_TEST(test_htmlDtdDumpOutput_public_only); |
|
|
RUN_TEST(test_htmlDtdDumpOutput_system_only_non_legacy); |
|
|
RUN_TEST(test_htmlDtdDumpOutput_system_only_legacy_compat_suppressed); |
|
|
RUN_TEST(test_htmlDtdDumpOutput_public_and_system); |
|
|
|
|
|
return UNITY_END(); |
|
|
} |
