diff --git a/internal/handlers/gnucash_test.go b/internal/handlers/gnucash_test.go new file mode 100644 index 0000000..5efd41e --- /dev/null +++ b/internal/handlers/gnucash_test.go @@ -0,0 +1,185 @@ +package handlers_test + +import ( + "bytes" + "github.com/aclindsa/moneygo/internal/handlers" + "io" + "io/ioutil" + "mime/multipart" + "net/http" + "os" + "testing" +) + +func importGnucash(client *http.Client, filename string) error { + var buf bytes.Buffer + mw := multipart.NewWriter(&buf) + + file, err := os.Open(filename) + if err != nil { + return err + } + defer file.Close() + + filewriter, err := mw.CreateFormFile("gnucash", filename) + if err != nil { + return err + } + if _, err := io.Copy(filewriter, file); err != nil { + return err + } + + mw.Close() + + response, err := client.Post(server.URL+"/v1/imports/gnucash", mw.FormDataContentType(), &buf) + if err != nil { + return err + } + + body, err := ioutil.ReadAll(response.Body) + response.Body.Close() + if err != nil { + return err + } + + var e handlers.Error + err = (&e).Read(string(body)) + if err != nil { + return err + } + if e.ErrorId != 0 || len(e.ErrorString) != 0 { + return &e + } + + return nil +} + +func gnucashAccountBalanceHelper(t *testing.T, client *http.Client, account *handlers.Account, balance string) { + t.Helper() + transactions, err := getAccountTransactions(client, account.AccountId, 0, 0, "") + if err != nil { + t.Fatalf("Couldn't fetch account transactions for '%s': %s\n", account.Name, err) + } + + if transactions.EndingBalance != balance { + t.Errorf("Expected ending balance for '%s' to be '%s', but found %s\n", account.Name, balance, transactions.EndingBalance) + } +} + +func TestImportGnucash(t *testing.T) { + RunWith(t, &data[0], func(t *testing.T, d *TestData) { + // Ensure there's only one USD currency + oldDefault, err := getSecurity(d.clients[0], d.users[0].DefaultCurrency) + if err != nil { + t.Fatalf("Error fetching default security: %s\n", err) + } + d.users[0].DefaultCurrency = d.securities[0].SecurityId + if _, err := updateUser(d.clients[0], &d.users[0]); err != nil { + t.Fatalf("Error updating user: %s\n", err) + } + if err := deleteSecurity(d.clients[0], oldDefault); err != nil { + t.Fatalf("Error removing default security: %s\n", err) + } + + // Import and ensure it didn't return a nasty error code + if err = importGnucash(d.clients[0], "handlers_testdata/example.gnucash"); err != nil { + t.Fatalf("Error importing from Gnucash: %s\n", err) + } + + // Next, find the Expenses/Groceries account and verify it's balance + var income, equity, liabilities, expenses, salary, creditcard, groceries, cable, openingbalances *handlers.Account + accounts, err := getAccounts(d.clients[0]) + if err != nil { + t.Fatalf("Error fetching accounts: %s\n", err) + } + for i, account := range *accounts.Accounts { + if account.Name == "Income" && account.Type == handlers.Income && account.ParentAccountId == -1 { + income = &(*accounts.Accounts)[i] + } else if account.Name == "Equity" && account.Type == handlers.Equity && account.ParentAccountId == -1 { + equity = &(*accounts.Accounts)[i] + } else if account.Name == "Liabilities" && account.Type == handlers.Liability && account.ParentAccountId == -1 { + liabilities = &(*accounts.Accounts)[i] + } else if account.Name == "Expenses" && account.Type == handlers.Expense && account.ParentAccountId == -1 { + expenses = &(*accounts.Accounts)[i] + } + } + if income == nil { + t.Fatalf("Couldn't find 'Income' account") + } + if equity == nil { + t.Fatalf("Couldn't find 'Equity' account") + } + if liabilities == nil { + t.Fatalf("Couldn't find 'Liabilities' account") + } + if expenses == nil { + t.Fatalf("Couldn't find 'Expenses' account") + } + for i, account := range *accounts.Accounts { + if account.Name == "Salary" && account.Type == handlers.Income && account.ParentAccountId == income.AccountId { + salary = &(*accounts.Accounts)[i] + } else if account.Name == "Opening Balances" && account.Type == handlers.Equity && account.ParentAccountId == equity.AccountId { + openingbalances = &(*accounts.Accounts)[i] + } else if account.Name == "Credit Card" && account.Type == handlers.Liability && account.ParentAccountId == liabilities.AccountId { + creditcard = &(*accounts.Accounts)[i] + } else if account.Name == "Groceries" && account.Type == handlers.Expense && account.ParentAccountId == expenses.AccountId { + groceries = &(*accounts.Accounts)[i] + } else if account.Name == "Cable" && account.Type == handlers.Expense && account.ParentAccountId == expenses.AccountId { + cable = &(*accounts.Accounts)[i] + } + } + if salary == nil { + t.Fatalf("Couldn't find 'Income/Salary' account") + } + if openingbalances == nil { + t.Fatalf("Couldn't find 'Equity/Opening Balances") + } + if creditcard == nil { + t.Fatalf("Couldn't find 'Liabilities/Credit Card' account") + } + if groceries == nil { + t.Fatalf("Couldn't find 'Expenses/Groceries' account") + } + if cable == nil { + t.Fatalf("Couldn't find 'Expenses/Cable' account") + } + + gnucashAccountBalanceHelper(t, d.clients[0], salary, "-998.34") + gnucashAccountBalanceHelper(t, d.clients[0], creditcard, "-272.03") + gnucashAccountBalanceHelper(t, d.clients[0], openingbalances, "-21014.33") + gnucashAccountBalanceHelper(t, d.clients[0], groceries, "287.56") // 87.19 from preexisting transactions and 200.37 from Gnucash + gnucashAccountBalanceHelper(t, d.clients[0], cable, "89.98") + + var ge *handlers.Security + securities, err := getSecurities(d.clients[0]) + if err != nil { + t.Fatalf("Error fetching securities: %s\n", err) + } + for i, security := range *securities.Securities { + if security.Symbol == "GE" { + ge = (*securities.Securities)[i] + } + } + if ge == nil { + t.Fatalf("Couldn't find GE security") + } + + prices, err := getPrices(d.clients[0], ge.SecurityId) + if err != nil { + t.Fatalf("Error fetching prices: %s\n", err) + } + var p1787, p2894, p3170 bool + for _, price := range *prices.Prices { + if price.CurrencyId == d.securities[0].SecurityId && price.Value == "17.87" { + p1787 = true + } else if price.CurrencyId == d.securities[0].SecurityId && price.Value == "28.94" { + p2894 = true + } else if price.CurrencyId == d.securities[0].SecurityId && price.Value == "31.70" { + p3170 = true + } + } + if !p1787 || !p2894 || !p3170 { + t.Errorf("Error finding expected prices\n") + } + }) +} diff --git a/internal/handlers/handlers_testdata/example.gnucash b/internal/handlers/handlers_testdata/example.gnucash new file mode 100644 index 0000000..e3a6ef2 Binary files /dev/null and b/internal/handlers/handlers_testdata/example.gnucash differ diff --git a/internal/handlers/securities.go b/internal/handlers/securities.go index 0ca96d0..5aabaa1 100644 --- a/internal/handlers/securities.go +++ b/internal/handlers/securities.go @@ -38,7 +38,7 @@ type Security struct { Symbol string // Number of decimal digits (to the right of the decimal point) this // security is precise to - Precision int + Precision int `db:"Preciseness"` Type SecurityType // AlternateId is CUSIP for Type=Stock, ISO4217 for Type=Currency AlternateId string @@ -206,7 +206,7 @@ func ImportGetCreateSecurity(tx *Tx, userid int64, security *Security) (*Securit var securities []*Security - _, err := tx.Select(&securities, "SELECT * from securities where UserId=? AND Type=? AND AlternateId=? AND Precision=?", userid, security.Type, security.AlternateId, security.Precision) + _, err := tx.Select(&securities, "SELECT * from securities where UserId=? AND Type=? AND AlternateId=? AND Preciseness=?", userid, security.Type, security.AlternateId, security.Precision) if err != nil { return nil, err }