]>
Commit | Line | Data |
---|---|---|
31f18b77 FG |
1 | // Copyright 2007, 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: wan@google.com (Zhanyong Wan) | |
31 | // | |
32 | // Tests using global test environments. | |
33 | ||
34 | #include <stdlib.h> | |
35 | #include <stdio.h> | |
36 | #include "gtest/gtest.h" | |
37 | ||
38 | #define GTEST_IMPLEMENTATION_ 1 // Required for the next #include. | |
39 | #include "src/gtest-internal-inl.h" | |
40 | #undef GTEST_IMPLEMENTATION_ | |
41 | ||
42 | namespace testing { | |
43 | GTEST_DECLARE_string_(filter); | |
44 | } | |
45 | ||
46 | namespace { | |
47 | ||
48 | enum FailureType { | |
49 | NO_FAILURE, NON_FATAL_FAILURE, FATAL_FAILURE | |
50 | }; | |
51 | ||
52 | // For testing using global test environments. | |
53 | class MyEnvironment : public testing::Environment { | |
54 | public: | |
55 | MyEnvironment() { Reset(); } | |
56 | ||
57 | // Depending on the value of failure_in_set_up_, SetUp() will | |
58 | // generate a non-fatal failure, generate a fatal failure, or | |
59 | // succeed. | |
60 | virtual void SetUp() { | |
61 | set_up_was_run_ = true; | |
62 | ||
63 | switch (failure_in_set_up_) { | |
64 | case NON_FATAL_FAILURE: | |
65 | ADD_FAILURE() << "Expected non-fatal failure in global set-up."; | |
66 | break; | |
67 | case FATAL_FAILURE: | |
68 | FAIL() << "Expected fatal failure in global set-up."; | |
69 | break; | |
70 | default: | |
71 | break; | |
72 | } | |
73 | } | |
74 | ||
75 | // Generates a non-fatal failure. | |
76 | virtual void TearDown() { | |
77 | tear_down_was_run_ = true; | |
78 | ADD_FAILURE() << "Expected non-fatal failure in global tear-down."; | |
79 | } | |
80 | ||
81 | // Resets the state of the environment s.t. it can be reused. | |
82 | void Reset() { | |
83 | failure_in_set_up_ = NO_FAILURE; | |
84 | set_up_was_run_ = false; | |
85 | tear_down_was_run_ = false; | |
86 | } | |
87 | ||
88 | // We call this function to set the type of failure SetUp() should | |
89 | // generate. | |
90 | void set_failure_in_set_up(FailureType type) { | |
91 | failure_in_set_up_ = type; | |
92 | } | |
93 | ||
94 | // Was SetUp() run? | |
95 | bool set_up_was_run() const { return set_up_was_run_; } | |
96 | ||
97 | // Was TearDown() run? | |
98 | bool tear_down_was_run() const { return tear_down_was_run_; } | |
99 | ||
100 | private: | |
101 | FailureType failure_in_set_up_; | |
102 | bool set_up_was_run_; | |
103 | bool tear_down_was_run_; | |
104 | }; | |
105 | ||
106 | // Was the TEST run? | |
107 | bool test_was_run; | |
108 | ||
109 | // The sole purpose of this TEST is to enable us to check whether it | |
110 | // was run. | |
111 | TEST(FooTest, Bar) { | |
112 | test_was_run = true; | |
113 | } | |
114 | ||
115 | // Prints the message and aborts the program if condition is false. | |
116 | void Check(bool condition, const char* msg) { | |
117 | if (!condition) { | |
118 | printf("FAILED: %s\n", msg); | |
119 | testing::internal::posix::Abort(); | |
120 | } | |
121 | } | |
122 | ||
123 | // Runs the tests. Return true iff successful. | |
124 | // | |
125 | // The 'failure' parameter specifies the type of failure that should | |
126 | // be generated by the global set-up. | |
127 | int RunAllTests(MyEnvironment* env, FailureType failure) { | |
128 | env->Reset(); | |
129 | env->set_failure_in_set_up(failure); | |
130 | test_was_run = false; | |
131 | testing::internal::GetUnitTestImpl()->ClearAdHocTestResult(); | |
132 | return RUN_ALL_TESTS(); | |
133 | } | |
134 | ||
135 | } // namespace | |
136 | ||
137 | int main(int argc, char **argv) { | |
138 | testing::InitGoogleTest(&argc, argv); | |
139 | ||
140 | // Registers a global test environment, and verifies that the | |
141 | // registration function returns its argument. | |
142 | MyEnvironment* const env = new MyEnvironment; | |
143 | Check(testing::AddGlobalTestEnvironment(env) == env, | |
144 | "AddGlobalTestEnvironment() should return its argument."); | |
145 | ||
146 | // Verifies that RUN_ALL_TESTS() runs the tests when the global | |
147 | // set-up is successful. | |
148 | Check(RunAllTests(env, NO_FAILURE) != 0, | |
149 | "RUN_ALL_TESTS() should return non-zero, as the global tear-down " | |
150 | "should generate a failure."); | |
151 | Check(test_was_run, | |
152 | "The tests should run, as the global set-up should generate no " | |
153 | "failure"); | |
154 | Check(env->tear_down_was_run(), | |
155 | "The global tear-down should run, as the global set-up was run."); | |
156 | ||
157 | // Verifies that RUN_ALL_TESTS() runs the tests when the global | |
158 | // set-up generates no fatal failure. | |
159 | Check(RunAllTests(env, NON_FATAL_FAILURE) != 0, | |
160 | "RUN_ALL_TESTS() should return non-zero, as both the global set-up " | |
161 | "and the global tear-down should generate a non-fatal failure."); | |
162 | Check(test_was_run, | |
163 | "The tests should run, as the global set-up should generate no " | |
164 | "fatal failure."); | |
165 | Check(env->tear_down_was_run(), | |
166 | "The global tear-down should run, as the global set-up was run."); | |
167 | ||
168 | // Verifies that RUN_ALL_TESTS() runs no test when the global set-up | |
169 | // generates a fatal failure. | |
170 | Check(RunAllTests(env, FATAL_FAILURE) != 0, | |
171 | "RUN_ALL_TESTS() should return non-zero, as the global set-up " | |
172 | "should generate a fatal failure."); | |
173 | Check(!test_was_run, | |
174 | "The tests should not run, as the global set-up should generate " | |
175 | "a fatal failure."); | |
176 | Check(env->tear_down_was_run(), | |
177 | "The global tear-down should run, as the global set-up was run."); | |
178 | ||
179 | // Verifies that RUN_ALL_TESTS() doesn't do global set-up or | |
180 | // tear-down when there is no test to run. | |
181 | testing::GTEST_FLAG(filter) = "-*"; | |
182 | Check(RunAllTests(env, NO_FAILURE) == 0, | |
183 | "RUN_ALL_TESTS() should return zero, as there is no test to run."); | |
184 | Check(!env->set_up_was_run(), | |
185 | "The global set-up should not run, as there is no test to run."); | |
186 | Check(!env->tear_down_was_run(), | |
187 | "The global tear-down should not run, " | |
188 | "as the global set-up was not run."); | |
189 | ||
190 | printf("PASS\n"); | |
191 | return 0; | |
192 | } |