#pragma once

#include "gtest/gtest.h"

#include "storm/adapters/RationalNumberAdapter.h"
#include "storm/utility/constants.h"
#include "storm/utility/initialize.h"


namespace testing {
    namespace internal {
    
        inline GTEST_API_ AssertionResult DoubleNearPredFormat(const char* expr1,
                                                const char* expr2,
                                                const char* abs_error_expr,
                                                storm::RationalNumber val1,
                                                storm::RationalNumber val2,
                                                storm::RationalNumber abs_error) {
            const storm::RationalNumber diff = storm::utility::abs<storm::RationalNumber>(val1 - val2);
            if (diff <= abs_error) return AssertionSuccess();
            return AssertionFailure()
                  << "The difference between " << expr1 << " and " << expr2
                  << " is " << diff << " (approx. " << storm::utility::convertNumber<double>(diff) << "), which exceeds " << abs_error_expr << ", where\n"
                  << expr1 << " evaluates to " << val1 << " (approx. " << storm::utility::convertNumber<double>(val1) << "),\n"
                  << expr2 << " evaluates to " << val2 << " (approx. " << storm::utility::convertNumber<double>(val2) << "),\n"
                  << abs_error_expr << " evaluates to " << abs_error << " (approx. " << storm::utility::convertNumber<double>(abs_error) << ").";
        }
    }
}

namespace storm {
    namespace test {
        inline void initialize() {
            storm::utility::initializeLogger();
            // Only enable error output by default.
            storm::utility::setLogLevel(l3pp::LogLevel::ERR);
        }
        
        inline void enableErrorOutput() {
            // Only decrease the log level
            if (storm::utility::getLogLevel() > l3pp::LogLevel::ERR) {
                storm::utility::setLogLevel(l3pp::LogLevel::ERR);
            }
        }
        
        inline void disableOutput() {
            storm::utility::setLogLevel(l3pp::LogLevel::OFF);
        }
    }
}

// Some tests have to be skipped for specific z3 versions because of a bug that was present in z3.
#ifdef STORM_HAVE_Z3
#include <z3.h>
namespace storm {
    namespace test {
        inline bool z3AtLeastVersion(unsigned expectedMajor, unsigned expectedMinor, unsigned expectedBuildNumber) {
            std::vector<unsigned> actual(4), expected({expectedMajor, expectedMinor, expectedBuildNumber, 0u});
            Z3_get_version(&actual[0], &actual[1], &actual[2], &actual[3]);
            for (uint64_t i = 0; i < 4; ++i) {
                if (actual[i] > expected[i]) {
                    return true;
                }
                if (actual[i] < expected[i]) {
                    return false;
                }
            }
            return true; // Equal versions
        }
    }
}
#endif


#define STORM_SILENT_ASSERT_THROW(statement, expected_exception) \
    storm::test::disableOutput(); \
    ASSERT_THROW(statement, expected_exception); \
    storm::test::enableErrorOutput()
    
#define STORM_SILENT_EXPECT_THROW(statement, expected_exception) \
    storm::test::disableOutput(); \
    EXPECT_THROW(statement, expected_exception); \
    storm::test::enableErrorOutput()