OLD | NEW |
| (Empty) |
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 private: | |
100 FailureType failure_in_set_up_; | |
101 bool set_up_was_run_; | |
102 bool tear_down_was_run_; | |
103 }; | |
104 | |
105 // Was the TEST run? | |
106 bool test_was_run; | |
107 | |
108 // The sole purpose of this TEST is to enable us to check whether it | |
109 // was run. | |
110 TEST(FooTest, Bar) { | |
111 test_was_run = true; | |
112 } | |
113 | |
114 // Prints the message and aborts the program if condition is false. | |
115 void Check(bool condition, const char* msg) { | |
116 if (!condition) { | |
117 printf("FAILED: %s\n", msg); | |
118 testing::internal::posix::Abort(); | |
119 } | |
120 } | |
121 | |
122 // Runs the tests. Return true iff successful. | |
123 // | |
124 // The 'failure' parameter specifies the type of failure that should | |
125 // be generated by the global set-up. | |
126 int RunAllTests(MyEnvironment* env, FailureType failure) { | |
127 env->Reset(); | |
128 env->set_failure_in_set_up(failure); | |
129 test_was_run = false; | |
130 testing::internal::GetUnitTestImpl()->ClearAdHocTestResult(); | |
131 return RUN_ALL_TESTS(); | |
132 } | |
133 | |
134 } // namespace | |
135 | |
136 int main(int argc, char **argv) { | |
137 testing::InitGoogleTest(&argc, argv); | |
138 | |
139 // Registers a global test environment, and verifies that the | |
140 // registration function returns its argument. | |
141 MyEnvironment* const env = new MyEnvironment; | |
142 Check(testing::AddGlobalTestEnvironment(env) == env, | |
143 "AddGlobalTestEnvironment() should return its argument."); | |
144 | |
145 // Verifies that RUN_ALL_TESTS() runs the tests when the global | |
146 // set-up is successful. | |
147 Check(RunAllTests(env, NO_FAILURE) != 0, | |
148 "RUN_ALL_TESTS() should return non-zero, as the global tear-down " | |
149 "should generate a failure."); | |
150 Check(test_was_run, | |
151 "The tests should run, as the global set-up should generate no " | |
152 "failure"); | |
153 Check(env->tear_down_was_run(), | |
154 "The global tear-down should run, as the global set-up was run."); | |
155 | |
156 // Verifies that RUN_ALL_TESTS() runs the tests when the global | |
157 // set-up generates no fatal failure. | |
158 Check(RunAllTests(env, NON_FATAL_FAILURE) != 0, | |
159 "RUN_ALL_TESTS() should return non-zero, as both the global set-up " | |
160 "and the global tear-down should generate a non-fatal failure."); | |
161 Check(test_was_run, | |
162 "The tests should run, as the global set-up should generate no " | |
163 "fatal failure."); | |
164 Check(env->tear_down_was_run(), | |
165 "The global tear-down should run, as the global set-up was run."); | |
166 | |
167 // Verifies that RUN_ALL_TESTS() runs no test when the global set-up | |
168 // generates a fatal failure. | |
169 Check(RunAllTests(env, FATAL_FAILURE) != 0, | |
170 "RUN_ALL_TESTS() should return non-zero, as the global set-up " | |
171 "should generate a fatal failure."); | |
172 Check(!test_was_run, | |
173 "The tests should not run, as the global set-up should generate " | |
174 "a fatal failure."); | |
175 Check(env->tear_down_was_run(), | |
176 "The global tear-down should run, as the global set-up was run."); | |
177 | |
178 // Verifies that RUN_ALL_TESTS() doesn't do global set-up or | |
179 // tear-down when there is no test to run. | |
180 testing::GTEST_FLAG(filter) = "-*"; | |
181 Check(RunAllTests(env, NO_FAILURE) == 0, | |
182 "RUN_ALL_TESTS() should return zero, as there is no test to run."); | |
183 Check(!env->set_up_was_run(), | |
184 "The global set-up should not run, as there is no test to run."); | |
185 Check(!env->tear_down_was_run(), | |
186 "The global tear-down should not run, " | |
187 "as the global set-up was not run."); | |
188 | |
189 printf("PASS\n"); | |
190 return 0; | |
191 } | |
OLD | NEW |