1 files changed, 141 insertions, 0 deletions
diff --git a/src/testlib/qtestxunitstreamer.cpp b/src/testlib/qtestxunitstreamer.cpp
new file mode 100644
index 0000000..2d8b7c4
--- /dev/null
+++ b/src/testlib/qtestxunitstreamer.cpp
@@ -0,0 +1,141 @@
+#include "qtestxunitstreamer.h"
+#include "qtestelement.h"
+
+#include "QtTest/private/qtestlog_p.h"
+#include "QtTest/private/qtestresult_p.h"
+#include "QtTest/private/qxmltestlogger_p.h"
+
+QT_BEGIN_NAMESPACE
+
+QTestXunitStreamer::QTestXunitStreamer()
+    :QTestBasicStreamer()
+{}
+
+QTestXunitStreamer::~QTestXunitStreamer()
+{}
+
+void QTestXunitStreamer::indentForElement(const QTestElement* element, char* buf, int size)
+{
+    if (size == 0) return;
+
+    buf[0] = 0;
+
+    if (!element) return;
+
+    char* endbuf = buf + size;
+    element = element->parentElement();
+    while (element && buf+2 < endbuf) {
+        *(buf++) = ' ';
+        *(buf++) = ' ';
+        *buf = 0;
+        element = element->parentElement();
+    }
+}
+
+void QTestXunitStreamer::formatStart(const QTestElement *element, char *formatted) const
+{
+    if(!element || !formatted )
+        return;
+
+    char indent[20];
+    indentForElement(element, indent, sizeof(indent));
+
+    QTest::qt_snprintf(formatted, 1024, "%s<%s", indent, element->elementName());
+}
+
+void QTestXunitStreamer::formatEnd(const QTestElement *element, char *formatted) const
+{
+    if(!element || !formatted )
+        return;
+
+    if(!element->childElements()){
+        QTest::qt_snprintf(formatted, 10, "");
+        return;
+    }
+
+    char indent[20];
+    indentForElement(element, indent, sizeof(indent));
+
+    QTest::qt_snprintf(formatted, 1024, "%s</%s>\n", indent, element->elementName());
+}
+
+void QTestXunitStreamer::formatAttributes(const QTestElementAttribute *attribute, char *formatted) const
+{
+    if(!attribute || !formatted )
+        return;
+
+    QTest::AttributeIndex attrindex = attribute->index();
+
+    char const* key = 0;
+    if (attrindex == QTest::AI_Description)
+        key = "message";
+    else if (attrindex != QTest::AI_File && attrindex != QTest::AI_Line)
+        key = attribute->name();
+
+    if (key) {
+        char quotedValue[900];
+        QXmlTestLogger::xmlQuote(quotedValue, attribute->value(), sizeof(quotedValue));
+        QTest::qt_snprintf(formatted, 1024, " %s=\"%s\"", key, quotedValue);
+    }
+    else {
+        QTest::qt_snprintf(formatted, 10, "");
+    }
+}
+
+void QTestXunitStreamer::formatAfterAttributes(const QTestElement *element, char *formatted) const
+{
+    if(!element || !formatted )
+        return;
+
+        if(!element->childElements())
+            QTest::qt_snprintf(formatted, 10, "/>\n");
+        else
+            QTest::qt_snprintf(formatted, 10, ">\n");
+}
+
+void QTestXunitStreamer::output(QTestElement *element) const
+{
+    char buf[1024];
+    QTest::qt_snprintf(buf, sizeof(buf), "<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n");
+    outputString(buf);
+    QTestBasicStreamer::output(element);
+}
+
+void QTestXunitStreamer::outputElements(QTestElement *element, bool) const
+{
+    char buf[1024];
+    bool hasChildren;
+    /*
+        Elements are in reverse order of occurrence, so start from the end and work
+        our way backwards.
+    */
+    while (element && element->nextElement()) {
+        element = element->nextElement();
+    }
+    while (element) {
+        hasChildren = element->childElements();
+
+        if(element->elementType() != QTest::LET_Benchmark){
+            formatStart(element, buf);
+            outputString(buf);
+
+            formatBeforeAttributes(element, buf);
+            outputString(buf);
+
+            outputElementAttributes(element->attributes());
+
+            formatAfterAttributes(element, buf);
+            outputString(buf);
+
+            if(hasChildren)
+                outputElements(element->childElements(), true);
+
+            formatEnd(element, buf);
+            outputString(buf);
+        }
+        element = element->previousElement();
+    }
+}
+
+QT_END_NAMESPACE
+