138 lines
2.4 KiB
Go
138 lines
2.4 KiB
Go
package ledger
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestFileDeltas(t *testing.T) {
|
|
d := func(s string) time.Time {
|
|
v, _ := time.Parse("2006-01-02", s)
|
|
return v
|
|
}
|
|
happy := []Delta{
|
|
{
|
|
Date: d("2022-12-12"),
|
|
Account: "AssetAccount:Cash:Fidelity76",
|
|
Value: -97.92,
|
|
Currency: USD,
|
|
},
|
|
{
|
|
Date: d("2022-12-12"),
|
|
Account: "Withdrawal:0:SharedHome:DominionEnergy",
|
|
Value: 97.92,
|
|
Currency: USD,
|
|
},
|
|
{
|
|
Date: d("2022-12-12"),
|
|
Account: "AssetAccount:Cash:Fidelity76",
|
|
Value: -1.00,
|
|
Currency: USD,
|
|
},
|
|
{
|
|
Date: d("2022-12-12"),
|
|
Account: "Debts:Credit:ChaseFreedomUltdVisa",
|
|
Value: 1.00,
|
|
Currency: USD,
|
|
},
|
|
}
|
|
|
|
cases := map[string][]Delta{
|
|
"empty": nil,
|
|
"one": happy[:2],
|
|
"happy": happy[:],
|
|
}
|
|
|
|
for name, d := range cases {
|
|
want := d
|
|
t.Run(name, func(t *testing.T) {
|
|
f, err := NewFile("./testdata/" + name + ".dat")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
deltas, err := f.Deltas()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if len(deltas) != len(want) {
|
|
t.Error(len(deltas))
|
|
}
|
|
for i := range want {
|
|
if i >= len(deltas) {
|
|
break
|
|
}
|
|
if want[i] != deltas[i] {
|
|
t.Errorf("[%d] \n\twant=%+v, \n\t got=%+v", i, want[i], deltas[i])
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReadTransaction(t *testing.T) {
|
|
d := func(s string) time.Time {
|
|
v, _ := time.Parse("2006-01-02", s)
|
|
return v
|
|
}
|
|
cases := map[string]struct {
|
|
input string
|
|
want transaction
|
|
err error
|
|
}{
|
|
"empty": {
|
|
input: "",
|
|
want: transaction{},
|
|
err: io.EOF,
|
|
},
|
|
"white space": {
|
|
input: " ",
|
|
want: transaction{},
|
|
err: io.EOF,
|
|
},
|
|
"verbose": {
|
|
input: `
|
|
2003-04-05 Reasoning here
|
|
A:B $1.00
|
|
C:D $-1.00
|
|
`,
|
|
want: transaction{
|
|
date: d("2003-04-05"),
|
|
description: "Reasoning here",
|
|
payee: "",
|
|
recipients: []transactionRecipient{
|
|
{
|
|
name: "A:B",
|
|
value: 1.0,
|
|
currency: "$",
|
|
},
|
|
{
|
|
name: "C:D",
|
|
value: -1.0,
|
|
currency: "$",
|
|
},
|
|
},
|
|
},
|
|
err: io.EOF,
|
|
},
|
|
}
|
|
|
|
for name, d := range cases {
|
|
c := d
|
|
t.Run(name, func(t *testing.T) {
|
|
got, err := readTransaction(strings.NewReader(c.input))
|
|
if err != c.err {
|
|
t.Error(err)
|
|
}
|
|
|
|
if fmt.Sprintf("%+v", got) != fmt.Sprintf("%+v", c.want) {
|
|
t.Errorf("want\n\t%+v, got\n\t%+v", c.want, got)
|
|
}
|
|
})
|
|
}
|
|
}
|