// Copyright 2007, Google Inc.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
//     * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
//     * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// Author: wan@google.com (Zhanyong Wan)
//
// Tests using global test environments.

#include <stdlib.h>
#include <stdio.h>
#include "gtest/gtest.h"

#define GTEST_IMPLEMENTATION_ 1  // Required for the next #include.
#include "src/gtest-internal-inl.h"
#undef GTEST_IMPLEMENTATION_

namespace testing
{
GTEST_DECLARE_string_(filter);
}

namespace
{

enum FailureType {
    NO_FAILURE, NON_FATAL_FAILURE, FATAL_FAILURE
};

// For testing using global test environments.
class MyEnvironment : public testing::Environment
{
    public:
        MyEnvironment() { Reset(); }

        // Depending on the value of failure_in_set_up_, SetUp() will
        // generate a non-fatal failure, generate a fatal failure, or
        // succeed.
        virtual void SetUp() {
            set_up_was_run_ = true;

            switch (failure_in_set_up_) {
                case NON_FATAL_FAILURE:
                    ADD_FAILURE() << "Expected non-fatal failure in global set-up.";
                    break;
                case FATAL_FAILURE:
                    FAIL() << "Expected fatal failure in global set-up.";
                    break;
                default:
                    break;
            }
        }

        // Generates a non-fatal failure.
        virtual void TearDown() {
            tear_down_was_run_ = true;
            ADD_FAILURE() << "Expected non-fatal failure in global tear-down.";
        }

        // Resets the state of the environment s.t. it can be reused.
        void Reset() {
            failure_in_set_up_ = NO_FAILURE;
            set_up_was_run_ = false;
            tear_down_was_run_ = false;
        }

        // We call this function to set the type of failure SetUp() should
        // generate.
        void set_failure_in_set_up(FailureType type) {
            failure_in_set_up_ = type;
        }

        // Was SetUp() run?
        bool set_up_was_run() const { return set_up_was_run_; }

        // Was TearDown() run?
        bool tear_down_was_run() const { return tear_down_was_run_; }

    private:
        FailureType failure_in_set_up_;
        bool set_up_was_run_;
        bool tear_down_was_run_;
};

// Was the TEST run?
bool test_was_run;

// The sole purpose of this TEST is to enable us to check whether it
// was run.
TEST(FooTest, Bar)
{
    test_was_run = true;
}

// Prints the message and aborts the program if condition is false.
void Check(bool condition, const char* msg)
{
    if (!condition) {
        printf("FAILED: %s\n", msg);
        testing::internal::posix::Abort();
    }
}

// Runs the tests.  Return true iff successful.
//
// The 'failure' parameter specifies the type of failure that should
// be generated by the global set-up.
int RunAllTests(MyEnvironment* env, FailureType failure)
{
    env->Reset();
    env->set_failure_in_set_up(failure);
    test_was_run = false;
    testing::internal::GetUnitTestImpl()->ClearAdHocTestResult();
    return RUN_ALL_TESTS();
}

}  // namespace

int main(int argc, char** argv)
{
    testing::InitGoogleTest(&argc, argv);

    // Registers a global test environment, and verifies that the
    // registration function returns its argument.
    MyEnvironment* const env = new MyEnvironment;
    Check(testing::AddGlobalTestEnvironment(env) == env,
          "AddGlobalTestEnvironment() should return its argument.");

    // Verifies that RUN_ALL_TESTS() runs the tests when the global
    // set-up is successful.
    Check(RunAllTests(env, NO_FAILURE) != 0,
          "RUN_ALL_TESTS() should return non-zero, as the global tear-down "
          "should generate a failure.");
    Check(test_was_run,
          "The tests should run, as the global set-up should generate no "
          "failure");
    Check(env->tear_down_was_run(),
          "The global tear-down should run, as the global set-up was run.");

    // Verifies that RUN_ALL_TESTS() runs the tests when the global
    // set-up generates no fatal failure.
    Check(RunAllTests(env, NON_FATAL_FAILURE) != 0,
          "RUN_ALL_TESTS() should return non-zero, as both the global set-up "
          "and the global tear-down should generate a non-fatal failure.");
    Check(test_was_run,
          "The tests should run, as the global set-up should generate no "
          "fatal failure.");
    Check(env->tear_down_was_run(),
          "The global tear-down should run, as the global set-up was run.");

    // Verifies that RUN_ALL_TESTS() runs no test when the global set-up
    // generates a fatal failure.
    Check(RunAllTests(env, FATAL_FAILURE) != 0,
          "RUN_ALL_TESTS() should return non-zero, as the global set-up "
          "should generate a fatal failure.");
    Check(!test_was_run,
          "The tests should not run, as the global set-up should generate "
          "a fatal failure.");
    Check(env->tear_down_was_run(),
          "The global tear-down should run, as the global set-up was run.");

    // Verifies that RUN_ALL_TESTS() doesn't do global set-up or
    // tear-down when there is no test to run.
    testing::GTEST_FLAG(filter) = "-*";
    Check(RunAllTests(env, NO_FAILURE) == 0,
          "RUN_ALL_TESTS() should return zero, as there is no test to run.");
    Check(!env->set_up_was_run(),
          "The global set-up should not run, as there is no test to run.");
    Check(!env->tear_down_was_run(),
          "The global tear-down should not run, "
          "as the global set-up was not run.");

    printf("PASS\n");
    return 0;
}
