/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: Unit test for mpcollectiondatamodel

*

*/

#include <QAbstractItemModel>

#include <QSignalSpy>

#include <QMetaType>

#include <QTranslator>

#include <QLocale>

#include <hbapplication.h>

#include <hbmainwindow.h>

#include <hbicon.h>

#include <hbnamespace.h>

#include "unittest_mpcollectiondatamodel.h"

#include "unittest_helper.h"

#include "stub/inc/mpcollectionalbumartmanager.h"

#include "stub/inc/mpmpxcollectiondata.h"

#include "stub/inc/mpplaybackdata.h"

// Do this so we can access all member variables.

#define private public

#include "mpcollectiondatamodel.h"

#undef private

/*!

 Make our test case a stand-alone executable that runs all the test functions.

 */

int main(int argc, char *argv[])

{

    HbApplication app(argc, argv);

    TestMpCollectionDataModel tv;

}

TestMpCollectionDataModel::TestMpCollectionDataModel()

    : mTest(0),

      mHelper(0),

      mStubData(0),

      mStubPlaybackData(0),

      mMpTranslator(0)

{

}

TestMpCollectionDataModel::~TestMpCollectionDataModel()

{

    delete mTest;

    delete mHelper;

    delete mStubData;

    delete mStubPlaybackData;

    delete mMpTranslator;

}

/*!

 Called before the first testfunction is executed.

 */

void TestMpCollectionDataModel::initTestCase()

{

    QString lang = QLocale::system().name();

    QString path = QString("z:/resource/qt/translations/");

    bool translatorLoaded = false;

    mMpTranslator = new QTranslator(this);

    translatorLoaded = mMpTranslator->load(path + "musicplayer_" + lang);

    if ( translatorLoaded ) {

        qApp->installTranslator(mMpTranslator);

    }

    mStubData = new MpMpxCollectionData();

    mHelper = new TestHelper();

}

/*!

 Called after the last testfunction was executed.

 */

void TestMpCollectionDataModel::cleanupTestCase()

{

    delete mStubData;

    mStubData = 0;

    delete mStubPlaybackData;

    mStubPlaybackData = 0;

    delete mHelper;

    mHelper = 0;

}

/*!

 Called before each testfunction is executed.

 */

void TestMpCollectionDataModel::init()

{

    mTest = new MpCollectionDataModel(mStubData, mStubPlaybackData);

    MpMpxCollectionData::resetItemDataCounter();

    mTest->mCollectionData->mItemDataReturn = true;

    mTest->mCollectionData->mCollectionTitleReturn = true;

}

/*!

 Called after every testfunction.

 */

void TestMpCollectionDataModel::cleanup()

{

    delete mTest;

    mTest = 0;

}

/*!

 Tests correct cleanup of member variables.

 */

void TestMpCollectionDataModel::testMemberCleanup()

{

    cleanup();

    MpCollectionAlbumArtManager::resetInitCounter();

    int count = MpMpxCollectionData::getInitCounter();

    init();

    cleanup();

    QCOMPARE(MpCollectionAlbumArtManager::getInitCounter(), 0);

    QCOMPARE(MpMpxCollectionData::getInitCounter(), count);

}

/*!

 Tests refreshModel() request.

 */

void TestMpCollectionDataModel::testRefreshModel()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextAlbums;

    mTest->refreshModel();

    // Verify that:

    // - It sets correct row count

    QCOMPARE(mTest->mRowCount, 100);

    QCOMPARE(mTest->mAlbumArtManager->mCacheAlbumArtCount, 1);

    mTest->mCollectionData->mCount = 50;

    mTest->mCollectionData->mContext = ECollectionContextArtistAlbums;

    mTest->refreshModel();

    // Verify that:

    // - It sets correct row count

    QCOMPARE(mTest->mRowCount, 50);

    QCOMPARE(mTest->mAlbumArtManager->mCacheAlbumArtCount, 2);

    mTest->mCollectionData->mCount = 150;

    mTest->mCollectionData->mContext = ECollectionContextAlbumsMediaWall;

    mTest->refreshModel();

    // Verify that:

    // - It sets correct row count

    QCOMPARE(mTest->mRowCount, 150);

    QCOMPARE(mTest->mAlbumArtManager->mCacheAlbumArtCount, 3);

    mTest->mCollectionData->mCount = 200;

    mTest->mCollectionData->mContext = ECollectionContextAllSongs;

    mTest->refreshModel();

    // Verify that:

    // - It sets correct row count

    // - It shouldn't trigger first screen cache

    QCOMPARE(mTest->mRowCount, 200);

    QCOMPARE(mTest->mAlbumArtManager->mCacheAlbumArtCount, 3);

}

/*!

 Tests collectionData() request.

 */

void TestMpCollectionDataModel::testCollectionData()

{

    QVERIFY(mTest->collectionData() == mStubData);

}

/*!

 Tests data() request for AllSongs context.

 */

void TestMpCollectionDataModel::testDataAllSongs()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextAllSongs;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 2);

    QCOMPARE(dataList.at(0), QString("Title1"));

    QCOMPARE(dataList.at(1), QString("Artist1"));

    // Qt::DecorationRole

    data = mTest->data(modelIndex, Qt::DecorationRole);

    QCOMPARE(data.isNull(), true);

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString("Title1"));

}

/*!

 Tests data() request for AllSongs context with no data available.

 */

void TestMpCollectionDataModel::testDataAllSongsNoData()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mItemDataReturn = false;

    mTest->mCollectionData->mContext = ECollectionContextAllSongs;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    // Qt::DecorationRole

    data = mTest->data(modelIndex, Qt::DecorationRole);

    QCOMPARE(data.isNull(), true);

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString(""));

}

/*!

 Tests data() request for Artists context.

 */

void TestMpCollectionDataModel::testDataArtists()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextArtists;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 1);

    QCOMPARE(dataList.at(0), QString("Title1"));

    // Qt::DecorationRole

    data = mTest->data(modelIndex, Qt::DecorationRole);

    QCOMPARE(data.isNull(), true);

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString("Title1"));

}

/*!

 Tests data() request for Artists context with no data available.

 */

void TestMpCollectionDataModel::testDataArtistsNoData()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mItemDataReturn = false;

    mTest->mCollectionData->mContext = ECollectionContextArtists;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 1);

    QCOMPARE(dataList.at(0), hbTrId("txt_mus_other_unknown4"));

    // Qt::DecorationRole

    data = mTest->data(modelIndex, Qt::DecorationRole);

    QCOMPARE(data.isNull(), true);

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString(""));

}

/*!

 Tests data() request for ArtistAlbums context.

 */

void TestMpCollectionDataModel::testDataArtistAlbums()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextArtistAlbums;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 1);

    QCOMPARE(dataList.at(0), QString("Title1"));

    // Qt::DecorationRole

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString("Title1"));

}

/*!

 Tests data() request for ArtistAlbums context with no data available.

 */

void TestMpCollectionDataModel::testDataArtistAlbumsNoData()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mItemDataReturn = false;

    mTest->mCollectionData->mContext = ECollectionContextArtistAlbums;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 1);

    QCOMPARE(dataList.at(0), hbTrId("txt_mus_other_unknown4"));

    // Qt::DecorationRole

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString(""));

}

/*!

 Tests data() request for ArtistAlbumsTBone context.

 */

void TestMpCollectionDataModel::testDataArtistAlbumsTBone()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextArtistAlbumsTBone;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 2);

    QCOMPARE(dataList.at(0), QString("Title1"));

    QCOMPARE(dataList.at(1), QString("CollectionTitle"));

    // Qt::DecorationRole

}

/*!

 Tests data() request for ArtistAlbumsTBone context with no data available.

 */

void TestMpCollectionDataModel::testDataArtistAlbumsTBoneNoData()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mItemDataReturn = false;

    mTest->mCollectionData->mCollectionTitleReturn = false;

    mTest->mCollectionData->mContext = ECollectionContextArtistAlbumsTBone;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 2);

    // Qt::DecorationRole

}

/*!

 Tests data() request for ArtistAllSongs context.

 */

void TestMpCollectionDataModel::testDataArtistAllSongs()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextArtistAllSongs;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 2);

    QCOMPARE(dataList.at(0), QString("Title1"));

    QCOMPARE(dataList.at(1), QString("Album1"));

    // Qt::DecorationRole

    data = mTest->data(modelIndex, Qt::DecorationRole);

    QCOMPARE(data.isNull(), true);

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString("Title1"));

}

/*!

 Tests data() request for ArtistAllSongs context with no data available.

 */

void TestMpCollectionDataModel::testDataArtistAllSongsNoData()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextArtistAllSongs;

    mTest->mCollectionData->mItemDataReturn = false;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    // Qt::DecorationRole

    data = mTest->data(modelIndex, Qt::DecorationRole);

    QCOMPARE(data.isNull(), true);

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString(""));

}

/*!

 Tests data() request for Albums context.

 */

void TestMpCollectionDataModel::testDataAlbums()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextAlbums;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 2);

    QCOMPARE(dataList.at(0), QString("Title1"));

    QCOMPARE(dataList.at(1), QString("Artist1"));

    // Qt::DecorationRole

    // Hb::IndexFeedbackRole

    data = mTest->data(modelIndex, Hb::IndexFeedbackRole);

    QCOMPARE(data.toString(), QString("Title1"));

}

/*!

 Tests data() request for Albums context with no data available.

 */

void TestMpCollectionDataModel::testDataAlbumsNoData()

{

    mTest->mCollectionData->mCount = 100;

    mTest->mRowCount = 100;

    mTest->mCollectionData->mContext = ECollectionContextAlbums;

    mTest->mCollectionData->mItemDataReturn = false;

    QModelIndex modelIndex = mHelper->indexFor(1);

    // Qt::DisplayRole

    QVariant data = mTest->data(modelIndex, Qt::DisplayRole);

    QCOMPARE(data.canConvert(QVariant::StringList), true);

    QStringList dataList = data.toStringList();

    QCOMPARE(dataList.count(), 2);