account store tests

BankAccount/pkg/tests/accountstore_test.go

@@ -0,0 +1,95 @@
+package tests
+
+import (
+	"bankacc/pkg/entities"
+	"bankacc/pkg/store"
+	"database/sql"
+	"log"
+
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/stretchr/testify/suite"
+)
+
+func MySQLInit() *sql.DB {
+	dbDriver := "mysql"
+	dbUser := "root"
+	dbPass := "Password1!"
+	dbName := "BankAccount?parseTime=true"
+	db, err := sql.Open(dbDriver, dbUser+":"+dbPass+"@/"+dbName)
+	if err != nil {
+		log.Println(err)
+	}
+	return db
+}
+
+type TearDownTestSuite interface {
+	TearDownTest()
+}
+
+type AccountTestSuite struct {
+	suite.Suite
+	Account      *entities.Account
+	Accounts     []entities.Account
+	AccountsP    []*entities.Account
+	AccountStore store.AccountModel
+	Db           *sql.DB
+}
+
+func (suite *AccountTestSuite) SetupTest() {
+	var err error
+	suite.Db = MySQLInit()
+	account := store.NewAccountStoreModel(suite.Db)
+	suite.Accounts = []entities.Account{
+		{
+			UserId:   1,
+			Balance:  1000,
+			Currency: "Euros",
+		},
+		{
+			UserId:   2,
+			Balance:  500,
+			Currency: "$",
+		},
+		{
+			UserId:   3,
+			Balance:  1000,
+			Currency: "Euros",
+		},
+		{
+			UserId:   1,
+			Balance:  2000,
+			Currency: "$",
+		},
+	}
+
+	for _, value := range suite.Accounts {
+		suite.Account, err = account.InsertAccount(value.UserId, value.Balance, value.Currency)
+		if err != nil {
+			suite.T().Fatal("Unable to run InsertAccount store func")
+		}
+		suite.AccountsP = append(suite.AccountsP, suite.Account)
+	}
+}
+
+func (suite *AccountTestSuite) TestGetAccountsById() {
+	store := store.NewAccountStoreModel(suite.Db)
+	var err error
+	var accounts []*entities.Account
+	for _, value := range suite.AccountsP {
+		accounts, err = store.GetAccountsByUserId(value.UserId)
+		if err != nil {
+			suite.T().Fatal("Unable to run GetAccountsById store func")
+		}
+	}
+
+	for _, value := range suite.AccountsP {
+		for i := range accounts {
+			if value.Id == accounts[i].Id {
+				suite.Equal(value.UserId, accounts[i].UserId)
+				suite.Equal(value.Balance, accounts[i].Balance)
+				suite.Equal(value.Currency, accounts[i].Currency)
+				suite.Equal(value.Status, accounts[i].Status)
+			}
+		}
+	}
+}