#include "gtest/gtest.h"

#include "RiaTestDataDirectory.h"

#include "RifElementPropertyTableReader.h"

#include <QString>
#include <numeric>


static const QString TEST_DATA_DIRECTORY = QString("%1/RifElementPropertyTableReader/").arg(TEST_DATA_DIR);

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RicElementPropertyTableReaderTest, BasicUsage)
{
    RifElementPropertyMetadata metadata = RifElementPropertyTableReader::readMetadata(TEST_DATA_DIRECTORY + "ELASTIC_TABLE.inp");

    RifElementPropertyTable table;
    RifElementPropertyTableReader::readData(&metadata, &table);

    EXPECT_TRUE(table.hasData);

    EXPECT_EQ(2, metadata.dataColumns.size());
    EXPECT_STREQ("MODULUS", metadata.dataColumns[0].toStdString().c_str());
    EXPECT_STREQ("RATIO", metadata.dataColumns[1].toStdString().c_str());

    EXPECT_EQ(2, table.data.size());
    EXPECT_EQ(4320, table.elementIds.size());
    EXPECT_EQ(4320, table.data[0].size());
    EXPECT_EQ(4320, table.data[1].size());
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RicElementPropertyTableReaderTest, ParseFailed)
{
    try
    {
        RifElementPropertyMetadata metadata = RifElementPropertyTableReader::readMetadata(TEST_DATA_DIRECTORY + "ELASTIC_TABLE_error.inp");

        RifElementPropertyTable table;
        RifElementPropertyTableReader::readData(&metadata, &table);

        EXPECT_TRUE(false);
    }
    catch (FileParseException e)
    {
        EXPECT_TRUE(e.message.startsWith("Number of columns mismatch"));
    }
}