599 lines
20 KiB
Haskell
599 lines
20 KiB
Haskell
module Tests
|
|
where
|
|
import qualified Data.Map as Map
|
|
import Text.ParserCombinators.Parsec
|
|
import Test.HUnit
|
|
import Ledger
|
|
import Utils
|
|
import Options
|
|
import BalanceCommand
|
|
import PrintCommand
|
|
import RegisterCommand
|
|
|
|
|
|
runtests args = do
|
|
putStrLn $ printf "Running %d tests%s ..\n" n s
|
|
runTestTT flattests
|
|
where
|
|
deeptests = tfilter matchname $ TestList tests
|
|
flattests = TestList $ filter matchname $ concatMap tflatten tests
|
|
matchname = matchpats args . tname
|
|
n = length ts where (TestList ts) = flattests
|
|
s | null args = ""
|
|
| otherwise = printf " matching %s"
|
|
(intercalate ", " $ map (printf "\"%s\"") args)
|
|
|
|
------------------------------------------------------------------------------
|
|
-- tests
|
|
|
|
tests =
|
|
[TestList []
|
|
,misc_tests
|
|
,balancereportacctnames_tests
|
|
,balancecommand_tests
|
|
,registercommand_tests
|
|
]
|
|
|
|
misc_tests = TestList [
|
|
"show dollars" ~: show (dollars 1) ~?= "$1.00"
|
|
,
|
|
"show hours" ~: show (hours 1) ~?= "1.0h"
|
|
,
|
|
"amount arithmetic" ~: do
|
|
let a1 = dollars 1.23
|
|
let a2 = Amount (comm "$") (-1.23) Nothing
|
|
let a3 = Amount (comm "$") (-1.23) Nothing
|
|
assertequal (Amount (comm "$") 0 Nothing) (a1 + a2)
|
|
assertequal (Amount (comm "$") 0 Nothing) (a1 + a3)
|
|
assertequal (Amount (comm "$") (-2.46) Nothing) (a2 + a3)
|
|
assertequal (Amount (comm "$") (-2.46) Nothing) (a3 + a3)
|
|
assertequal (Amount (comm "$") (-2.46) Nothing) (sum [a2,a3])
|
|
assertequal (Amount (comm "$") (-2.46) Nothing) (sum [a3,a3])
|
|
assertequal (Amount (comm "$") 0 Nothing) (sum [a1,a2,a3,-a3])
|
|
,
|
|
"ledgertransaction" ~: do
|
|
assertparseequal rawtransaction1 (parsewith ledgertransaction rawtransaction1_str)
|
|
,
|
|
"ledgerentry" ~: do
|
|
assertparseequal entry1 (parsewith ledgerentry entry1_str)
|
|
,
|
|
"balanceEntry" ~: do
|
|
assertequal
|
|
(Mixed [dollars (-47.18)])
|
|
(tamount $ last $ etransactions $ balanceEntry entry1)
|
|
,
|
|
"punctuatethousands" ~: punctuatethousands "" @?= ""
|
|
,
|
|
"punctuatethousands" ~: punctuatethousands "1234567.8901" @?= "1,234,567.8901"
|
|
,
|
|
"punctuatethousands" ~: punctuatethousands "-100" @?= "-100"
|
|
,
|
|
"expandAccountNames" ~: do
|
|
assertequal
|
|
["assets","assets:cash","assets:checking","expenses","expenses:vacation"]
|
|
(expandAccountNames ["assets:cash","assets:checking","expenses:vacation"])
|
|
,
|
|
"ledgerAccountNames" ~: do
|
|
assertequal
|
|
["assets","assets:cash","assets:checking","assets:saving","equity","equity:opening balances",
|
|
"expenses","expenses:food","expenses:food:dining","expenses:phone","expenses:vacation",
|
|
"liabilities","liabilities:credit cards","liabilities:credit cards:discover"]
|
|
(accountnames ledger7)
|
|
,
|
|
"cacheLedger" ~: do
|
|
assertequal 15 (length $ Map.keys $ accountmap $ cacheLedger rawledger7)
|
|
,
|
|
"transactionamount" ~: do
|
|
assertparseequal (Mixed [dollars 47.18]) (parsewith transactionamount " $47.18")
|
|
assertparseequal (Mixed [Amount (Commodity {symbol="$",side=L,spaced=False,comma=False,precision=0}) 1 Nothing]) (parsewith transactionamount " $1.")
|
|
,
|
|
"setAmountDisplayPrefs" ~: do
|
|
let l = setAmountDisplayPrefs $ rawLedgerWithAmounts ["1","2.00"]
|
|
assertequal [2,2] (rawLedgerPrecisions l) -- use greatest precision everywhere
|
|
|
|
] where
|
|
rawLedgerWithAmounts as =
|
|
RawLedger
|
|
[]
|
|
[]
|
|
[nullentry{etransactions=[nullrawtxn{tamount=parse a}]} | a <- as]
|
|
""
|
|
where parse = fromparse . parsewith transactionamount . (" "++)
|
|
|
|
balancereportacctnames_tests = TestList
|
|
[
|
|
"balancereportacctnames0" ~: ("-s",[]) `gives` ["assets","assets:cash","assets:checking","assets:saving",
|
|
"expenses","expenses:food","expenses:supplies","income",
|
|
"income:gifts","income:salary","liabilities","liabilities:debts"]
|
|
,"balancereportacctnames1" ~: ("", []) `gives` ["assets","expenses","income","liabilities"]
|
|
,"balancereportacctnames2" ~: ("", ["assets"]) `gives` ["assets"]
|
|
,"balancereportacctnames3" ~: ("", ["as"]) `gives` ["assets","assets:cash"]
|
|
,"balancereportacctnames4" ~: ("", ["assets:cash"]) `gives` ["assets:cash"]
|
|
,"balancereportacctnames5" ~: ("", ["-assets"]) `gives` ["expenses","income","liabilities"]
|
|
,"balancereportacctnames6" ~: ("", ["-e"]) `gives` []
|
|
,"balancereportacctnames7" ~: ("-s",["assets"]) `gives` ["assets","assets:cash","assets:checking","assets:saving"]
|
|
,"balancereportacctnames8" ~: ("-s",["-e"]) `gives` []
|
|
] where
|
|
gives (opt,pats) e = do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
let t = pruneZeroBalanceLeaves $ ledgerAccountTree 999 l
|
|
assertequal e (balancereportacctnames l (opt=="-s") pats t)
|
|
|
|
balancecommand_tests = TestList [
|
|
"simple balance report" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $-1 assets\n\
|
|
\ $2 expenses\n\
|
|
\ $-2 income\n\
|
|
\ $1 liabilities\n\
|
|
\" --"
|
|
(showBalanceReport [] [] l)
|
|
,
|
|
"balance report with --subtotal" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $-1 assets\n\
|
|
\ $-2 cash\n\
|
|
\ $1 saving\n\
|
|
\ $2 expenses\n\
|
|
\ $1 food\n\
|
|
\ $1 supplies\n\
|
|
\ $-2 income\n\
|
|
\ $-1 gifts\n\
|
|
\ $-1 salary\n\
|
|
\ $1 liabilities:debts\n\
|
|
\" --"
|
|
(showBalanceReport [SubTotal] [] l)
|
|
,
|
|
"balance report with account pattern o" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $1 expenses:food\n\
|
|
\ $-2 income\n\
|
|
\--------------------\n\
|
|
\ $-1\n\
|
|
\" --"
|
|
(showBalanceReport [] ["o"] l)
|
|
,
|
|
"balance report with account pattern o and --subtotal" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $1 expenses:food\n\
|
|
\ $-2 income\n\
|
|
\ $-1 gifts\n\
|
|
\ $-1 salary\n\
|
|
\--------------------\n\
|
|
\ $-1\n\
|
|
\" --"
|
|
(showBalanceReport [SubTotal] ["o"] l)
|
|
,
|
|
"balance report with account pattern a" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $-1 assets\n\
|
|
\ $-2 cash\n\
|
|
\ $1 saving\n\
|
|
\ $-1 income:salary\n\
|
|
\ $1 liabilities\n\
|
|
\--------------------\n\
|
|
\ $-1\n\
|
|
\" --"
|
|
(showBalanceReport [] ["a"] l)
|
|
,
|
|
"balance report with account pattern e" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $-1 assets\n\
|
|
\ $2 expenses\n\
|
|
\ $1 supplies\n\
|
|
\ $-2 income\n\
|
|
\ $1 liabilities:debts\n\
|
|
\" --"
|
|
(showBalanceReport [] ["e"] l)
|
|
,
|
|
"balance report with unmatched parent of two matched subaccounts" ~:
|
|
do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $-2 assets:cash\n\
|
|
\ $1 assets:saving\n\
|
|
\--------------------\n\
|
|
\ $-1\n\
|
|
\" --"
|
|
(showBalanceReport [] ["cash","saving"] l)
|
|
,
|
|
"balance report with multi-part account name" ~:
|
|
do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal
|
|
" $1 expenses:food\n\
|
|
\--------------------\n\
|
|
\ $1\n\
|
|
\" --"
|
|
$ showBalanceReport [] ["expenses:food"] l
|
|
,
|
|
"balance report with negative account pattern" ~: do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal (
|
|
" $2 expenses\n" ++
|
|
" $-2 income\n" ++
|
|
" $1 liabilities\n" ++
|
|
"--------------------\n" ++
|
|
" $1\n" ++
|
|
"")
|
|
(showBalanceReport [] ["-assets"] l)
|
|
,
|
|
"balance report negative account pattern always matches full name" ~:
|
|
do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal "" $ showBalanceReport [] ["-e"] l
|
|
,
|
|
"balance report negative patterns affect totals" ~:
|
|
do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal (
|
|
" $1 expenses\n" ++
|
|
"--------------------\n" ++
|
|
" $1\n" ++
|
|
"")
|
|
$ showBalanceReport [] ["expenses","-food"] l
|
|
]
|
|
|
|
registercommand_tests = TestList [
|
|
"register report" ~:
|
|
do
|
|
l <- ledgerfromfile "sample.ledger"
|
|
assertequal (
|
|
"2007/01/01 income assets:checking $1 $1\n" ++
|
|
" income:salary $-1 0\n" ++
|
|
"2007/01/01 gift assets:checking $1 $1\n" ++
|
|
" income:gifts $-1 0\n" ++
|
|
"2007/01/01 save assets:saving $1 $1\n" ++
|
|
" assets:checking $-1 0\n" ++
|
|
"2007/01/01 eat & shop expenses:food $1 $1\n" ++
|
|
" expenses:supplies $1 $2\n" ++
|
|
" assets:cash $-2 0\n" ++
|
|
"2008/01/01 pay off liabilities:debts $1 $1\n" ++
|
|
" assets:checking $-1 0\n" ++
|
|
"")
|
|
$ showRegisterReport [] [] l
|
|
]
|
|
|
|
------------------------------------------------------------------------------
|
|
-- test data
|
|
|
|
rawtransaction1_str = " expenses:food:dining $10.00\n"
|
|
|
|
rawtransaction1 = RawTransaction "expenses:food:dining"(Mixed [dollars 10]) "" RegularTransaction
|
|
|
|
entry1_str = "\
|
|
\2007/01/28 coopportunity\n\
|
|
\ expenses:food:groceries $47.18\n\
|
|
\ assets:checking\n\
|
|
\\n" --"
|
|
|
|
entry1 =
|
|
(Entry "2007/01/28" False "" "coopportunity" ""
|
|
[RawTransaction "expenses:food:groceries" (Mixed [dollars 47.18]) "" RegularTransaction,
|
|
RawTransaction "assets:checking" (Mixed [dollars (-47.18)]) "" RegularTransaction] "")
|
|
|
|
|
|
entry2_str = "\
|
|
\2007/01/27 * joes diner\n\
|
|
\ expenses:food:dining $10.00\n\
|
|
\ expenses:gifts $10.00\n\
|
|
\ assets:checking $-20.00\n\
|
|
\\n" --"
|
|
|
|
entry3_str = "\
|
|
\2007/01/01 * opening balance\n\
|
|
\ assets:cash $4.82\n\
|
|
\ equity:opening balances\n\
|
|
\\n\
|
|
\2007/01/01 * opening balance\n\
|
|
\ assets:cash $4.82\n\
|
|
\ equity:opening balances\n\
|
|
\\n\
|
|
\2007/01/28 coopportunity\n\
|
|
\ expenses:food:groceries $47.18\n\
|
|
\ assets:checking\n\
|
|
\\n" --"
|
|
|
|
periodic_entry1_str = "\
|
|
\~ monthly from 2007/2/2\n\
|
|
\ assets:saving $200.00\n\
|
|
\ assets:checking\n\
|
|
\\n" --"
|
|
|
|
periodic_entry2_str = "\
|
|
\~ monthly from 2007/2/2\n\
|
|
\ assets:saving $200.00 ;auto savings\n\
|
|
\ assets:checking\n\
|
|
\\n" --"
|
|
|
|
periodic_entry3_str = "\
|
|
\~ monthly from 2007/01/01\n\
|
|
\ assets:cash $4.82\n\
|
|
\ equity:opening balances\n\
|
|
\\n\
|
|
\~ monthly from 2007/01/01\n\
|
|
\ assets:cash $4.82\n\
|
|
\ equity:opening balances\n\
|
|
\\n" --"
|
|
|
|
ledger1_str = "\
|
|
\\n\
|
|
\2007/01/27 * joes diner\n\
|
|
\ expenses:food:dining $10.00\n\
|
|
\ expenses:gifts $10.00\n\
|
|
\ assets:checking $-20.00\n\
|
|
\\n\
|
|
\\n\
|
|
\2007/01/28 coopportunity\n\
|
|
\ expenses:food:groceries $47.18\n\
|
|
\ assets:checking $-47.18\n\
|
|
\\n\
|
|
\" --"
|
|
|
|
ledger2_str = "\
|
|
\;comment\n\
|
|
\2007/01/27 * joes diner\n\
|
|
\ expenses:food:dining $10.00\n\
|
|
\ assets:checking $-47.18\n\
|
|
\\n" --"
|
|
|
|
ledger3_str = "\
|
|
\2007/01/27 * joes diner\n\
|
|
\ expenses:food:dining $10.00\n\
|
|
\;intra-entry comment\n\
|
|
\ assets:checking $-47.18\n\
|
|
\\n" --"
|
|
|
|
ledger4_str = "\
|
|
\!include \"somefile\"\n\
|
|
\2007/01/27 * joes diner\n\
|
|
\ expenses:food:dining $10.00\n\
|
|
\ assets:checking $-47.18\n\
|
|
\\n" --"
|
|
|
|
ledger5_str = ""
|
|
|
|
ledger6_str = "\
|
|
\~ monthly from 2007/1/21\n\
|
|
\ expenses:entertainment $16.23 ;netflix\n\
|
|
\ assets:checking\n\
|
|
\\n\
|
|
\; 2007/01/01 * opening balance\n\
|
|
\; assets:saving $200.04\n\
|
|
\; equity:opening balances \n\
|
|
\\n" --"
|
|
|
|
ledger7_str = "\
|
|
\2007/01/01 * opening balance\n\
|
|
\ assets:cash $4.82\n\
|
|
\ equity:opening balances \n\
|
|
\\n\
|
|
\2007/01/01 * opening balance\n\
|
|
\ income:interest $-4.82\n\
|
|
\ equity:opening balances \n\
|
|
\\n\
|
|
\2007/01/02 * ayres suites\n\
|
|
\ expenses:vacation $179.92\n\
|
|
\ assets:checking \n\
|
|
\\n\
|
|
\2007/01/02 * auto transfer to savings\n\
|
|
\ assets:saving $200.00\n\
|
|
\ assets:checking \n\
|
|
\\n\
|
|
\2007/01/03 * poquito mas\n\
|
|
\ expenses:food:dining $4.82\n\
|
|
\ assets:cash \n\
|
|
\\n\
|
|
\2007/01/03 * verizon\n\
|
|
\ expenses:phone $95.11\n\
|
|
\ assets:checking \n\
|
|
\\n\
|
|
\2007/01/03 * discover\n\
|
|
\ liabilities:credit cards:discover $80.00\n\
|
|
\ assets:checking \n\
|
|
\\n\
|
|
\2007/01/04 * blue cross\n\
|
|
\ expenses:health:insurance $90.00\n\
|
|
\ assets:checking \n\
|
|
\\n\
|
|
\2007/01/05 * village market liquor\n\
|
|
\ expenses:food:dining $6.48\n\
|
|
\ assets:checking \n\
|
|
\\n" --"
|
|
|
|
rawledger7 = RawLedger
|
|
[]
|
|
[]
|
|
[
|
|
Entry {
|
|
edate="2007/01/01",
|
|
estatus=False,
|
|
ecode="*",
|
|
edescription="opening balance",
|
|
ecomment="",
|
|
etransactions=[
|
|
RawTransaction {
|
|
taccount="assets:cash",
|
|
tamount=(Mixed [dollars 4.82]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
},
|
|
RawTransaction {
|
|
taccount="equity:opening balances",
|
|
tamount=(Mixed [dollars (-4.82)]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
}
|
|
],
|
|
epreceding_comment_lines=""
|
|
}
|
|
,
|
|
Entry {
|
|
edate="2007/02/01",
|
|
estatus=False,
|
|
ecode="*",
|
|
edescription="ayres suites",
|
|
ecomment="",
|
|
etransactions=[
|
|
RawTransaction {
|
|
taccount="expenses:vacation",
|
|
tamount=(Mixed [dollars 179.92]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
},
|
|
RawTransaction {
|
|
taccount="assets:checking",
|
|
tamount=(Mixed [dollars (-179.92)]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
}
|
|
],
|
|
epreceding_comment_lines=""
|
|
}
|
|
,
|
|
Entry {
|
|
edate="2007/01/02",
|
|
estatus=False,
|
|
ecode="*",
|
|
edescription="auto transfer to savings",
|
|
ecomment="",
|
|
etransactions=[
|
|
RawTransaction {
|
|
taccount="assets:saving",
|
|
tamount=(Mixed [dollars 200]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
},
|
|
RawTransaction {
|
|
taccount="assets:checking",
|
|
tamount=(Mixed [dollars (-200)]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
}
|
|
],
|
|
epreceding_comment_lines=""
|
|
}
|
|
,
|
|
Entry {
|
|
edate="2007/01/03",
|
|
estatus=False,
|
|
ecode="*",
|
|
edescription="poquito mas",
|
|
ecomment="",
|
|
etransactions=[
|
|
RawTransaction {
|
|
taccount="expenses:food:dining",
|
|
tamount=(Mixed [dollars 4.82]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
},
|
|
RawTransaction {
|
|
taccount="assets:cash",
|
|
tamount=(Mixed [dollars (-4.82)]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
}
|
|
],
|
|
epreceding_comment_lines=""
|
|
}
|
|
,
|
|
Entry {
|
|
edate="2007/01/03",
|
|
estatus=False,
|
|
ecode="*",
|
|
edescription="verizon",
|
|
ecomment="",
|
|
etransactions=[
|
|
RawTransaction {
|
|
taccount="expenses:phone",
|
|
tamount=(Mixed [dollars 95.11]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
},
|
|
RawTransaction {
|
|
taccount="assets:checking",
|
|
tamount=(Mixed [dollars (-95.11)]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
}
|
|
],
|
|
epreceding_comment_lines=""
|
|
}
|
|
,
|
|
Entry {
|
|
edate="2007/01/03",
|
|
estatus=False,
|
|
ecode="*",
|
|
edescription="discover",
|
|
ecomment="",
|
|
etransactions=[
|
|
RawTransaction {
|
|
taccount="liabilities:credit cards:discover",
|
|
tamount=(Mixed [dollars 80]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
},
|
|
RawTransaction {
|
|
taccount="assets:checking",
|
|
tamount=(Mixed [dollars (-80)]),
|
|
tcomment="",
|
|
rttype=RegularTransaction
|
|
}
|
|
],
|
|
epreceding_comment_lines=""
|
|
}
|
|
]
|
|
""
|
|
|
|
ledger7 = cacheLedger rawledger7
|
|
|
|
timelogentry1_str = "i 2007/03/11 16:19:00 hledger\n"
|
|
timelogentry1 = TimeLogEntry 'i' "2007/03/11 16:19:00" "hledger"
|
|
|
|
timelogentry2_str = "o 2007/03/11 16:30:00\n"
|
|
timelogentry2 = TimeLogEntry 'o' "2007/03/11 16:30:00" ""
|
|
|
|
timelog1_str = concat [
|
|
timelogentry1_str,
|
|
timelogentry2_str
|
|
]
|
|
timelog1 = TimeLog [
|
|
timelogentry1,
|
|
timelogentry2
|
|
]
|
|
|
|
------------------------------------------------------------------------------
|
|
-- test utils
|
|
|
|
-- | Get a Test's label, or the empty string.
|
|
tname :: Test -> String
|
|
tname (TestLabel n _) = n
|
|
tname _ = ""
|
|
|
|
-- | Flatten a Test containing TestLists into a list of single tests.
|
|
tflatten :: Test -> [Test]
|
|
tflatten (TestLabel _ t@(TestList _)) = tflatten t
|
|
tflatten (TestList ts) = concatMap tflatten ts
|
|
tflatten t = [t]
|
|
|
|
-- | Filter TestLists in a Test, recursively, preserving the structure.
|
|
tfilter :: (Test -> Bool) -> Test -> Test
|
|
tfilter p (TestLabel l ts) = TestLabel l (tfilter p ts)
|
|
tfilter p (TestList ts) = TestList $ filter (any p . tflatten) $ map (tfilter p) ts
|
|
tfilter _ t = t
|
|
|
|
-- | Combine a list of TestLists into one.
|
|
tlistconcat :: [Test] -> Test
|
|
tlistconcat = foldr (\(TestList as) (TestList bs) -> TestList (as ++ bs)) (TestList [])
|
|
|
|
-- | Assert a parsed thing equals some expected thing, or print a parse error.
|
|
assertparseequal :: (Show a, Eq a) => a -> (Either ParseError a) -> Assertion
|
|
assertparseequal expected parsed = either printParseError (assertequal expected) parsed
|
|
|