Files
payments/internal/events/balance_test.go
fgueryandGitHub a63e3a20a8 feat(EN-214): create events via storage instead of manually calling event activity (#594)
* feat(EN-214): add outbox table migration

* feat(EN-214): add outbox models and storage layer

* feat(EN-214): implement outbox publisher activity

* feat(EN-214): implement outbox schedule activity

* feat(EN-214): add outbox publisher workflow

* feat(EN-214): migrate accounts to outbox pattern

* feat(EN-214): migrate balances to outbox pattern

* feat(EN-214): migrate payments to outbox pattern

* feat(EN-214): migrate bank accounts to outbox pattern

* feat(EN-214): migrate pools to outbox pattern

* feat(EN-214): migrate payment initiations to outbox pattern

* feat(EN-214): migrate PSU and connections to outbox pattern

* feat(EN-214): migrate connectors to outbox pattern

* feat(EN-214): remove legacy send_events workflow

* feat(EN-214): update workflows to use outbox pattern

* feat(EN-214): improve outbox idempotency and error handling

* feat(EN-214): update storage tasks and tests

* test(EN-214): update e2e tests for outbox pattern

* test(EN-214): clean up workflow tests

* chore(EN-214): update docker compose configuration

* feat(EN-214): add e2e run config in ide

* fix: add back docker-compose-dev run config

* chore: improve connector uninstall comments & remove connector id being passed

* fix: ensure events are not pushed if balances are not created

* fix: ensure err is not overwritten when using rollback

* fix: ensure we use v3 provider format

* chore: apply the nitpick comments from coderabbit

* chore: consistentely use e() error handling

* feat: optionally disable publisher schedule to make e2e test more consistent

* fix: improve error handling on uninstall connector

* avoid using custom payload and rely on existing interfaces; also avoid using different types for events

* fix: remove unecessary unmarshalling

* fix: ensure that rollbackOnTxError picks up errors on tx.Commit()

* fix: add extra validation on iban & account number in the test

* test: cover marshall/unmarshalling of event payload

* fix: increase test timeout

* try to make test more resiliant

* fix: wrong pointer use in loop

* fix: error formatting issue

* fix: move oubtox schedule creation out of an activity and do it in the worker itself

* feat: OutboxEventDeleteAndSend as batch

* change outbox_event to have id only instead of idempotency key

* make e2e test less brittle, save eventID as idempotency key in publication

* feat: make outbox_event keep an history

* feat(EN-214): added a schedule to cleanup outbox events

* test: make cleanup actually clean

* ignore linting rules for now

* feat: do not add psu metadata to event anymore as it's irrelevant

* fix: create local variable instead of relying on pointers, even if not 100% necessary

* fix: remove temporal schedule describe calls
2025-11-24 16:05:42 +00:00

186 lines
4.7 KiB
Go

package events
import (
"encoding/json"
"math/big"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestBalanceMessagePayload_MarshalJSON(t *testing.T) {
t.Parallel()
t.Run("with balance", func(t *testing.T) {
t.Parallel()
balance := big.NewInt(123456789)
payload := BalanceMessagePayload{
AccountID: "acc123",
ConnectorID: "conn123",
Provider: "test",
CreatedAt: time.Date(2024, 1, 1, 12, 0, 0, 0, time.UTC),
LastUpdatedAt: time.Date(2024, 1, 2, 12, 0, 0, 0, time.UTC),
Asset: "USD/2",
Balance: balance,
}
data, err := json.Marshal(&payload)
require.NoError(t, err)
var result map[string]interface{}
err = json.Unmarshal(data, &result)
require.NoError(t, err)
assert.Equal(t, "acc123", result["accountID"])
assert.Equal(t, "conn123", result["connectorID"])
assert.Equal(t, "test", result["provider"])
assert.Equal(t, "USD/2", result["asset"])
assert.Equal(t, "123456789", result["balance"])
})
t.Run("with nil balance", func(t *testing.T) {
t.Parallel()
payload := BalanceMessagePayload{
AccountID: "acc123",
ConnectorID: "conn123",
Provider: "test",
CreatedAt: time.Date(2024, 1, 1, 12, 0, 0, 0, time.UTC),
LastUpdatedAt: time.Date(2024, 1, 2, 12, 0, 0, 0, time.UTC),
Asset: "USD/2",
Balance: nil,
}
data, err := json.Marshal(&payload)
require.NoError(t, err)
var result map[string]interface{}
err = json.Unmarshal(data, &result)
require.NoError(t, err)
assert.Nil(t, result["balance"])
})
t.Run("with zero balance", func(t *testing.T) {
t.Parallel()
balance := big.NewInt(0)
payload := BalanceMessagePayload{
AccountID: "acc123",
ConnectorID: "conn123",
Provider: "test",
CreatedAt: time.Date(2024, 1, 1, 12, 0, 0, 0, time.UTC),
LastUpdatedAt: time.Date(2024, 1, 2, 12, 0, 0, 0, time.UTC),
Asset: "USD/2",
Balance: balance,
}
data, err := json.Marshal(&payload)
require.NoError(t, err)
var result map[string]interface{}
err = json.Unmarshal(data, &result)
require.NoError(t, err)
assert.Equal(t, "0", result["balance"])
})
}
func TestBalanceMessagePayload_UnmarshalJSON(t *testing.T) {
t.Parallel()
t.Run("with valid balance string", func(t *testing.T) {
t.Parallel()
jsonData := `{
"accountID": "acc123",
"connectorID": "conn123",
"provider": "test",
"createdAt": "2024-01-01T12:00:00Z",
"lastUpdatedAt": "2024-01-02T12:00:00Z",
"asset": "USD/2",
"balance": "123456789"
}`
var payload BalanceMessagePayload
err := json.Unmarshal([]byte(jsonData), &payload)
require.NoError(t, err)
assert.Equal(t, "acc123", payload.AccountID)
assert.Equal(t, "conn123", payload.ConnectorID)
assert.Equal(t, "test", payload.Provider)
assert.Equal(t, "USD/2", payload.Asset)
assert.NotNil(t, payload.Balance)
assert.Equal(t, "123456789", payload.Balance.String())
})
t.Run("with nil balance", func(t *testing.T) {
t.Parallel()
jsonData := `{
"accountID": "acc123",
"connectorID": "conn123",
"provider": "test",
"createdAt": "2024-01-01T12:00:00Z",
"lastUpdatedAt": "2024-01-02T12:00:00Z",
"asset": "USD/2"
}`
var payload BalanceMessagePayload
err := json.Unmarshal([]byte(jsonData), &payload)
require.NoError(t, err)
assert.Nil(t, payload.Balance)
})
t.Run("with invalid balance string", func(t *testing.T) {
t.Parallel()
jsonData := `{
"accountID": "acc123",
"connectorID": "conn123",
"provider": "test",
"createdAt": "2024-01-01T12:00:00Z",
"lastUpdatedAt": "2024-01-02T12:00:00Z",
"asset": "USD/2",
"balance": "invalid-number"
}`
var payload BalanceMessagePayload
err := json.Unmarshal([]byte(jsonData), &payload)
require.Error(t, err)
assert.Contains(t, err.Error(), "invalid balance string")
})
t.Run("round-trip marshalling", func(t *testing.T) {
t.Parallel()
original := BalanceMessagePayload{
AccountID: "acc123",
ConnectorID: "conn123",
Provider: "test",
CreatedAt: time.Date(2024, 1, 1, 12, 0, 0, 0, time.UTC),
LastUpdatedAt: time.Date(2024, 1, 2, 12, 0, 0, 0, time.UTC),
Asset: "USD/2",
Balance: big.NewInt(987654321),
}
data, err := json.Marshal(&original)
require.NoError(t, err)
var unmarshaled BalanceMessagePayload
err = json.Unmarshal(data, &unmarshaled)
require.NoError(t, err)
assert.Equal(t, original.AccountID, unmarshaled.AccountID)
assert.Equal(t, original.ConnectorID, unmarshaled.ConnectorID)
assert.Equal(t, original.Provider, unmarshaled.Provider)
assert.Equal(t, original.Asset, unmarshaled.Asset)
assert.Equal(t, original.Balance.String(), unmarshaled.Balance.String())
})
}