From c5e5a2cab931c52bd46d35ed150fa05c5e6ef4c7 Mon Sep 17 00:00:00 2001
From: "Fred L. Drake, Jr." <fdrake@users.sourceforge.net>
Date: Mon, 22 Apr 2002 18:46:00 +0000
Subject: [PATCH] Test support enhancements:

_xml_failure():  Clean up the output message so the first line doesn't
    wrap in the typical case.

CharData typedef:  If count is less than 0, consider it "unset".  This
    is used in the new support to check attribute values.

accumulate_characters():  Handle storage->count < 0.  Allow XML_Char
    to be more than one byte.

accumulate_attribute():  New function: store the value of the first
    attribute encountered in parsing.  This needs the concept of a
    CharData that is unset to allow capturing a value that is an empty
    string.

check_characters():  Handle storage->count < 0.  Allow XML_Char to be
    more than one byte.

run_character_check():  Create a local parser so that both
    run_character_check() and run_attribute_check() can be used in a
    single test function.

run_attribute_check():  New function: check that the value of the
    first attribute in a document matches the expected value.
---
 expat/tests/runtests.c | 59 ++++++++++++++++++++++++++++++++++++++----
 1 file changed, 54 insertions(+), 5 deletions(-)

diff --git a/expat/tests/runtests.c b/expat/tests/runtests.c
index 0b680681..e1d2e657 100644
--- a/expat/tests/runtests.c
+++ b/expat/tests/runtests.c
@@ -33,7 +33,8 @@ static void
 _xml_failure(const char *file, int line)
 {
     char buffer[1024];
-    sprintf(buffer, "%s (line %d, offset %d)\n    reported from %s, line %d",
+    sprintf(buffer,
+            "\n    %s (line %d, offset %d)\n    reported from %s, line %d",
             XML_ErrorString(XML_GetErrorCode(parser)),
             XML_GetCurrentLineNumber(parser),
             XML_GetCurrentColumnNumber(parser),
@@ -117,7 +118,7 @@ END_TEST
 
 typedef struct 
 {
-    int count;
+    int count;                          /* # of chars, < 0 if not set */
     XML_Char data[1024];
 } CharData;
 
@@ -125,24 +126,48 @@ static void
 accumulate_characters(void *userData, const XML_Char *s, int len)
 {
     CharData *storage = (CharData *)userData;
+    if (storage->count < 0)
+        storage->count = 0;
     if (len + storage->count < sizeof(storage->data)) {
-        memcpy(storage->data + storage->count, s, len);
+        memcpy(storage->data + storage->count, s,
+               len * sizeof(storage->data[0]));
         storage->count += len;
     }
 }
 
+static void
+accumulate_attribute(void *userData, const XML_Char *name,
+                     const XML_Char **atts)
+{
+    CharData *storage = (CharData *)userData;
+    if (storage->count < 0 && atts != NULL && atts[0] != NULL) {
+        /* "accumulate" the value of the first attribute we see */
+        int maxchars = sizeof(storage->data) / sizeof(storage->data[0]);
+        int i;
+        for (i = 0; i < maxchars; ++i) {
+            XML_Char ch = atts[1][i];
+            if (ch == 0)
+                break;
+            storage->data[i] = ch;
+        }
+        storage->count = i;
+    }
+}
+
 static void
 check_characters(CharData *storage, XML_Char *expected)
 {
     char buffer[1024];
     int len = strlen(expected);
+    if (storage->count < 0)
+        storage->count = 0;
     if (len != storage->count) {
         sprintf(buffer, "wrong number of data characters: got %d, expected %d",
                 storage->count, len);
         fail(buffer);
         return;
     }
-    if (memcmp(expected, storage->data, len) != 0)
+    if (memcmp(expected, storage->data, len * sizeof(storage->data[0])) != 0)
         fail("got bad data bytes");
 }
 
@@ -150,12 +175,36 @@ static void
 run_character_check(XML_Char *text, XML_Char *expected)
 {
     CharData storage;
-    storage.count = 0;
+    XML_Parser parser = XML_ParserCreate(NULL);
+
+    if (parser == NULL)
+        fail("Parser not created.");
+
+    storage.count = -1;
     XML_SetUserData(parser, &storage);
     XML_SetCharacterDataHandler(parser, accumulate_characters);
     if (!XML_Parse(parser, text, strlen(text), 1))
         xml_failure();
     check_characters(&storage, expected);
+    XML_ParserFree(parser);
+}
+
+static void
+run_attribute_check(XML_Char *text, XML_Char *expected)
+{
+    CharData storage;
+    XML_Parser parser = XML_ParserCreate(NULL);
+
+    if (parser == NULL)
+        fail("Parser not created.");
+
+    storage.count = -1; /* magical "not-set" value */
+    XML_SetUserData(parser, &storage);
+    XML_SetStartElementHandler(parser, accumulate_attribute);
+    if (!XML_Parse(parser, text, strlen(text), 1))
+        xml_failure();
+    check_characters(&storage, expected);
+    XML_ParserFree(parser);
 }
 
 /* Regression test for SF bug #491986. */
-- 
2.40.0