/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2011-     Statoil ASA
//  Copyright (C) 2013-     Ceetron Solutions AS
//  Copyright (C) 2011-2012 Ceetron AS
// 
//  ResInsight is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
// 
//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
//  FITNESS FOR A PARTICULAR PURPOSE.
// 
//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html> 
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#include "gtest/gtest.h"

#include "RigActiveCellInfo.h"


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RigActiveCellInfo, BasicTest)
{
    RigActiveCellInfo rigActiveCellInfo;

    size_t globalActiveCellCount = 10;
    rigActiveCellInfo.setReservoirCellCount(globalActiveCellCount);

    for (size_t i = 0; i < globalActiveCellCount; i++)
    {
        EXPECT_TRUE(rigActiveCellInfo.cellResultIndex(i) == cvf::UNDEFINED_SIZE_T);
        EXPECT_FALSE(rigActiveCellInfo.isActive(i));
    }

    rigActiveCellInfo.setCellResultIndex(3, 1);
    EXPECT_TRUE(rigActiveCellInfo.cellResultIndex(3) == 1);
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(RigActiveCellInfo, GridCellCounts)
{
    {
        RigActiveCellInfo rigActiveCellInfo;
        rigActiveCellInfo.setGridCount(3);
        rigActiveCellInfo.setGridActiveCellCounts(0, 0);
        rigActiveCellInfo.setGridActiveCellCounts(1, 1);
        rigActiveCellInfo.setGridActiveCellCounts(2, 2);
        rigActiveCellInfo.computeDerivedData();

        EXPECT_TRUE(rigActiveCellInfo.reservoirActiveCellCount() == 3);
    }

    {
        RigActiveCellInfo rigActiveCellInfo;
        rigActiveCellInfo.setGridCount(3);
        rigActiveCellInfo.setGridActiveCellCounts(0, 3 );
        rigActiveCellInfo.setGridActiveCellCounts(1, 4 );
        rigActiveCellInfo.setGridActiveCellCounts(2, 5 );
        rigActiveCellInfo.computeDerivedData();

        EXPECT_TRUE(rigActiveCellInfo.reservoirActiveCellCount() == 12);
    }
}