| 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 |