payments/internal/connectors/engine/workflow/fetch_payments_test.go at main

mirror of https://github.com/hanzo-fi/payments.git synced 2026-10-30 20:31:20 +00:00

Files

T

zeekay 7d727874d5 chore: rebrand Formance Payments -> Hanzo Payments (hanzo-fi)

- go.mod module path + all self-imports: github.com/formancehq/payments -> github.com/hanzo-fi/payments
  (1590 .go files / 3152 refs; 20 modules: root, pkg/client, pkg/domain, 18 ce/plugins).
  External dep kept: formancehq/go-libs.
- plugins: each ce/plugins/* consumed a *published* pkg/domain; added local
  'replace .../pkg/domain => ../../../pkg/domain' (consistent with root) so every
  standalone plugin module builds in-monorepo. compile-plugin.sh MOD var rebranded too.
- branding: README, LICENSE (dual MIT+EE; upstream copyright retained + Hanzo added),
  goreleaser (release images -> registry.hanzo.ai/hanzo-fi, OCI labels, homepage),
  proto go_package, renovate, speakeasy gen.yaml, connector-template *.gotpl.
- CI: removed Numary/Formance workflows + composite action; added canonical hanzo.yml +
  cicd.yml (hanzoai/ci, gated HANZO_CI_ENABLED) + secretless ci.yml (go build/vet).
- kept as data/compat keys: wire capability enums ALLOW_FORMANCE_*, generated SDK type
  client.Formance, domain field FormanceRedirectURL, OTEL_SERVICE_NAME, upstream base image.
- verified: go build ./... (default + -tags ee) + all 20 nested modules green.

2026-07-05 13:08:55 -07:00

595 lines

24 KiB

Go

Raw Permalink Blame History

 package workflow
 import (
 	"context"
 	"errors"
 	"fmt"
 	"github.com/hanzo-fi/payments/internal/connectors/engine/activities"
 	"github.com/hanzo-fi/payments/pkg/domain/models"
 	"github.com/stretchr/testify/mock"
 	"go.temporal.io/sdk/temporal"
 )
 func (s *UnitTestSuite) Test_FetchNextPayments_WithoutInstance_Success() {
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   fmt.Sprintf("%s-%s", models.CAPABILITY_FETCH_PAYMENTS.String(), "1"),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{
 				s.pspPayment,
 			},
 			NewState: []byte(`{}`),
 			HasMore:  false,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, payments []models.Payment) error {
 		s.Equal(1, len(payments))
 		s.Equal(s.paymentPayoutID, payments[0].ID)
 		return nil
 	})
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID: s.connectorID,
 		FromPayload: &FromPayload{
 			ID:      "1",
 			Payload: []byte(`{}`),
 		},
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err := s.env.GetWorkflowError()
 	s.NoError(err)
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_WithNextTasks_Success() {
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   fmt.Sprintf("%s-%s", models.CAPABILITY_FETCH_PAYMENTS.String(), "1"),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{
 				s.pspPayment,
 			},
 			NewState: []byte(`{}`),
 			HasMore:  false,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, payments []models.Payment) error {
 		s.Equal(1, len(payments))
 		s.Equal(s.paymentPayoutID, payments[0].ID)
 		return nil
 	})
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageSchedulesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.TemporalScheduleCreateActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID: s.connectorID,
 		FromPayload: &FromPayload{
 			ID:      "1",
 			Payload: []byte(`{}`),
 		},
 		Periodically: false,
 	}, []models.ConnectorTaskTree{
 		{
 			TaskType:     models.TASK_FETCH_PAYMENTS,
 			Name:         "test",
 			Periodically: true,
 		},
 	})
 	s.True(s.env.IsWorkflowCompleted())
 	err := s.env.GetWorkflowError()
 	s.NoError(err)
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_WithNextTasks_ConnectorScheduledForDeletion_Success() {
 	s.configuredConnector.ScheduledForDeletion = true
 	_, _, err := s.w.connectors.Load(s.configuredConnector, true, false)
 	s.NoError(err)
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   fmt.Sprintf("%s-%s", models.CAPABILITY_FETCH_PAYMENTS.String(), "1"),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{
 				s.pspPayment,
 			},
 			NewState: []byte(`{}`),
 			HasMore:  false,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, payments []models.Payment) error {
 		s.Equal(1, len(payments))
 		s.Equal(s.paymentPayoutID, payments[0].ID)
 		return nil
 	})
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID: s.connectorID,
 		FromPayload: &FromPayload{
 			ID:      "1",
 			Payload: []byte(`{}`),
 		},
 		Periodically: false,
 	}, []models.ConnectorTaskTree{
 		{
 			TaskType:     models.TASK_FETCH_PAYMENTS,
 			Name:         "test",
 			Periodically: true,
 		},
 	})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.NoError(err)
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_Success() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.Equal("test", instance.ScheduleID)
 		s.Equal(s.connectorID, instance.ConnectorID)
 		s.False(instance.Terminated)
 		return nil
 	})
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{
 				s.pspPayment,
 			},
 			NewState: []byte(`{}`),
 			HasMore:  false,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, payments []models.Payment) error {
 		s.Equal(1, len(payments))
 		s.Equal(s.paymentPayoutID, payments[0].ID)
 		return nil
 	})
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.Equal("test", instance.ScheduleID)
 		s.Equal(s.connectorID, instance.ConnectorID)
 		s.True(instance.Terminated)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.NoError(err)
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_WithoutNextTasks_Success() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.Equal("test", instance.ScheduleID)
 		s.Equal(s.connectorID, instance.ConnectorID)
 		s.False(instance.Terminated)
 		return nil
 	})
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{
 				s.pspPayment,
 			},
 			NewState: []byte(`{}`),
 			HasMore:  false,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, payments []models.Payment) error {
 		s.Equal(1, len(payments))
 		s.Equal(s.paymentPayoutID, payments[0].ID)
 		return nil
 	})
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.Equal("test", instance.ScheduleID)
 		s.Equal(s.connectorID, instance.ConnectorID)
 		s.True(instance.Terminated)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.NoError(err)
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_HasMoreLoop_Success() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.Equal("test", instance.ScheduleID)
 		s.Equal(s.connectorID, instance.ConnectorID)
 		s.False(instance.Terminated)
 		return nil
 	})
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{
 				s.pspPayment,
 			},
 			NewState: []byte(`{}`),
 			HasMore:  true,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, payments []models.Payment) error {
 		s.Equal(1, len(payments))
 		s.Equal(s.paymentPayoutID, payments[0].ID)
 		return nil
 	})
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, req activities.FetchNextPaymentsRequest) (*models.FetchNextPaymentsResponse, error) {
 		return &models.FetchNextPaymentsResponse{
 			Payments: []models.PSPPayment{},
 			NewState: []byte(`{}`),
 			HasMore:  false,
 		}, nil
 	})
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.Equal("test", instance.ScheduleID)
 		s.Equal(s.connectorID, instance.ConnectorID)
 		s.True(instance.Terminated)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.NoError(err)
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_StorageInstancesStore_Error() {
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(
 		temporal.NewNonRetryableApplicationError("error-test", "STORAGE", expectedErr),
 	)
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_StorageStatesGet_Error() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		nil,
 		temporal.NewNonRetryableApplicationError("error-test", "STORAGE", expectedErr),
 	)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.True(instance.Terminated)
 		s.NotNil(instance.Error)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_PluginFetchNextPayments_Error() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(
 		nil,
 		temporal.NewNonRetryableApplicationError("error-test", "PLUGIN", expectedErr),
 	)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.True(instance.Terminated)
 		s.NotNil(instance.Error)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_StoragePaymentsStore_Error() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(&models.FetchNextPaymentsResponse{
 		Payments: []models.PSPPayment{
 			s.pspPayment,
 		},
 		NewState: []byte(`{}`),
 		HasMore:  false,
 	}, nil)
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(
 		temporal.NewNonRetryableApplicationError("error-test", "STORAGE", expectedErr),
 	)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.True(instance.Terminated)
 		s.NotNil(instance.Error)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_StoragePaymentInitiationUpdateFromPaymentActivity_Error() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(&models.FetchNextPaymentsResponse{
 		Payments: []models.PSPPayment{
 			s.pspPayment,
 		},
 		NewState: []byte(`{}`),
 		HasMore:  false,
 	}, nil)
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(
 		temporal.NewNonRetryableApplicationError("error-test", "WORKFLOW", expectedErr),
 	)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.True(instance.Terminated)
 		s.NotNil(instance.Error)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_StorageStatesStore_Error() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(&models.FetchNextPaymentsResponse{
 		Payments: []models.PSPPayment{
 			s.pspPayment,
 		},
 		NewState: []byte(`{}`),
 		HasMore:  false,
 	}, nil)
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(
 		temporal.NewNonRetryableApplicationError("error-test", "STORAGE", expectedErr),
 	)
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.True(instance.Terminated)
 		s.NotNil(instance.Error)
 		return nil
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }
 func (s *UnitTestSuite) Test_FetchNextPayments_StorageInstancesUpdate_Error() {
 	s.env.OnActivity(activities.StorageInstancesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesGetActivity, mock.Anything, mock.Anything).Once().Return(
 		&models.State{
 			ID: models.StateID{
 				Reference:   models.CAPABILITY_FETCH_PAYMENTS.String(),
 				ConnectorID: s.connectorID,
 			},
 			ConnectorID: s.connectorID,
 			State:       []byte(`{}`),
 		},
 		nil,
 	)
 	s.env.OnActivity(activities.PluginFetchNextPaymentsActivity, mock.Anything, mock.Anything).Once().Return(&models.FetchNextPaymentsResponse{
 		Payments: []models.PSPPayment{
 			s.pspPayment,
 		},
 		NewState: []byte(`{}`),
 		HasMore:  false,
 	}, nil)
 	s.env.OnActivity(activities.StoragePaymentsStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	s.env.OnActivity(activities.StoragePaymentInitiationUpdateFromPaymentActivity, mock.Anything, s.pspPayment.Status, s.pspPayment.CreatedAt, s.paymentPayoutID).Once().Return(nil)
 	s.env.OnActivity(activities.StorageStatesStoreActivity, mock.Anything, mock.Anything).Once().Return(nil)
 	expectedErr := errors.New("error-test")
 	s.env.OnActivity(activities.StorageInstancesUpdateActivity, mock.Anything, mock.Anything).Once().Return(func(ctx context.Context, instance models.Instance) error {
 		s.True(instance.Terminated)
 		s.Nil(instance.Error)
 		return temporal.NewNonRetryableApplicationError("error-test", "STORAGE", expectedErr)
 	})
 	err := s.env.SetTypedSearchAttributesOnStart(temporal.NewSearchAttributes(temporal.NewSearchAttributeKeyKeyword(SearchAttributeScheduleID).ValueSet("test")))
 	s.NoError(err)
 	s.env.ExecuteWorkflow(RunFetchNextPayments, FetchNextPayments{
 		ConnectorID:  s.connectorID,
 		FromPayload:  nil,
 		Periodically: false,
 	}, []models.ConnectorTaskTree{})
 	s.True(s.env.IsWorkflowCompleted())
 	err = s.env.GetWorkflowError()
 	s.Error(err)
 	s.ErrorContains(err, expectedErr.Error())
 }

595 lines 24 KiB Go Raw Permalink Blame History

595 lines

24 KiB

Go

Raw Permalink Blame History