rules_test.go

package coin

import (
	"strings"
	"testing"

	"github.com/mkobetic/coin/assert"
)

func init() {
	r := strings.NewReader(`
commodity CAD
  format 1.00 CAD

commodity USD
  format 1.00 USD

account Assets:Bank:Checking
account Assets:Bank:Savings
account Expenses:Groceries
account Expenses:Auto
account Expenses:Auto:Gas
account Expenses:Miscellaneous
account Income:Salary
account Income:Interest
account Liabilities:Credit:MC
`)
	Load(r, "")
	ResolveAccounts()
}

var sample = `
common
  Expenses:Groceries       FRESHCO|COSTCO WHOLESALE|FARM BOY|LOBLAWS
  Expenses:Auto:Gas        COSTCO GAS|PETROCAN|SHELL
389249328477983 Assets:Bank:Savings
  Income:Interest     Interest
392843029797099 Assets:Bank:Checking
    @common
	Income:Salary       ACME PAY 
479347938749398 Liabilities:Credit:MC
  Expenses:Auto            HUYNDAI|TOYOTA
  ; service/maintenance
  @common
  Expenses:Miscellaneous   
`

func Test_ReadRules(t *testing.T) {
	r := strings.NewReader(sample)
	rules, err := ReadRules(r)
	assert.NoError(t, err)
	assert.Equal(t, len(rules.Accounts), 3)
	mc := rules.Accounts["479347938749398"]
	assert.NotNil(t, mc)
	assert.Equal(t, len(mc.Rules), 3)
	r1, ok := mc.Rules[0].(*Rule)
	assert.True(t, ok)
	assert.Equal(t, len(r1.Notes), 1)
	assert.Equal(t, r1.Notes[0], "service/maintenance")
}

func Test_Classification(t *testing.T) {
	r := strings.NewReader(sample)
	rules, err := ReadRules(r)
	assert.NoError(t, err)
	for _, fix := range []struct {
		from  string
		payee string
		to    string
	}{
		{"479347938749398", "[TR] COSTCO WHOLESALE #9239", "Expenses:Groceries"},
		{"479347938749398", "JOE'S DINER", "Expenses:Miscellaneous"},
	} {
		rule := rules.AccountRulesFor(fix.from).RuleFor(fix.payee)
		assert.NotNil(t, rule)
		account := rule.Account
		if account.FullName != fix.to {
			t.Errorf("mismatched\nexp: %s\ngot: %s\n", fix.to, account.FullName)
		}
	}
}