| #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(); |
| } |
