src/testlib/qabstracttestlogger.cpp

   1 /****************************************************************************
   2 **
   3 ** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).
   4 ** All rights reserved.
   5 ** Contact: Nokia Corporation (qt-info@nokia.com)
   6 **
   7 ** This file is part of the QtTest module of the Qt Toolkit.
   8 **
   9 ** $QT_BEGIN_LICENSE:LGPL$
  10 ** No Commercial Usage
  11 ** This file contains pre-release code and may not be distributed.
  12 ** You may use this file in accordance with the terms and conditions
  13 ** contained in the Technology Preview License Agreement accompanying
  14 ** this package.
  15 **
  16 ** GNU Lesser General Public License Usage
  17 ** Alternatively, this file may be used under the terms of the GNU Lesser
  18 ** General Public License version 2.1 as published by the Free Software
  19 ** Foundation and appearing in the file LICENSE.LGPL included in the
  20 ** packaging of this file.  Please review the following information to
  21 ** ensure the GNU Lesser General Public License version 2.1 requirements
  22 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
  23 **
  24 ** In addition, as a special exception, Nokia gives you certain additional
  25 ** rights.  These rights are described in the Nokia Qt LGPL Exception
  26 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
  27 **
  28 ** If you have questions regarding the use of this file, please contact
  29 ** Nokia at qt-info@nokia.com.
  30 **
  31 **
  32 **
  33 **
  34 **
  35 **
  36 **
  37 **
  38 ** $QT_END_LICENSE$
  39 **
  40 ****************************************************************************/
  41
  42 #include "QtTest/private/qabstracttestlogger_p.h"
  43 #include "QtTest/private/qtestlog_p.h"
  44 #include "QtTest/qtestassert.h"
  45
  46 #include "QtCore/qbytearray.h"
  47
  48 #include <stdio.h>
  49 #include <stdlib.h>
  50 #include <stdarg.h>
  51
  52 #ifndef Q_OS_WIN
  53 #include <unistd.h>
  54 #endif
  55
  56 QT_BEGIN_NAMESPACE
  57
  58 namespace QTest
  59 {
  60     static FILE *stream = 0;
  61 }
  62
  63 void QAbstractTestLogger::outputString(const char *msg)
  64 {
  65     QTEST_ASSERT(QTest::stream);
  66
  67     ::fputs(msg, QTest::stream);
  68     ::fflush(QTest::stream);
  69 }
  70
  71 bool QAbstractTestLogger::isTtyOutput()
  72 {
  73     QTEST_ASSERT(QTest::stream);
  74
  75 #if defined(Q_OS_WIN) || defined(Q_OS_INTEGRITY)
  76     return true;
  77 #else
  78     static bool ttyoutput = isatty(fileno(QTest::stream));
  79     return ttyoutput;
  80 #endif
  81 }
  82
  83
  84 void QAbstractTestLogger::startLogging()
  85 {
  86     QTEST_ASSERT(!QTest::stream);
  87
  88     const char *out = QTestLog::outputFileName();
  89     if (!out) {
  90         QTest::stream = stdout;
  91         return;
  92     }
  93 #if defined(_MSC_VER) && _MSC_VER >= 1400 && !defined(Q_OS_WINCE)
  94     if (::fopen_s(&QTest::stream, out, "wt")) {
  95 #else
  96     QTest::stream = ::fopen(out, "wt");
  97     if (!QTest::stream) {
  98 #endif
  99         printf("Unable to open file for logging: %s", out);
 100         ::exit(1);
 101     }
 102 }
 103
 104 void QAbstractTestLogger::stopLogging()
 105 {
 106     QTEST_ASSERT(QTest::stream);
 107     if (QTest::stream != stdout)
 108         fclose(QTest::stream);
 109     QTest::stream = 0;
 110 }
 111
 112 namespace QTest
 113 {
 114
 115 extern void filter_unprintable(char *str);
 116
 117 /*!
 118     \fn int QTest::qt_asprintf(QTestCharBuffer *buf, const char *format, ...);
 119     \internal
 120  */
 121 int qt_asprintf(QTestCharBuffer *str, const char *format, ...)
 122 {
 123     static const int MAXSIZE = 1024*1024*2;
 124
 125     Q_ASSERT(str);
 126
 127     int size = str->size();
 128
 129     va_list ap;
 130     int res = 0;
 131
 132     for (;;) {
 133         va_start(ap, format);
 134         res = qvsnprintf(str->data(), size, format, ap);
 135         va_end(ap);
 136         str->data()[size - 1] = '\0';
 137         if (res >= 0 && res < size) {
 138             // We succeeded
 139             break;
 140         }
 141         // buffer wasn't big enough, try again.
 142         // Note, we're assuming that a result of -1 is always due to running out of space.
 143         size *= 2;
 144         if (size > MAXSIZE) {
 145             break;
 146         }
 147         if (!str->reset(size))
 148             break; // out of memory - take what we have
 149     }
 150
 151     filter_unprintable(str->data());
 152
 153     return res;
 154 }
 155
 156 }
 157
 158 QT_END_NAMESPACE