mirror of
https://github.com/aclindsa/moneygo.git
synced 2024-10-30 15:50:04 -04:00
Aaron Lindsay
30d4515780
Also fix up a "bug" caused by financial institutions not deciding on which sign to use for an OFX field...
100 lines
3.5 KiB
Go
100 lines
3.5 KiB
Go
package handlers_test
|
|
|
|
import (
|
|
"github.com/aclindsa/moneygo/internal/handlers"
|
|
"net/http"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
func importOFX(client *http.Client, accountid int64, filename string) error {
|
|
return uploadFile(client, filename, "/v1/accounts/"+strconv.FormatInt(accountid, 10)+"/imports/ofxfile")
|
|
}
|
|
|
|
func TestImportOFXChecking(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 = importOFX(d.clients[0], d.accounts[1].AccountId, "handlers_testdata/checking_20171126.ofx"); err != nil {
|
|
t.Fatalf("Error importing OFX: %s\n", err)
|
|
}
|
|
accountBalanceHelper(t, d.clients[0], &d.accounts[1], "2493.19")
|
|
|
|
if err = importOFX(d.clients[0], d.accounts[1].AccountId, "handlers_testdata/checking_20171129.ofx"); err != nil {
|
|
t.Fatalf("Error importing OFX: %s\n", err)
|
|
}
|
|
accountBalanceHelper(t, d.clients[0], &d.accounts[1], "5336.27")
|
|
})
|
|
}
|
|
|
|
func TestImportOFXCreditCard(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 = importOFX(d.clients[0], d.accounts[7].AccountId, "handlers_testdata/creditcard.ofx"); err != nil {
|
|
t.Fatalf("Error importing OFX: %s\n", err)
|
|
}
|
|
accountBalanceHelper(t, d.clients[0], &d.accounts[7], "-4.49")
|
|
})
|
|
}
|
|
|
|
func TestImportOFX401kMutualFunds(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)
|
|
}
|
|
|
|
account := &handlers.Account{
|
|
SecurityId: d.securities[0].SecurityId,
|
|
UserId: d.users[0].UserId,
|
|
ParentAccountId: -1,
|
|
Type: handlers.Investment,
|
|
Name: "401k",
|
|
}
|
|
|
|
account, err = createAccount(d.clients[0], account)
|
|
if err != nil {
|
|
t.Fatalf("Error creating 401k account: %s\n", err)
|
|
}
|
|
|
|
// Import and ensure it didn't return a nasty error code
|
|
if err = importOFX(d.clients[0], account.AccountId, "handlers_testdata/401k_mutualfunds.ofx"); err != nil {
|
|
t.Fatalf("Error importing OFX: %s\n", err)
|
|
}
|
|
accountBalanceHelper(t, d.clients[0], account, "-192.10")
|
|
})
|
|
}
|