/***************************************************************************
 *   Copyright (C) 2006-2008 by Ian Reinhart Geiser                        *
 *   geiseri@yahoo.com                                                     *
 *                                                                         *
 *   This program 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 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program 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 for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#include "freemindtests.h"
#include "block.h"
#include "freemindimporter.h"
#include "mindmap.h"

#include <QtTest/QtTest>
#include <QDomDocument>
#include <QGraphicsScene>
#include <QColor>

QTEST_MAIN(FreeMindTests);


FreeMindTests::FreeMindTests( QObject *parent )
		: QObject( parent )
{
}


FreeMindTests::~FreeMindTests()
{
}

void FreeMindTests::testFindNode( )
{
	QFile inputFile("freemind.mm");
	inputFile.open(QIODevice::ReadOnly);

	QDomDocument doc;
	QVERIFY( doc.setContent( &inputFile ) );

	QDomElement rootNode = doc.firstChildElement("map").firstChildElement("node");
	QCOMPARE( rootNode.attribute("TEXT"), QString("New Mindmap"));
	QCOMPARE( rootNode.attribute("COLOR"), QString("#000000"));
	QCOMPARE( rootNode.attribute("STYLE"), QString("bubble"));
}

void FreeMindTests::testFindChildNode( )
{
	QFile inputFile("freemind.mm");
	inputFile.open(QIODevice::ReadOnly);

	QDomDocument doc;
	QVERIFY( doc.setContent( &inputFile ) );

	QDomElement rootNode = doc.firstChildElement("map").firstChildElement("node");
	QVERIFY( !rootNode.firstChildElement("node").isNull() );

	QCOMPARE( rootNode.firstChildElement("node").attribute("TEXT"), QString("\nhghggh"));
	QCOMPARE( rootNode.firstChildElement("node").attribute("COLOR"), QString("#0033ff"));
	QCOMPARE( rootNode.firstChildElement("node").attribute("STYLE"), QString("bubble"));

}

void FreeMindTests::testCreateBlock( )
{
	QFile inputFile("freemind.mm");
	inputFile.open(QIODevice::ReadOnly);

	QDomDocument doc;
	QVERIFY( doc.setContent( &inputFile ) );
	Block *block = FreeMindImporter::blockFromNode( doc.firstChildElement("map").firstChildElement("node"), 0, 0 );

	QVERIFY( block );
	QCOMPARE( block->title(), QLatin1String("New Mindmap") );
}

void FreeMindTests::testImportChildren( )
{
	MindMap map;
	QFile inputFile("freemind.mm");
	inputFile.open(QIODevice::ReadOnly);

	QDomDocument doc;
	QVERIFY( doc.setContent( &inputFile ) );

	FreeMindImporter importer( &map );

	QVERIFY( importer.importChildren( 0, doc.firstChildElement("map").firstChildElement("node") ) );
}


void FreeMindTests::testFullImport( )
{
	MindMap map;
	QFile inputFile("freemind.mm");
	inputFile.open(QIODevice::ReadOnly);

	QDomDocument doc;
	QVERIFY( doc.setContent( &inputFile ) );

	QVERIFY( map.importFreemindMindMap( doc ) );
	QCOMPARE( map.root()->title(), QLatin1String("New Mindmap") );
	QCOMPARE( map.root()->childCount(), 10 );
}