#include "qtestlightxmlstreamer.h"
#include "qtestelement.h"
#include "qtestelementattribute.h"

#include "QtTest/private/qtestlog_p.h"
#include "QtTest/private/qtestresult_p.h"
#include "QtTest/private/qxmltestlogger_p.h"

#include <string.h>

QT_BEGIN_NAMESPACE

QTestLightXmlStreamer::QTestLightXmlStreamer()
    :QTestBasicStreamer()
{
}

QTestLightXmlStreamer::~QTestLightXmlStreamer()
{}

void QTestLightXmlStreamer::formatStart(const QTestElement *element, char *formatted) const
{
    if(!element || !formatted)
        return;

    switch(element->elementType()){
    case QTest::LET_TestCase: {
        char quotedTf[950];
        QXmlTestLogger::xmlQuote(quotedTf, element->attributeValue(QTest::AI_Name),
            sizeof(quotedTf));

        QTest::qt_snprintf(formatted, 1024, "<TestFunction name=\"%s\">\n", quotedTf);
        break;
    }
    case QTest::LET_Failure: {
        char cdataDesc[900];
        QXmlTestLogger::xmlCdata(cdataDesc, element->attributeValue(QTest::AI_Description),
            sizeof(cdataDesc));

        QTest::qt_snprintf(formatted, 1024, "    <Description><![CDATA[%s]]></Description>\n",
                           cdataDesc);
         break;
    }
    case QTest::LET_Error: {
        // assuming type and attribute names don't need quoting
        char quotedFile[128];
        char cdataDesc[700];
        QXmlTestLogger::xmlQuote(quotedFile, element->attributeValue(QTest::AI_File),
            sizeof(quotedFile));
        QXmlTestLogger::xmlCdata(cdataDesc, element->attributeValue(QTest::AI_Description),
            sizeof(cdataDesc));

        QTest::qt_snprintf(formatted, 1024, "<Message type=\"%s\" %s=\"%s\" %s=\"%s\">\n    <Description><![CDATA[%s]]></Description>\n</Message>\n",
                           element->attributeValue(QTest::AI_Type),
                           element->attributeName(QTest::AI_File),
                           quotedFile,
                           element->attributeName(QTest::AI_Line),
                           element->attributeValue(QTest::AI_Line),
                           cdataDesc);
        break;
    }
    case QTest::LET_Benchmark: {
        // assuming value and iterations don't need quoting
        char quotedMetric[256];
        char quotedTag[256];
        QXmlTestLogger::xmlQuote(quotedMetric, element->attributeValue(QTest::AI_Metric),
            sizeof(quotedMetric));
        QXmlTestLogger::xmlQuote(quotedTag, element->attributeValue(QTest::AI_Tag),
            sizeof(quotedTag));

        QTest::qt_snprintf(formatted, 1024, "<BenchmarkResult %s=\"%s\" %s=\"%s\" %s=\"%s\" %s=\"%s\" />\n",
                           element->attributeName(QTest::AI_Metric),
                           quotedMetric,
                           element->attributeName(QTest::AI_Tag),
                           quotedTag,
                           element->attributeName(QTest::AI_Value),
                           element->attributeValue(QTest::AI_Value),
                           element->attributeName(QTest::AI_Iterations),
                           element->attributeValue(QTest::AI_Iterations) );
        break;
    }
    default:
        QTest::qt_snprintf(formatted, 10, "");
    }
}

void QTestLightXmlStreamer::formatEnd(const QTestElement *element, char *formatted) const
{
    if(!element || !formatted)
        return;

    if (element->elementType() == QTest::LET_TestCase) {
        if( element->attribute(QTest::AI_Result) && element->childElements())
            QTest::qt_snprintf(formatted, 1024, "</Incident>\n</TestFunction>\n");
        else
            QTest::qt_snprintf(formatted, 1024, "</TestFunction>\n");
    }
    else
        QTest::qt_snprintf(formatted, 10, "");
}

void QTestLightXmlStreamer::formatBeforeAttributes(const QTestElement *element, char *formatted) const
{
    if(!element || !formatted)
        return;

    if (element->elementType() == QTest::LET_TestCase && element->attribute(QTest::AI_Result)){
    char buf[900];
    char quotedFile[700];
    QXmlTestLogger::xmlQuote(quotedFile, element->attributeValue(QTest::AI_File),
        sizeof(quotedFile));

    QTest::qt_snprintf(buf, sizeof(buf), "%s=\"%s\" %s=\"%s\"",
                       element->attributeName(QTest::AI_File),
                       quotedFile,
                       element->attributeName(QTest::AI_Line),
                       element->attributeValue(QTest::AI_Line));

    if( !element->childElements() )
        QTest::qt_snprintf(formatted, 1024, "<Incident type=\"%s\" %s/>\n",
                           element->attributeValue(QTest::AI_Result), buf);
    else
        QTest::qt_snprintf(formatted, 1024, "<Incident type=\"%s\" %s>\n",
                               element->attributeValue(QTest::AI_Result), buf);
    }else{
        QTest::qt_snprintf(formatted, 10, "");
    }
}

void QTestLightXmlStreamer::output(QTestElement *element) const
{
    char buf[1024];
    QTest::qt_snprintf(buf, sizeof(buf), "<Environment>\n    <QtVersion>%s</QtVersion>\n    <QTestVersion>%s</QTestVersion>\n",
                       qVersion(), QTEST_VERSION_STR );
    outputString(buf);

    QTest::qt_snprintf(buf, sizeof(buf), "</Environment>\n");
    outputString(buf);

    QTestBasicStreamer::output(element);
}

QT_END_NAMESPACE