. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/sporto/kic/api/models"
"github.com/sporto/kic/api/services/accounts"
"github.com/sporto/kic/api/services/transactions"
"time"
// "os"
"log"
)
var _ = Describe("CreateServ", func() {
var (
createAccountServ accounts.CreateServ
updateAccountServ accounts.UpdateServ
service transactions.CreateServ
account models.Account
transaction *models.Transaction
err error
)
BeforeEach(func() {
// log.Println("---- BeforeEach -------")
// log.Println("dbSession", dbSession)
accountIn := *&models.Account{CurrentBalance: 100, LastInterestPaid: time.Now()}
account, err = createAccountServ.Run(dbSession, accountIn)
// log.Println("Account created ", account)
if err != nil {
log.Println("Account not created because ", err)
}
transaction = &models.Transaction{AccountId: account.Id, Debit: 50, Credit: 100}
})
func (serv *CreateServ) Run(dbSession *r.Session, transactionIn models.Transaction) (transactionOut models.Transaction, err error) {
// fail if transaction is already saved
if transactionIn.Id != "" {
err = errors.New("Transaction Id must be nil")
return
}
// fail if not account id provided
if transactionIn.AccountId == "" {
err = errors.New("Account Id cannot be nil")
return
}
// fail if no credit or debit provided
if transactionIn.Credit <= 0 && transactionIn.Debit <= 0 {
err = errors.New("Credit or Debit must be provided")
return
}
// check that the account exist
getServ := new(accounts.GetServ)
account, err := getServ.Run(dbSession, transactionIn.AccountId)
if err != nil {
return
}
// check that the transaction is valid e.g. enough balance
if transactionIn.Debit > account.CurrentBalance {
err = errors.New("Not enough balance")
return
}
// check that interest has been p aid
// unless kind is interest
if transactionIn.Kind != KindInterest {
dur := time.Now().Sub(account.LastInterestPaid)
if dur.Hours() > 24 {
err = errors.New("Interest not updated")
return
}
}
transactionIn.CreatedAt = time.Now()
transactionIn.UpdatedAt = time.Now()
// update the current account balance
account.CurrentBalance += transactionIn.Credit
account.CurrentBalance -= transactionIn.Debit
// also update the balance in the transaction
transactionIn.Balance = account.CurrentBalance
// save the transaction
response, err := r.Table("transactions").Insert(transactionIn).RunWrite(dbSession)
if err != nil {
return
}
id := response.GeneratedKeys[0]
// get the transaction out
transactionOut, err = new(GetServ).Run(dbSession, id)
if err != nil {
return
}
updateAccountServ := new(accounts.UpdateServ)
_, err = updateAccountServ.Run(dbSession, account)
if err != nil {
return
}
return
}
package accounts_test
import (
"fmt"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/sporto/kic/api/models"
"github.com/sporto/kic/api/services/accounts"
)
var _ = Describe("UpdateServ", func() {
var (
service accounts.UpdateServ
createServ accounts.CreateServ
account models.Account
err error
)
BeforeEach(func() {
// create an account for testing
accountIn := &models.Account{}
account, err = createServ.Run(dbSession, *accountIn)
if err != nil {
fmt.Println(err)
}
// fmt.Println("Account saved", accountIn.Id)
})
It("Saved the account", func() {