mirror of
https://github.com/hanzo-fi/payments.git
synced 2026-10-30 20:31:20 +00:00
CrimsonandGitHub
5bc95257a2
* Add method to fetch latest balance from storage * Add functionality to fetch latest pool balance if at value is current time * Add new API endpoint for fetching latest pool balances * Generate openapi specs for new latest balances endpoint * Add e2e test for new pool balances endpoint
116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/big"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/formancehq/payments/internal/connectors/engine"
|
|
"github.com/formancehq/payments/internal/models"
|
|
"github.com/formancehq/payments/internal/storage"
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
func TestPoolsBalancesLatest(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
store := storage.NewMockStorage(ctrl)
|
|
eng := engine.NewMockEngine(ctrl)
|
|
|
|
s := New(store, eng, false)
|
|
|
|
id := uuid.New()
|
|
poolsAccount := []models.AccountID{{}}
|
|
balancesResponse := []*models.Balance{
|
|
{
|
|
Asset: "EUR/2",
|
|
Balance: big.NewInt(100),
|
|
},
|
|
{
|
|
Asset: "USD/2",
|
|
Balance: big.NewInt(200),
|
|
},
|
|
{
|
|
Asset: "EUR/2",
|
|
Balance: big.NewInt(300),
|
|
},
|
|
}
|
|
|
|
tests := []struct {
|
|
name string
|
|
poolsGetStorageErr error
|
|
accountsBalancesErr error
|
|
expectedError error
|
|
}{
|
|
{
|
|
name: "success",
|
|
poolsGetStorageErr: nil,
|
|
expectedError: nil,
|
|
},
|
|
{
|
|
name: "storage error not found",
|
|
poolsGetStorageErr: storage.ErrNotFound,
|
|
expectedError: newStorageError(storage.ErrNotFound, "cannot get pool"),
|
|
},
|
|
{
|
|
name: "other error",
|
|
poolsGetStorageErr: fmt.Errorf("error"),
|
|
expectedError: newStorageError(fmt.Errorf("error"), "cannot get pool"),
|
|
},
|
|
{
|
|
name: "storage error not found",
|
|
accountsBalancesErr: storage.ErrNotFound,
|
|
expectedError: newStorageError(storage.ErrNotFound, "cannot get latest balances"),
|
|
},
|
|
{
|
|
name: "other error",
|
|
accountsBalancesErr: fmt.Errorf("error"),
|
|
expectedError: newStorageError(fmt.Errorf("error"), "cannot get latest balances"),
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
store.EXPECT().PoolsGet(gomock.Any(), id).Return(&models.Pool{
|
|
ID: id,
|
|
Name: "test",
|
|
CreatedAt: time.Now().Add(-time.Hour),
|
|
PoolAccounts: poolsAccount,
|
|
}, test.poolsGetStorageErr)
|
|
if test.poolsGetStorageErr == nil {
|
|
store.EXPECT().BalancesGetLatest(gomock.Any(), models.AccountID{}).Return(balancesResponse, test.accountsBalancesErr)
|
|
}
|
|
|
|
balances, err := s.PoolsBalances(context.Background(), id)
|
|
if test.expectedError == nil {
|
|
require.NoError(t, err)
|
|
require.NotNil(t, balances)
|
|
foundEUR := false
|
|
foundUSD := false
|
|
for _, balance := range balances {
|
|
switch balance.Asset {
|
|
case "EUR/2":
|
|
require.Equal(t, big.NewInt(400), balance.Amount)
|
|
foundEUR = true
|
|
case "USD/2":
|
|
require.Equal(t, big.NewInt(200), balance.Amount)
|
|
foundUSD = true
|
|
default:
|
|
require.Fail(t, "unexpected asset")
|
|
}
|
|
}
|
|
require.True(t, foundEUR)
|
|
require.True(t, foundUSD)
|
|
} else {
|
|
require.Equal(t, test.expectedError, err)
|
|
}
|
|
})
|
|
}
|
|
}
|