#include "gtest/gtest.h"

#include "cafPdmUiTreeSelectionQModel.h"



QList<caf::PdmOptionItemInfo> createOptions()
{
    QList<caf::PdmOptionItemInfo> options;

    {
        QString text;

        text = "First";
        options.push_back(caf::PdmOptionItemInfo(text, text));

        text = "Second";
        options.push_back(caf::PdmOptionItemInfo(text, text));


        {
            text = "Second_a";
            caf::PdmOptionItemInfo itemInfo = caf::PdmOptionItemInfo(text, text);
            itemInfo.setLevel(1);
            options.push_back(itemInfo);
        }

        {
            text = "Second_b";
            caf::PdmOptionItemInfo itemInfo = caf::PdmOptionItemInfo(text, text);
            itemInfo.setLevel(1);
            options.push_back(itemInfo);
        }


        text = "Third";
        options.push_back(caf::PdmOptionItemInfo(text, text));

        text = "Fourth";
        options.push_back(caf::PdmOptionItemInfo(text, text));
    }

    return options;
}




//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(PdmUiTreeSelectionQModelTest, BasicUsage)
{
    QList<caf::PdmOptionItemInfo> options = createOptions();

    caf::PdmUiTreeSelectionQModel myModel;
    myModel.setOptions(nullptr, options);

    EXPECT_EQ(options.size(), myModel.optionItemCount());

    EXPECT_EQ(4, myModel.rowCount(myModel.index(-1, -1)));
    
    EXPECT_EQ(0, myModel.rowCount(myModel.index(0, 0)));
    EXPECT_EQ(2, myModel.rowCount(myModel.index(1, 0)));
    EXPECT_EQ(0, myModel.rowCount(myModel.index(2, 0)));
    EXPECT_EQ(0, myModel.rowCount(myModel.index(3, 0)));

    // Test for row out of bounds
    EXPECT_FALSE(myModel.index(4, 0).isValid());
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(PdmUiTreeSelectionQModelTest, ParentBehaviour)
{
    QList<caf::PdmOptionItemInfo> options = createOptions();

    caf::PdmUiTreeSelectionQModel myModel;
    myModel.setOptions(nullptr, options);

    QModelIndex parentIndex = myModel.index(1, 0);
    EXPECT_EQ(2, myModel.rowCount(parentIndex));

    {
        QModelIndex firstChildIndex = myModel.index(0, 0, parentIndex);
        EXPECT_STREQ("Second_a", myModel.data(firstChildIndex).toString().toLatin1());
        EXPECT_TRUE(parentIndex == myModel.parent(firstChildIndex));
    }

    {
        QModelIndex secondChildIndex = myModel.index(1, 0, parentIndex);
        EXPECT_STREQ("Second_b", myModel.data(secondChildIndex).toString().toLatin1());
        EXPECT_TRUE(parentIndex == myModel.parent(secondChildIndex));
    }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(PdmUiTreeSelectionQModelTest, SetDataAndSignal)
{
    QList<caf::PdmOptionItemInfo> options = createOptions();

    caf::PdmUiTreeSelectionQModel myModel;
    myModel.setOptions(nullptr, options);

    QModelIndex parentIndex = myModel.index(0, 0);

    myModel.setData(parentIndex, QVariant(true));
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
TEST(PdmUiTreeSelectionQModelTest, SetCheckedStateForItems)
{
    QList<caf::PdmOptionItemInfo> options = createOptions();

    caf::PdmUiTreeSelectionQModel myModel;
    myModel.setOptions(nullptr, options);

    QModelIndex parentIndex = myModel.index(1, 0);
    QModelIndex firstChildIndex = myModel.index(0, 0, parentIndex);

    QModelIndexList indices;
    indices << firstChildIndex;

    myModel.setCheckedStateForItems(indices, false);

    // No test code for this behaviour, only making sure this runs without any errors
}