summaryrefslogtreecommitdiffstats
path: root/kmymoney2/reports/reportstestcommon.cpp
diff options
context:
space:
mode:
authortpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2011-07-04 22:38:03 +0000
committertpearson <tpearson@283d02a7-25f6-0310-bc7c-ecb5cbfe19da>2011-07-04 22:38:03 +0000
commitdadc34655c3ab961b0b0b94a10eaaba710f0b5e8 (patch)
tree99e72842fe687baea16376a147619b6048d7e441 /kmymoney2/reports/reportstestcommon.cpp
downloadkmymoney-dadc34655c3ab961b0b0b94a10eaaba710f0b5e8.tar.gz
kmymoney-dadc34655c3ab961b0b0b94a10eaaba710f0b5e8.zip
Added kmymoney
git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/kmymoney@1239792 283d02a7-25f6-0310-bc7c-ecb5cbfe19da
Diffstat (limited to 'kmymoney2/reports/reportstestcommon.cpp')
-rw-r--r--kmymoney2/reports/reportstestcommon.cpp494
1 files changed, 494 insertions, 0 deletions
diff --git a/kmymoney2/reports/reportstestcommon.cpp b/kmymoney2/reports/reportstestcommon.cpp
new file mode 100644
index 0000000..31e6c1d
--- /dev/null
+++ b/kmymoney2/reports/reportstestcommon.cpp
@@ -0,0 +1,494 @@
+/***************************************************************************
+ reportstestcommon.cpp
+ -------------------
+ copyright : (C) 2002-2005 by Thomas Baumgart
+ email : ipwizard@users.sourceforge.net
+ Ace Jones <ace.j@hotpop.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. *
+ * *
+ ***************************************************************************/
+
+#include <qvaluelist.h>
+#include <qvaluevector.h>
+#include <qdom.h>
+#include <qfile.h>
+
+#include <kdebug.h>
+#include <kdeversion.h>
+#include <kglobal.h>
+#include <kglobalsettings.h>
+#include <klocale.h>
+#include <kstandarddirs.h>
+
+#include "kreportsviewtest.h"
+
+#define private public
+#include "pivottable.h"
+#include "querytable.h"
+#undef private
+using namespace reports;
+
+#include "../mymoney/mymoneysecurity.h"
+#include "../mymoney/mymoneyprice.h"
+#include "../mymoney/storage/mymoneystoragedump.h"
+#include "../mymoney/mymoneyreport.h"
+#include "../mymoney/mymoneystatement.h"
+#include "../mymoney/storage/mymoneystoragexml.h"
+#include "reportstestcommon.h"
+
+namespace test {
+
+const MyMoneyMoney moCheckingOpen(0.0);
+const MyMoneyMoney moCreditOpen(-0.0);
+const MyMoneyMoney moConverterCheckingOpen(1418.0);
+const MyMoneyMoney moConverterCreditOpen(-418.0);
+const MyMoneyMoney moZero(0.0);
+const MyMoneyMoney moSolo(234.12);
+const MyMoneyMoney moParent1(88.01);
+const MyMoneyMoney moParent2(133.22);
+const MyMoneyMoney moParent(moParent1+moParent2);
+const MyMoneyMoney moChild(14.00);
+const MyMoneyMoney moThomas(5.11);
+const MyMoneyMoney moNoPayee(8944.70);
+
+QString acAsset;
+QString acLiability;
+QString acExpense;
+QString acIncome;
+QString acChecking;
+QString acCredit;
+QString acSolo;
+QString acParent;
+QString acChild;
+QString acSecondChild;
+QString acGrandChild1;
+QString acGrandChild2;
+QString acForeign;
+QString acCanChecking;
+QString acJpyChecking;
+QString acCanCash;
+QString acJpyCash;
+QString inBank;
+QString eqStock1;
+QString eqStock2;
+QString acInvestment;
+QString acStock1;
+QString acStock2;
+QString acDividends;
+QString acInterest;
+QString acTax;
+QString acCash;
+
+TransactionHelper::TransactionHelper( const QDate& _date, const QString& _action, MyMoneyMoney _value, const QString& _accountid, const QString& _categoryid, const QString& _currencyid, const QString& _payee )
+{
+ // _currencyid is the currency of the transaction, and of the _value
+ // both the account and category can have their own currency (athough the category having
+ // a foreign currency is not yet supported by the program, the reports will still allow it,
+ // so it must be tested.)
+
+ MyMoneyFile* file = MyMoneyFile::instance();
+ bool haspayee = ! _payee.isEmpty();
+ MyMoneyPayee payeeTest = file->payeeByName(_payee);
+
+ MyMoneyFileTransaction ft;
+ setPostDate(_date);
+
+ QString currencyid = _currencyid;
+ if ( currencyid.isEmpty() )
+ currencyid=MyMoneyFile::instance()->baseCurrency().id();
+ setCommodity(currencyid);
+
+ MyMoneyMoney price;
+ MyMoneySplit splitLeft;
+ if ( haspayee )
+ splitLeft.setPayeeId(payeeTest.id());
+ splitLeft.setAction(_action);
+ splitLeft.setValue(-_value);
+ price = MyMoneyFile::instance()->price(currencyid, file->account(_accountid).currencyId(),_date).rate(file->account(_accountid).currencyId());
+ splitLeft.setShares(-_value * price);
+ splitLeft.setAccountId(_accountid);
+ addSplit(splitLeft);
+
+ MyMoneySplit splitRight;
+ if ( haspayee )
+ splitRight.setPayeeId(payeeTest.id());
+ splitRight.setAction(_action);
+ splitRight.setValue(_value);
+ price = MyMoneyFile::instance()->price(currencyid, file->account(_categoryid).currencyId(),_date).rate(file->account(_categoryid).currencyId());
+ splitRight.setShares(_value * price );
+ splitRight.setAccountId(_categoryid);
+ addSplit(splitRight);
+
+ MyMoneyFile::instance()->addTransaction(*this);
+ ft.commit();
+}
+
+TransactionHelper::~TransactionHelper()
+{
+ MyMoneyFileTransaction ft;
+ MyMoneyFile::instance()->removeTransaction(*this);
+ ft.commit();
+}
+
+void TransactionHelper::update(void)
+{
+ MyMoneyFileTransaction ft;
+ MyMoneyFile::instance()->modifyTransaction(*this);
+ ft.commit();
+}
+
+InvTransactionHelper::InvTransactionHelper( const QDate& _date, const QString& _action, MyMoneyMoney _shares, MyMoneyMoney _price, const QString& _stockaccountid, const QString& _transferid, const QString& _categoryid )
+{
+ init(_date, _action, _shares, _price, _stockaccountid, _transferid, _categoryid);
+}
+
+void InvTransactionHelper::init( const QDate& _date, const QString& _action, MyMoneyMoney _shares, MyMoneyMoney _price, const QString& _stockaccountid, const QString& _transferid, const QString& _categoryid )
+{
+ MyMoneyFile* file = MyMoneyFile::instance();
+ MyMoneyAccount stockaccount = file->account(_stockaccountid);
+ MyMoneyMoney value = _shares * _price;
+
+ setPostDate(_date);
+
+ setCommodity("USD");
+ MyMoneySplit s1;
+ s1.setValue(value);
+ s1.setAccountId(_stockaccountid);
+
+ if ( _action == MyMoneySplit::ActionReinvestDividend )
+ {
+ s1.setShares(_shares);
+ s1.setAction(MyMoneySplit::ActionReinvestDividend);
+
+ MyMoneySplit s2;
+ s2.setAccountId(_categoryid);
+ s2.setShares(-value);
+ s2.setValue(-value);
+ addSplit(s2);
+ }
+ else if ( _action == MyMoneySplit::ActionDividend || _action == MyMoneySplit::ActionYield )
+ {
+ s1.setAccountId(_categoryid);
+ s1.setShares(-value);
+ s1.setValue(-value);
+
+ // Split 2 will be the zero-amount investment split that serves to
+ // mark this transaction as a cash dividend and note which stock account
+ // it belongs to.
+ MyMoneySplit s2;
+ s2.setValue(0);
+ s2.setShares(0);
+ s2.setAction(_action);
+ s2.setAccountId(_stockaccountid);
+ addSplit(s2);
+
+ MyMoneySplit s3;
+ s3.setAccountId(_transferid);
+ s3.setShares(value);
+ s3.setValue(value);
+ addSplit(s3);
+ }
+ else if ( _action == MyMoneySplit::ActionBuyShares )
+ {
+ s1.setShares(_shares);
+ s1.setAction(MyMoneySplit::ActionBuyShares);
+
+ MyMoneySplit s3;
+ s3.setAccountId(_transferid);
+ s3.setShares(-value);
+ s3.setValue(-value);
+ addSplit(s3);
+ }
+ addSplit(s1);
+
+ //kdDebug(2) << "created transaction, now adding..." << endl;
+
+ MyMoneyFileTransaction ft;
+ file->addTransaction(*this);
+
+ //kdDebug(2) << "updating price..." << endl;
+
+ // update the price, while we're here
+ QString stockid = stockaccount.currencyId();
+ QString basecurrencyid = file->baseCurrency().id();
+ MyMoneyPrice price = file->price( stockid, basecurrencyid, _date, true );
+ if ( !price.isValid() )
+ {
+ MyMoneyPrice newprice( stockid, basecurrencyid, _date, _price, "test" );
+ file->addPrice(newprice);
+ }
+ ft.commit();
+ //kdDebug(2) << "successfully added " << id() << endl;
+}
+
+QString makeAccount( const QString& _name, MyMoneyAccount::accountTypeE _type, MyMoneyMoney _balance, const QDate& _open, const QString& _parent, QString _currency, bool _taxReport )
+{
+ MyMoneyAccount info;
+ MyMoneyFileTransaction ft;
+
+ info.setName(_name);
+ info.setAccountType(_type);
+ info.setOpeningDate(_open);
+ if ( _currency != "" )
+ info.setCurrencyId(_currency);
+ else
+ info.setCurrencyId(MyMoneyFile::instance()->baseCurrency().id());
+
+ if(_taxReport)
+ info.setValue("Tax", "Yes");
+
+ MyMoneyAccount parent = MyMoneyFile::instance()->account(_parent);
+ MyMoneyFile::instance()->addAccount( info, parent );
+ // create the opening balance transaction if any
+ if(!_balance.isZero()) {
+ MyMoneySecurity sec = MyMoneyFile::instance()->currency(info.currencyId());
+ MyMoneyFile::instance()->openingBalanceAccount(sec);
+ MyMoneyFile::instance()->createOpeningBalanceTransaction(info, _balance);
+ }
+ ft.commit();
+
+ return info.id();
+}
+
+void makePrice(const QString& _currencyid, const QDate& _date, const MyMoneyMoney& _price )
+{
+ MyMoneyFileTransaction ft;
+ MyMoneyFile* file = MyMoneyFile::instance();
+ MyMoneySecurity curr = file->currency(_currencyid);
+ MyMoneyPrice price(_currencyid, file->baseCurrency().id(), _date, _price, "test");
+ file->addPrice(price);
+ ft.commit();
+}
+
+QString makeEquity(const QString& _name, const QString& _symbol )
+{
+ MyMoneySecurity equity;
+ MyMoneyFileTransaction ft;
+
+ equity.setName( _name );
+ equity.setTradingSymbol( _symbol );
+ equity.setSmallestAccountFraction( 1000 );
+ equity.setSecurityType( MyMoneySecurity::SECURITY_NONE /*MyMoneyEquity::ETYPE_STOCK*/ );
+ MyMoneyFile::instance()->addSecurity( equity );
+ ft.commit();
+
+ return equity.id();
+}
+
+void makeEquityPrice(const QString& _id, const QDate& _date, const MyMoneyMoney& _price )
+{
+ MyMoneyFile* file = MyMoneyFile::instance();
+ MyMoneyFileTransaction ft;
+ QString basecurrencyid = file->baseCurrency().id();
+ MyMoneyPrice price = file->price( _id, basecurrencyid, _date, true );
+ if ( !price.isValid() )
+ {
+ MyMoneyPrice newprice( _id, basecurrencyid, _date, _price, "test" );
+ file->addPrice(newprice);
+ }
+ ft.commit();
+}
+
+void writeRCFtoXMLDoc( const MyMoneyReport& filter, QDomDocument* doc )
+{
+ QDomProcessingInstruction instruct = doc->createProcessingInstruction(QString("xml"), QString("version=\"1.0\" encoding=\"utf-8\""));
+ doc->appendChild(instruct);
+
+ QDomElement root = doc->createElement("KMYMONEY-FILE");
+ doc->appendChild(root);
+
+ QDomElement reports = doc->createElement("REPORTS");
+ root.appendChild(reports);
+
+ QDomElement report = doc->createElement("REPORT");
+ filter.write(report,doc);
+ reports.appendChild(report);
+
+}
+
+void writeTabletoHTML( const PivotTable& table, const QString& _filename )
+{
+ static unsigned filenumber = 1;
+ QString filename = _filename;
+ if ( filename.isEmpty() )
+ {
+ filename = QString("report-%1%2.html").arg((filenumber<10)?"0":"").arg(filenumber);
+ ++filenumber;
+ }
+
+ QFile g( filename );
+ g.open( IO_WriteOnly );
+ QTextStream(&g) << table.renderHTML();
+ g.close();
+
+}
+
+void writeTabletoHTML( const QueryTable& table, const QString& _filename )
+{
+ static unsigned filenumber = 1;
+ QString filename = _filename;
+ if ( filename.isEmpty() )
+ {
+ filename = QString("report-%1%2.html").arg((filenumber<10)?"0":"").arg(filenumber);
+ ++filenumber;
+ }
+
+ QFile g( filename );
+ g.open( IO_WriteOnly );
+ QTextStream(&g) << table.renderHTML();
+ g.close();
+}
+
+void writeTabletoCSV( const PivotTable& table, const QString& _filename )
+{
+ static unsigned filenumber = 1;
+ QString filename = _filename;
+ if ( filename.isEmpty() )
+ {
+ filename = QString("report-%1%2.csv").arg((filenumber<10)?"0":"").arg(filenumber);
+ ++filenumber;
+ }
+
+ QFile g( filename );
+ g.open( IO_WriteOnly );
+ QTextStream(&g) << table.renderCSV();
+ g.close();
+
+}
+
+void writeTabletoCSV( const QueryTable& table, const QString& _filename )
+{
+ static unsigned filenumber = 1;
+ QString filename = _filename;
+ if ( filename.isEmpty() )
+ {
+ filename = QString("qreport-%1%2.csv").arg((filenumber<10)?"0":"").arg(filenumber);
+ ++filenumber;
+ }
+
+ QFile g( filename );
+ g.open( IO_WriteOnly );
+ QTextStream(&g) << table.renderCSV();
+ g.close();
+
+}
+
+void writeRCFtoXML( const MyMoneyReport& filter, const QString& _filename )
+{
+ static unsigned filenum = 1;
+ QString filename = _filename;
+ if ( filename.isEmpty() ) {
+ filename = QString("report-%1%2.xml").arg(QString::number(filenum).rightJustify(2, '0'));
+ ++filenum;
+ }
+
+ QDomDocument* doc = new QDomDocument("KMYMONEY-FILE");
+ Q_CHECK_PTR(doc);
+
+ writeRCFtoXMLDoc(filter,doc);
+
+ QFile g( filename );
+ g.open( IO_WriteOnly );
+
+ QTextStream stream(&g);
+#if KDE_IS_VERSION(3,2,0)
+ stream.setEncoding(QTextStream::UnicodeUTF8);
+ stream << doc->toString();
+#else
+ //stream.setEncoding(QTextStream::Locale);
+ QString temp = doc->toString();
+ stream << temp.data();
+#endif
+ g.close();
+
+ delete doc;
+}
+
+bool readRCFfromXMLDoc( QValueList<MyMoneyReport>& list, QDomDocument* doc )
+{
+ bool result = false;
+
+ QDomElement rootElement = doc->documentElement();
+ if(!rootElement.isNull())
+ {
+ QDomNode child = rootElement.firstChild();
+ while(!child.isNull() && child.isElement())
+ {
+ QDomElement childElement = child.toElement();
+ if("REPORTS" == childElement.tagName())
+ {
+ result = true;
+ QDomNode subchild = child.firstChild();
+ while(!subchild.isNull() && subchild.isElement())
+ {
+ MyMoneyReport filter;
+ if ( filter.read(subchild.toElement()))
+ {
+ list += filter;
+ }
+ subchild = subchild.nextSibling();
+ }
+ }
+ child = child.nextSibling();
+ }
+ }
+ return result;
+}
+
+bool readRCFfromXML( QValueList<MyMoneyReport>& list, const QString& filename )
+{
+ int result = false;
+ QFile f( filename );
+ f.open( IO_ReadOnly );
+ QDomDocument* doc = new QDomDocument;
+ if(doc->setContent(&f, FALSE))
+ {
+ result = readRCFfromXMLDoc(list,doc);
+ }
+ delete doc;
+
+ return result;
+
+}
+
+void XMLandback( MyMoneyReport& filter )
+{
+ // this function writes the filter to XML, and then reads
+ // it back from XML overwriting the original filter;
+ // in all cases, the result should be the same if the read
+ // & write methods are working correctly.
+
+ QDomDocument* doc = new QDomDocument("KMYMONEY-FILE");
+ Q_CHECK_PTR(doc);
+
+ writeRCFtoXMLDoc(filter,doc);
+ QValueList<MyMoneyReport> list;
+ if ( readRCFfromXMLDoc(list,doc) && list.count() > 0 )
+ filter = list[0];
+ else
+ throw new MYMONEYEXCEPTION("Failed to load report from XML");
+
+ delete doc;
+
+}
+
+MyMoneyMoney searchHTML(const QString& _html, const QString& _search)
+{
+ QRegExp re(QString("%1[<>/td]*([\\-.0-9,]*)").arg(_search));
+ re.search(_html);
+ QString found = re.cap(1);
+ found.remove(',');
+
+ return MyMoneyMoney(found.toDouble());
+}
+
+} // end namespace test
+
+// vim:cin:si:ai:et:ts=2:sw=2: