Haiku porting patches, Credit to Paul Davey.
[oota-llvm.git] / utils / unittest / googletest / gtest-test-part.cc
1 // Copyright 2008, Google Inc.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 //
8 //     * Redistributions of source code must retain the above copyright
9 // notice, this list of conditions and the following disclaimer.
10 //     * Redistributions in binary form must reproduce the above
11 // copyright notice, this list of conditions and the following disclaimer
12 // in the documentation and/or other materials provided with the
13 // distribution.
14 //     * Neither the name of Google Inc. nor the names of its
15 // contributors may be used to endorse or promote products derived from
16 // this software without specific prior written permission.
17 //
18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 //
30 // Author: mheule@google.com (Markus Heule)
31 //
32 // The Google C++ Testing Framework (Google Test)
33
34 #include <gtest/gtest-test-part.h>
35
36 // Indicates that this translation unit is part of Google Test's
37 // implementation.  It must come before gtest-internal-inl.h is
38 // included, or there will be a compiler error.  This trick is to
39 // prevent a user from accidentally including gtest-internal-inl.h in
40 // his code.
41 #define GTEST_IMPLEMENTATION
42 #include "gtest/internal/gtest-internal-inl.h"
43 #undef GTEST_IMPLEMENTATION
44
45 namespace testing {
46
47 // Gets the summary of the failure message by omitting the stack trace
48 // in it.
49 internal::String TestPartResult::ExtractSummary(const char* message) {
50   const char* const stack_trace = strstr(message, internal::kStackTraceMarker);
51   return stack_trace == NULL ? internal::String(message) :
52       internal::String(message, stack_trace - message);
53 }
54
55 // Prints a TestPartResult object.
56 std::ostream& operator<<(std::ostream& os, const TestPartResult& result) {
57   return os << result.file_name() << ":"
58             << result.line_number() << ": "
59             << (result.type() == TPRT_SUCCESS ? "Success" :
60                 result.type() == TPRT_FATAL_FAILURE ? "Fatal failure" :
61                 "Non-fatal failure") << ":\n"
62             << result.message() << std::endl;
63 }
64
65 // Constructs an empty TestPartResultArray.
66 TestPartResultArray::TestPartResultArray()
67     : list_(new internal::List<TestPartResult>) {
68 }
69
70 // Destructs a TestPartResultArray.
71 TestPartResultArray::~TestPartResultArray() {
72   delete list_;
73 }
74
75 // Appends a TestPartResult to the array.
76 void TestPartResultArray::Append(const TestPartResult& result) {
77   list_->PushBack(result);
78 }
79
80 // Returns the TestPartResult at the given index (0-based).
81 const TestPartResult& TestPartResultArray::GetTestPartResult(int index) const {
82   if (index < 0 || index >= size()) {
83     printf("\nInvalid index (%d) into TestPartResultArray.\n", index);
84     internal::abort();
85   }
86
87   const internal::ListNode<TestPartResult>* p = list_->Head();
88   for (int i = 0; i < index; i++) {
89     p = p->next();
90   }
91
92   return p->element();
93 }
94
95 // Returns the number of TestPartResult objects in the array.
96 int TestPartResultArray::size() const {
97   return list_->size();
98 }
99
100 namespace internal {
101
102 HasNewFatalFailureHelper::HasNewFatalFailureHelper()
103     : has_new_fatal_failure_(false),
104       original_reporter_(UnitTest::GetInstance()->impl()->
105                          GetTestPartResultReporterForCurrentThread()) {
106   UnitTest::GetInstance()->impl()->SetTestPartResultReporterForCurrentThread(
107       this);
108 }
109
110 HasNewFatalFailureHelper::~HasNewFatalFailureHelper() {
111   UnitTest::GetInstance()->impl()->SetTestPartResultReporterForCurrentThread(
112       original_reporter_);
113 }
114
115 void HasNewFatalFailureHelper::ReportTestPartResult(
116     const TestPartResult& result) {
117   if (result.fatally_failed())
118     has_new_fatal_failure_ = true;
119   original_reporter_->ReportTestPartResult(result);
120 }
121
122 }  // namespace internal
123
124 }  // namespace testing