148 lines
3.7 KiB
Go
148 lines
3.7 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/itsscb/df/util"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func createRandomPayment(t *testing.T) Payment {
|
|
account := createRandomCustomer(t)
|
|
require.NotEmpty(t, account)
|
|
|
|
creator := util.RandomUser()
|
|
|
|
arg := CreatePaymentParams{
|
|
CustomerID: account.ID,
|
|
PaymentCategory: util.RandomUser(),
|
|
Bankname: sql.NullString{
|
|
Valid: true,
|
|
String: util.RandomUser(),
|
|
},
|
|
IBAN: sql.NullString{
|
|
Valid: true,
|
|
String: util.RandomUser(),
|
|
},
|
|
BIC: sql.NullString{
|
|
Valid: true,
|
|
String: util.RandomUser(),
|
|
},
|
|
PaypalAccount: sql.NullString{
|
|
Valid: true,
|
|
String: util.RandomUser(),
|
|
},
|
|
PaypalID: sql.NullString{
|
|
Valid: true,
|
|
String: util.RandomUser(),
|
|
},
|
|
PaymentSystem: sql.NullString{
|
|
Valid: true,
|
|
String: util.RandomUser(),
|
|
},
|
|
Type: util.RandomUser(),
|
|
Creator: creator,
|
|
Changer: creator,
|
|
}
|
|
|
|
person, err := testQueries.CreatePayment(context.Background(), arg)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, person)
|
|
|
|
require.Equal(t, arg.PaymentCategory, person.PaymentCategory)
|
|
require.Equal(t, arg.Bankname, person.Bankname)
|
|
require.Equal(t, arg.CustomerID, person.CustomerID)
|
|
require.Equal(t, arg.IBAN, person.IBAN)
|
|
require.Equal(t, arg.BIC, person.BIC)
|
|
require.Equal(t, arg.PaypalAccount, person.PaypalAccount)
|
|
require.Equal(t, arg.PaymentSystem, person.PaymentSystem)
|
|
require.Equal(t, arg.PaypalID, person.PaypalID)
|
|
require.Equal(t, arg.Creator, person.Creator)
|
|
require.Equal(t, arg.Type, person.Type)
|
|
|
|
require.NotZero(t, person.ID)
|
|
require.NotZero(t, person.Created)
|
|
|
|
return person
|
|
}
|
|
|
|
func TestCreatePayment(t *testing.T) {
|
|
createRandomPayment(t)
|
|
}
|
|
|
|
func TestGetPayment(t *testing.T) {
|
|
newperson := createRandomPayment(t)
|
|
require.NotEmpty(t, newperson)
|
|
|
|
person, err := testQueries.GetPayment(context.Background(), newperson.ID)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, person)
|
|
|
|
require.Equal(t, newperson.PaymentCategory, person.PaymentCategory)
|
|
require.Equal(t, newperson.Bankname, person.Bankname)
|
|
require.Equal(t, newperson.CustomerID, person.CustomerID)
|
|
require.Equal(t, newperson.IBAN, person.IBAN)
|
|
require.Equal(t, newperson.BIC, person.BIC)
|
|
require.Equal(t, newperson.PaypalAccount, person.PaypalAccount)
|
|
require.Equal(t, newperson.PaymentSystem, person.PaymentSystem)
|
|
require.Equal(t, newperson.PaypalID, person.PaypalID)
|
|
require.Equal(t, newperson.Creator, person.Creator)
|
|
require.Equal(t, newperson.Type, person.Type)
|
|
|
|
require.WithinDuration(t, newperson.Created, person.Created, time.Second)
|
|
}
|
|
|
|
func TestDeletePayment(t *testing.T) {
|
|
person1 := createRandomPayment(t)
|
|
err := testQueries.DeletePayment(context.Background(), person1.ID)
|
|
require.NoError(t, err)
|
|
|
|
person2, err := testQueries.GetPayment(context.Background(), person1.ID)
|
|
require.Error(t, err)
|
|
require.EqualError(t, err, sql.ErrNoRows.Error())
|
|
require.Empty(t, person2)
|
|
}
|
|
|
|
func TestUpdatePayment(t *testing.T) {
|
|
person1 := createRandomPayment(t)
|
|
require.NotEmpty(t, person1)
|
|
|
|
arg := UpdatePaymentParams{
|
|
ID: person1.ID,
|
|
Bankname: sql.NullString{
|
|
String: util.RandomUser(),
|
|
Valid: true,
|
|
},
|
|
}
|
|
|
|
person2, err := testQueries.UpdatePayment(context.Background(), arg)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, person2)
|
|
|
|
require.Equal(t, person1.ID, person2.ID)
|
|
require.Equal(t, person1.PaymentCategory, person2.PaymentCategory)
|
|
require.NotEqual(t, person1.Bankname, person2.Bankname)
|
|
}
|
|
|
|
func TestListPayments(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
createRandomPayment(t)
|
|
}
|
|
|
|
arg := ListPaymentsParams{
|
|
Limit: 5,
|
|
Offset: 5,
|
|
}
|
|
|
|
persons, err := testQueries.ListPayments(context.Background(), arg)
|
|
require.NoError(t, err)
|
|
require.Len(t, persons, 5)
|
|
|
|
for _, person := range persons {
|
|
require.NotEmpty(t, person)
|
|
}
|
|
}
|