#include "gtest/gtest.h"

#include "RifWellPathImporter.h"

#include <QTemporaryFile>
#include <QTextStream>

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RimWellPathAsciiFileReaderTest, TestWellNameNoColon)
{
    QTemporaryFile file;
    if (file.open())
    {
        QString wellName = "My test Wellname";
        {
            QTextStream out(&file);
            out << "name " << wellName << "\n";
            out << "1 2 3";
        }

        RifWellPathImporter reader;
        RifWellPathImporter::WellData wpData = reader.readWellData(file.fileName(), 0);
        EXPECT_TRUE(wpData.m_name == wellName);
     }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RimWellPathAsciiFileReaderTest, TestWellNameWithColon)
{
    QTemporaryFile file;
    if (file.open())
    {
        QString wellName = "My test Wellname";
        {
            QTextStream out(&file);
            out << "WELLNAME:" << wellName << "\n";
            out << "1 2 3";
        }

        RifWellPathImporter reader;
        RifWellPathImporter::WellData wpData = reader.readWellData(file.fileName(), 0);
        EXPECT_TRUE(wpData.m_name == wellName);
     }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RimWellPathAsciiFileReaderTest, TestWellNameWithColonAndSpace)
{
    QTemporaryFile file;
    if (file.open())
    {
        QString wellName = "My test Wellname";
        {
            QTextStream out(&file);
            out << "WELLNAME  :  " << wellName << "\n";
            out << "1 2 3";
        }

        RifWellPathImporter reader;
        RifWellPathImporter::WellData wpData = reader.readWellData(file.fileName(), 0);
        EXPECT_TRUE(wpData.m_name == wellName);
     }
}