/*
* 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:
* NotesModel unit-test class.
*
*/

// System includes
#include <QtTest/QtTest>
#include <QtGui>
#include <QPointer>
#include <e32base.h>

// User includes
#include "notesmodel.h"
#include "agendautil.h"

class TestNotesModel : public QObject
{
	Q_OBJECT

public slots:
	void handleEntriesDeletion(int var);

private slots:
	// Test case setup.
	void initTestCase();
	void cleanupTestCase();

	// Test cases.
	void test_construction();
	void test_sourceModelAddNote();
	void test_sourceModelDeleteNote();
	void test_sourceModelUpdateNote();

private:
	QPointer<NotesModel> mNotesModel;
	QPointer<AgendaUtil> mAgendaUtil;

	QDateTime mStart;
	QDateTime mEnd;

	CActiveSchedulerWait *mWait;
};

/*!
	\class TestNotesModel

	This is unittest class for NotesModel apis.
 */

/*!
	Slot called when all the entries have been deleted.
 */
void TestNotesModel::handleEntriesDeletion(int var)
{
	Q_UNUSED(var)

	if (mWait && mWait->IsStarted()) {
		mWait->AsyncStop();
	}
}

/*!
	This function is called to init the testcase.
	Called for every test case execution.
 */
void TestNotesModel::initTestCase()
{
	QT_TRAP_THROWING(mAgendaUtil = new AgendaUtil(this););
	QVERIFY(mAgendaUtil != 0);

	// Connect to the required signals
	connect(
			mAgendaUtil, SIGNAL(entriesDeleted(int)),
			this, SLOT(handleEntriesDeletion(int)));

	mWait = 0;

	mStart.setDate(QDate(1900, 1, 1));
	mStart.setTime(QTime(0, 0, 0));
	mEnd.setDate(QDate(2100, 30, 12));
	mEnd.setTime(QTime(23, 59, 0));
}

/*!
	This function is called when the testcase execution is
	completed, for cleaning up.
 */
void TestNotesModel::cleanupTestCase()
{
	if (mAgendaUtil) {
		delete mAgendaUtil;
	}
	if (mNotesModel) {
		delete mNotesModel;
	}
	if (mWait && mWait->IsStarted()) {
		mWait->AsyncStop();
	}
	if (mWait) {
		delete mWait;
	}
}

/*!
	This test functions tests the construction of NotesModel object.
 */
void TestNotesModel::test_construction()
{
	mNotesModel = new NotesModel(mAgendaUtil, this);
	QVERIFY(mNotesModel);

	if (mNotesModel) {
		delete mNotesModel;
		QVERIFY(!mNotesModel);
	}
}

/*!
	This test function tests if the source model is updated when a note is
	added.
 */
void TestNotesModel::test_sourceModelAddNote()
{
	mAgendaUtil->deleteEntries(mStart, mEnd, AgendaUtil::IncludeAll);
	if (!mWait) {
		mWait =  new CActiveSchedulerWait;
	}
	if (!mWait->IsStarted()) {
		mWait->Start();
	}
	mNotesModel = new NotesModel(mAgendaUtil, this);
	QVERIFY(mNotesModel);

	QAbstractItemModel *model = 0;
	model = mNotesModel->sourceModel();
	QVERIFY(model);
	QVERIFY(!model->rowCount());

	// Now save a note.
	AgendaEntry entry;
	entry.setType(AgendaEntry::TypeNote);
	entry.setDescription("Hello");
	entry.setLastModifiedDateTime(
			QDateTime(QDate::currentDate(), QTime::currentTime()));
	QVERIFY(mAgendaUtil->addEntry(entry));
	
	// Now check the count in the model.
	startTimer(1000);
	QVERIFY(model->rowCount());
	QVERIFY(model->rowCount() == 1);
}

/*!
	This test function tests if the source model is updated when a note is
	deleted.
 */
void TestNotesModel::test_sourceModelDeleteNote()
{
	mAgendaUtil->deleteEntries(mStart, mEnd, AgendaUtil::IncludeAll);
	if (!mWait) {
		mWait =  new CActiveSchedulerWait;
	}
	if (!mWait->IsStarted()) {
		mWait->Start();
	}
	mNotesModel = new NotesModel(mAgendaUtil, this);
	QVERIFY(mNotesModel);

	QAbstractItemModel *model = 0;
	model = mNotesModel->sourceModel();
	QVERIFY(model);
	QVERIFY(!model->rowCount());

	// Now save a note.
	AgendaEntry entry;
	entry.setType(AgendaEntry::TypeNote);
	entry.setDescription("Hello");
	entry.setLastModifiedDateTime(
			QDateTime(QDate::currentDate(), QTime::currentTime()));
	ulong id;
	QVERIFY(id = mAgendaUtil->addEntry(entry));

	// Now check the count in the model.
	startTimer(1000);
	QVERIFY(model->rowCount());
	QVERIFY(model->rowCount() == 1);

	// Verify if the model has the updated content.
	startTimer(1000);
	QModelIndex index = model->index(0, 0);
	QVERIFY(index.isValid());
	QStringList list = index.data(Qt::DisplayRole).value<QStringList>();
	QCOMPARE(entry.description(), list.at(0));

	// Delete the note saved above.
	QVERIFY(mAgendaUtil->deleteEntry(id));
	// Now check the count in the model.
	startTimer(1000);
	QVERIFY(!model->rowCount());
}

/*!
	This test function tests if the source model is updated when a note is
	modified.
 */
void TestNotesModel::test_sourceModelUpdateNote()
{
	mAgendaUtil->deleteEntries(mStart, mEnd, AgendaUtil::IncludeAll);
	if (!mWait) {
		mWait =  new CActiveSchedulerWait;
	}
	if (!mWait->IsStarted()) {
		mWait->Start();
	}
	mNotesModel = new NotesModel(mAgendaUtil, this);
	QVERIFY(mNotesModel);

	QAbstractItemModel *model = 0;
	model = mNotesModel->sourceModel();
	QVERIFY(model);
	QVERIFY(!model->rowCount());

	// Now save a note.
	AgendaEntry entry;
	entry.setType(AgendaEntry::TypeNote);
	entry.setDescription("Hello");
	entry.setLastModifiedDateTime(
			QDateTime(QDate::currentDate(), QTime::currentTime()));
	ulong id;
	QVERIFY(id = mAgendaUtil->addEntry(entry));
	// Now check the count in the model.
	startTimer(1000);
	QVERIFY(model->rowCount());
	QVERIFY(model->rowCount() == 1);

	// Now update the content on of the note created above.
	entry = mAgendaUtil->fetchById(id);
	QVERIFY(!entry.isNull());
	entry.setDescription("Hello world");
	QVERIFY(mAgendaUtil->updateEntry(entry));

	// Verify if the model has the updated content.
	startTimer(1000);
	QModelIndex index = model->index(0, 0);
	QVERIFY(index.isValid());
	QStringList list = index.data(Qt::DisplayRole).value<QStringList>();
	QCOMPARE(entry.description(), list.at(0));

	// Delete the note saved above.
	QVERIFY(mAgendaUtil->deleteEntry(id));
	// Now check the count in the model.
	startTimer(1000);
	QVERIFY(!model->rowCount());
}

QTEST_MAIN(TestNotesModel)
#include "unittest_notesmodel.moc"

// End of file	--Don't remove this.