113481Sgiacomo.travaglini@arm.com// Copyright 2007, Google Inc. 213481Sgiacomo.travaglini@arm.com// All rights reserved. 313481Sgiacomo.travaglini@arm.com// 413481Sgiacomo.travaglini@arm.com// Redistribution and use in source and binary forms, with or without 513481Sgiacomo.travaglini@arm.com// modification, are permitted provided that the following conditions are 613481Sgiacomo.travaglini@arm.com// met: 713481Sgiacomo.travaglini@arm.com// 813481Sgiacomo.travaglini@arm.com// * Redistributions of source code must retain the above copyright 913481Sgiacomo.travaglini@arm.com// notice, this list of conditions and the following disclaimer. 1013481Sgiacomo.travaglini@arm.com// * Redistributions in binary form must reproduce the above 1113481Sgiacomo.travaglini@arm.com// copyright notice, this list of conditions and the following disclaimer 1213481Sgiacomo.travaglini@arm.com// in the documentation and/or other materials provided with the 1313481Sgiacomo.travaglini@arm.com// distribution. 1413481Sgiacomo.travaglini@arm.com// * Neither the name of Google Inc. nor the names of its 1513481Sgiacomo.travaglini@arm.com// contributors may be used to endorse or promote products derived from 1613481Sgiacomo.travaglini@arm.com// this software without specific prior written permission. 1713481Sgiacomo.travaglini@arm.com// 1813481Sgiacomo.travaglini@arm.com// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 1913481Sgiacomo.travaglini@arm.com// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 2013481Sgiacomo.travaglini@arm.com// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 2113481Sgiacomo.travaglini@arm.com// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2213481Sgiacomo.travaglini@arm.com// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2313481Sgiacomo.travaglini@arm.com// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 2413481Sgiacomo.travaglini@arm.com// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2513481Sgiacomo.travaglini@arm.com// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2613481Sgiacomo.travaglini@arm.com// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2713481Sgiacomo.travaglini@arm.com// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 2813481Sgiacomo.travaglini@arm.com// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 2913481Sgiacomo.travaglini@arm.com// 3013481Sgiacomo.travaglini@arm.com// Author: wan@google.com (Zhanyong Wan) 3113481Sgiacomo.travaglini@arm.com// 3213481Sgiacomo.travaglini@arm.com// Tests using global test environments. 3313481Sgiacomo.travaglini@arm.com 3413481Sgiacomo.travaglini@arm.com#include <stdlib.h> 3513481Sgiacomo.travaglini@arm.com#include <stdio.h> 3613481Sgiacomo.travaglini@arm.com#include "gtest/gtest.h" 3713481Sgiacomo.travaglini@arm.com 3813481Sgiacomo.travaglini@arm.com#define GTEST_IMPLEMENTATION_ 1 // Required for the next #include. 3913481Sgiacomo.travaglini@arm.com#include "src/gtest-internal-inl.h" 4013481Sgiacomo.travaglini@arm.com#undef GTEST_IMPLEMENTATION_ 4113481Sgiacomo.travaglini@arm.com 4213481Sgiacomo.travaglini@arm.comnamespace testing { 4313481Sgiacomo.travaglini@arm.comGTEST_DECLARE_string_(filter); 4413481Sgiacomo.travaglini@arm.com} 4513481Sgiacomo.travaglini@arm.com 4613481Sgiacomo.travaglini@arm.comnamespace { 4713481Sgiacomo.travaglini@arm.com 4813481Sgiacomo.travaglini@arm.comenum FailureType { 4913481Sgiacomo.travaglini@arm.com NO_FAILURE, NON_FATAL_FAILURE, FATAL_FAILURE 5013481Sgiacomo.travaglini@arm.com}; 5113481Sgiacomo.travaglini@arm.com 5213481Sgiacomo.travaglini@arm.com// For testing using global test environments. 5313481Sgiacomo.travaglini@arm.comclass MyEnvironment : public testing::Environment { 5413481Sgiacomo.travaglini@arm.com public: 5513481Sgiacomo.travaglini@arm.com MyEnvironment() { Reset(); } 5613481Sgiacomo.travaglini@arm.com 5713481Sgiacomo.travaglini@arm.com // Depending on the value of failure_in_set_up_, SetUp() will 5813481Sgiacomo.travaglini@arm.com // generate a non-fatal failure, generate a fatal failure, or 5913481Sgiacomo.travaglini@arm.com // succeed. 6013481Sgiacomo.travaglini@arm.com virtual void SetUp() { 6113481Sgiacomo.travaglini@arm.com set_up_was_run_ = true; 6213481Sgiacomo.travaglini@arm.com 6313481Sgiacomo.travaglini@arm.com switch (failure_in_set_up_) { 6413481Sgiacomo.travaglini@arm.com case NON_FATAL_FAILURE: 6513481Sgiacomo.travaglini@arm.com ADD_FAILURE() << "Expected non-fatal failure in global set-up."; 6613481Sgiacomo.travaglini@arm.com break; 6713481Sgiacomo.travaglini@arm.com case FATAL_FAILURE: 6813481Sgiacomo.travaglini@arm.com FAIL() << "Expected fatal failure in global set-up."; 6913481Sgiacomo.travaglini@arm.com break; 7013481Sgiacomo.travaglini@arm.com default: 7113481Sgiacomo.travaglini@arm.com break; 7213481Sgiacomo.travaglini@arm.com } 7313481Sgiacomo.travaglini@arm.com } 7413481Sgiacomo.travaglini@arm.com 7513481Sgiacomo.travaglini@arm.com // Generates a non-fatal failure. 7613481Sgiacomo.travaglini@arm.com virtual void TearDown() { 7713481Sgiacomo.travaglini@arm.com tear_down_was_run_ = true; 7813481Sgiacomo.travaglini@arm.com ADD_FAILURE() << "Expected non-fatal failure in global tear-down."; 7913481Sgiacomo.travaglini@arm.com } 8013481Sgiacomo.travaglini@arm.com 8113481Sgiacomo.travaglini@arm.com // Resets the state of the environment s.t. it can be reused. 8213481Sgiacomo.travaglini@arm.com void Reset() { 8313481Sgiacomo.travaglini@arm.com failure_in_set_up_ = NO_FAILURE; 8413481Sgiacomo.travaglini@arm.com set_up_was_run_ = false; 8513481Sgiacomo.travaglini@arm.com tear_down_was_run_ = false; 8613481Sgiacomo.travaglini@arm.com } 8713481Sgiacomo.travaglini@arm.com 8813481Sgiacomo.travaglini@arm.com // We call this function to set the type of failure SetUp() should 8913481Sgiacomo.travaglini@arm.com // generate. 9013481Sgiacomo.travaglini@arm.com void set_failure_in_set_up(FailureType type) { 9113481Sgiacomo.travaglini@arm.com failure_in_set_up_ = type; 9213481Sgiacomo.travaglini@arm.com } 9313481Sgiacomo.travaglini@arm.com 9413481Sgiacomo.travaglini@arm.com // Was SetUp() run? 9513481Sgiacomo.travaglini@arm.com bool set_up_was_run() const { return set_up_was_run_; } 9613481Sgiacomo.travaglini@arm.com 9713481Sgiacomo.travaglini@arm.com // Was TearDown() run? 9813481Sgiacomo.travaglini@arm.com bool tear_down_was_run() const { return tear_down_was_run_; } 9913481Sgiacomo.travaglini@arm.com 10013481Sgiacomo.travaglini@arm.com private: 10113481Sgiacomo.travaglini@arm.com FailureType failure_in_set_up_; 10213481Sgiacomo.travaglini@arm.com bool set_up_was_run_; 10313481Sgiacomo.travaglini@arm.com bool tear_down_was_run_; 10413481Sgiacomo.travaglini@arm.com}; 10513481Sgiacomo.travaglini@arm.com 10613481Sgiacomo.travaglini@arm.com// Was the TEST run? 10713481Sgiacomo.travaglini@arm.combool test_was_run; 10813481Sgiacomo.travaglini@arm.com 10913481Sgiacomo.travaglini@arm.com// The sole purpose of this TEST is to enable us to check whether it 11013481Sgiacomo.travaglini@arm.com// was run. 11113481Sgiacomo.travaglini@arm.comTEST(FooTest, Bar) { 11213481Sgiacomo.travaglini@arm.com test_was_run = true; 11313481Sgiacomo.travaglini@arm.com} 11413481Sgiacomo.travaglini@arm.com 11513481Sgiacomo.travaglini@arm.com// Prints the message and aborts the program if condition is false. 11613481Sgiacomo.travaglini@arm.comvoid Check(bool condition, const char* msg) { 11713481Sgiacomo.travaglini@arm.com if (!condition) { 11813481Sgiacomo.travaglini@arm.com printf("FAILED: %s\n", msg); 11913481Sgiacomo.travaglini@arm.com testing::internal::posix::Abort(); 12013481Sgiacomo.travaglini@arm.com } 12113481Sgiacomo.travaglini@arm.com} 12213481Sgiacomo.travaglini@arm.com 12313481Sgiacomo.travaglini@arm.com// Runs the tests. Return true iff successful. 12413481Sgiacomo.travaglini@arm.com// 12513481Sgiacomo.travaglini@arm.com// The 'failure' parameter specifies the type of failure that should 12613481Sgiacomo.travaglini@arm.com// be generated by the global set-up. 12713481Sgiacomo.travaglini@arm.comint RunAllTests(MyEnvironment* env, FailureType failure) { 12813481Sgiacomo.travaglini@arm.com env->Reset(); 12913481Sgiacomo.travaglini@arm.com env->set_failure_in_set_up(failure); 13013481Sgiacomo.travaglini@arm.com test_was_run = false; 13113481Sgiacomo.travaglini@arm.com testing::internal::GetUnitTestImpl()->ClearAdHocTestResult(); 13213481Sgiacomo.travaglini@arm.com return RUN_ALL_TESTS(); 13313481Sgiacomo.travaglini@arm.com} 13413481Sgiacomo.travaglini@arm.com 13513481Sgiacomo.travaglini@arm.com} // namespace 13613481Sgiacomo.travaglini@arm.com 13713481Sgiacomo.travaglini@arm.comint main(int argc, char **argv) { 13813481Sgiacomo.travaglini@arm.com testing::InitGoogleTest(&argc, argv); 13913481Sgiacomo.travaglini@arm.com 14013481Sgiacomo.travaglini@arm.com // Registers a global test environment, and verifies that the 14113481Sgiacomo.travaglini@arm.com // registration function returns its argument. 14213481Sgiacomo.travaglini@arm.com MyEnvironment* const env = new MyEnvironment; 14313481Sgiacomo.travaglini@arm.com Check(testing::AddGlobalTestEnvironment(env) == env, 14413481Sgiacomo.travaglini@arm.com "AddGlobalTestEnvironment() should return its argument."); 14513481Sgiacomo.travaglini@arm.com 14613481Sgiacomo.travaglini@arm.com // Verifies that RUN_ALL_TESTS() runs the tests when the global 14713481Sgiacomo.travaglini@arm.com // set-up is successful. 14813481Sgiacomo.travaglini@arm.com Check(RunAllTests(env, NO_FAILURE) != 0, 14913481Sgiacomo.travaglini@arm.com "RUN_ALL_TESTS() should return non-zero, as the global tear-down " 15013481Sgiacomo.travaglini@arm.com "should generate a failure."); 15113481Sgiacomo.travaglini@arm.com Check(test_was_run, 15213481Sgiacomo.travaglini@arm.com "The tests should run, as the global set-up should generate no " 15313481Sgiacomo.travaglini@arm.com "failure"); 15413481Sgiacomo.travaglini@arm.com Check(env->tear_down_was_run(), 15513481Sgiacomo.travaglini@arm.com "The global tear-down should run, as the global set-up was run."); 15613481Sgiacomo.travaglini@arm.com 15713481Sgiacomo.travaglini@arm.com // Verifies that RUN_ALL_TESTS() runs the tests when the global 15813481Sgiacomo.travaglini@arm.com // set-up generates no fatal failure. 15913481Sgiacomo.travaglini@arm.com Check(RunAllTests(env, NON_FATAL_FAILURE) != 0, 16013481Sgiacomo.travaglini@arm.com "RUN_ALL_TESTS() should return non-zero, as both the global set-up " 16113481Sgiacomo.travaglini@arm.com "and the global tear-down should generate a non-fatal failure."); 16213481Sgiacomo.travaglini@arm.com Check(test_was_run, 16313481Sgiacomo.travaglini@arm.com "The tests should run, as the global set-up should generate no " 16413481Sgiacomo.travaglini@arm.com "fatal failure."); 16513481Sgiacomo.travaglini@arm.com Check(env->tear_down_was_run(), 16613481Sgiacomo.travaglini@arm.com "The global tear-down should run, as the global set-up was run."); 16713481Sgiacomo.travaglini@arm.com 16813481Sgiacomo.travaglini@arm.com // Verifies that RUN_ALL_TESTS() runs no test when the global set-up 16913481Sgiacomo.travaglini@arm.com // generates a fatal failure. 17013481Sgiacomo.travaglini@arm.com Check(RunAllTests(env, FATAL_FAILURE) != 0, 17113481Sgiacomo.travaglini@arm.com "RUN_ALL_TESTS() should return non-zero, as the global set-up " 17213481Sgiacomo.travaglini@arm.com "should generate a fatal failure."); 17313481Sgiacomo.travaglini@arm.com Check(!test_was_run, 17413481Sgiacomo.travaglini@arm.com "The tests should not run, as the global set-up should generate " 17513481Sgiacomo.travaglini@arm.com "a fatal failure."); 17613481Sgiacomo.travaglini@arm.com Check(env->tear_down_was_run(), 17713481Sgiacomo.travaglini@arm.com "The global tear-down should run, as the global set-up was run."); 17813481Sgiacomo.travaglini@arm.com 17913481Sgiacomo.travaglini@arm.com // Verifies that RUN_ALL_TESTS() doesn't do global set-up or 18013481Sgiacomo.travaglini@arm.com // tear-down when there is no test to run. 18113481Sgiacomo.travaglini@arm.com testing::GTEST_FLAG(filter) = "-*"; 18213481Sgiacomo.travaglini@arm.com Check(RunAllTests(env, NO_FAILURE) == 0, 18313481Sgiacomo.travaglini@arm.com "RUN_ALL_TESTS() should return zero, as there is no test to run."); 18413481Sgiacomo.travaglini@arm.com Check(!env->set_up_was_run(), 18513481Sgiacomo.travaglini@arm.com "The global set-up should not run, as there is no test to run."); 18613481Sgiacomo.travaglini@arm.com Check(!env->tear_down_was_run(), 18713481Sgiacomo.travaglini@arm.com "The global tear-down should not run, " 18813481Sgiacomo.travaglini@arm.com "as the global set-up was not run."); 18913481Sgiacomo.travaglini@arm.com 19013481Sgiacomo.travaglini@arm.com printf("PASS\n"); 19113481Sgiacomo.travaglini@arm.com return 0; 19213481Sgiacomo.travaglini@arm.com} 193