initial commit

core/model/transaction_test.go

@@ -0,0 +1,141 @@
+package model
+
+import (
+	"errors"
+	"github.com/openaccounting/oa-server/core/model/db"
+	"github.com/openaccounting/oa-server/core/model/types"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	"testing"
+	"time"
+)
+
+type TdTransaction struct {
+	db.Datastore
+	mock.Mock
+}
+
+func (td *TdTransaction) GetOrg(orgId string, userId string) (*types.Org, error) {
+	org := &types.Org{
+		Currency: "USD",
+	}
+
+	return org, nil
+}
+
+func (td *TdTransaction) GetPermissionedAccountIds(userId string, orgId string, tokenId string) ([]string, error) {
+	return []string{"1", "2"}, nil
+}
+
+func (td *TdTransaction) GetAccountsByOrgId(orgId string) ([]*types.Account, error) {
+	return []*types.Account{&types.Account{Id: "1", Currency: "USD"}, &types.Account{Id: "2"}}, nil
+}
+
+func (td *TdTransaction) InsertTransaction(transaction *types.Transaction) (err error) {
+	return nil
+}
+
+func (td *TdTransaction) GetTransactionById(id string) (*types.Transaction, error) {
+	args := td.Called(id)
+	return args.Get(0).(*types.Transaction), args.Error(1)
+}
+
+func (td *TdTransaction) UpdateTransaction(oldId string, transaction *types.Transaction) error {
+	args := td.Called(oldId, transaction)
+	return args.Error(0)
+}
+
+func (td *TdTransaction) GetOrgUserIds(id string) ([]string, error) {
+	return []string{"1"}, nil
+}
+
+func TestCreateTransaction(t *testing.T) {
+	tests := map[string]struct {
+		err error
+		tx  *types.Transaction
+	}{
+		"successful": {
+			err: nil,
+			tx: &types.Transaction{
+				"1",
+				"2",
+				"3",
+				time.Now(),
+				time.Now(),
+				time.Now(),
+				"description",
+				"",
+				false,
+				[]*types.Split{
+					&types.Split{"1", "1", 1000, 1000},
+					&types.Split{"1", "2", -1000, -1000},
+				},
+			},
+		},
+		"bad split amounts": {
+			err: errors.New("splits must add up to 0"),
+			tx: &types.Transaction{
+				"1",
+				"2",
+				"3",
+				time.Now(),
+				time.Now(),
+				time.Now(),
+				"description",
+				"",
+				false,
+				[]*types.Split{
+					&types.Split{"1", "1", 1000, 1000},
+					&types.Split{"1", "2", -500, -500},
+				},
+			},
+		},
+		"lacking permission": {
+			err: errors.New("user does not have permission to access account 3"),
+			tx: &types.Transaction{
+				"1",
+				"2",
+				"3",
+				time.Now(),
+				time.Now(),
+				time.Now(),
+				"description",
+				"",
+				false,
+				[]*types.Split{
+					&types.Split{"1", "1", 1000, 1000},
+					&types.Split{"1", "3", -1000, -1000},
+				},
+			},
+		},
+		"nativeAmount mismatch": {
+			err: errors.New("nativeAmount must equal amount for native currency splits"),
+			tx: &types.Transaction{
+				"1",
+				"2",
+				"3",
+				time.Now(),
+				time.Now(),
+				time.Now(),
+				"description",
+				"",
+				false,
+				[]*types.Split{
+					&types.Split{"1", "1", 1000, 500},
+					&types.Split{"1", "2", -1000, -500},
+				},
+			},
+		},
+	}
+
+	for name, test := range tests {
+		t.Logf("Running test case: %s", name)
+
+		td := &TdTransaction{}
+		model := NewModel(td, nil, types.Config{})
+
+		err := model.CreateTransaction(test.tx)
+
+		assert.Equal(t, err, test.err)
+	}
+}