#include <cstdint>

#include <iostream>
#include <stdexcept>

#include <adios2.h>

#include <gtest/gtest.h>

#define str_helper(X) #X
#define str(X) str_helper(X)

class XMLConfigTest : public ::testing::Test
{
public:
    XMLConfigTest() : configDir(str(XML_CONFIG_DIR)) {}

protected:
    // virtual void SetUp() { }

    // virtual void TearDown() { }
    std::string configDir;
};

TEST_F(XMLConfigTest, TwoIOs)
{
    std::string configFile = configDir + "/config1.xml";

#ifdef ADIOS2_HAVE_MPI
    adios2::ADIOS adios(configFile, MPI_COMM_WORLD, adios2::DebugON);
#else
    adios2::ADIOS adios(configFile, adios2::DebugON);
#endif

    EXPECT_NO_THROW({
        adios2::IO &io = adios.GetIO("Test IO 1");
        const adios2::Params &params = io.GetParameters();
        ASSERT_EQ(params.size(), 5);
        EXPECT_THROW(params.at("DoesNotExist"), std::out_of_range);
        EXPECT_EQ(params.at("Threads"), "1");
        EXPECT_EQ(params.at("ProfileUnits"), "Microseconds");
        EXPECT_EQ(params.at("MaxBufferSize"), "20Mb");
        EXPECT_EQ(params.at("InitialBufferSize"), "1Mb");
        EXPECT_EQ(params.at("BufferGrowthFactor"), "2");
        auto engine = io.Open("Test BP Writer 1", adios2::OpenMode::Write);
    });

    EXPECT_NO_THROW({
        adios2::IO &io = adios.GetIO("Test IO 2");
        const adios2::Params &params = io.GetParameters();
        ASSERT_EQ(params.size(), 0);
    });
}

TEST_F(XMLConfigTest, TwoEnginesException)
{
    std::string configFile = configDir + "/config2.xml";

#ifdef ADIOS2_HAVE_MPI
    EXPECT_THROW(
        adios2::ADIOS adios(configFile, MPI_COMM_WORLD, adios2::DebugON),
        std::invalid_argument);
#else
    EXPECT_THROW(adios2::ADIOS adios(configFile, adios2::DebugON),
                 std::invalid_argument);
#endif
}

int main(int argc, char **argv)
{
#ifdef ADIOS2_HAVE_MPI
    MPI_Init(nullptr, nullptr);
#endif

    ::testing::InitGoogleTest(&argc, argv);
    int result = RUN_ALL_TESTS();

#ifdef ADIOS2_HAVE_MPI
    MPI_Finalize();
#endif

    return result;
}