mirror of
https://github.com/hanzo-fi/payments.git
synced 2026-10-30 20:31:20 +00:00
CrimsonandGitHub
4ae387a5d1
* Use formance-sdk-go/v3 to in connector uninstall e2e test * Compile API docs from openapi spec * Generate go SDK locally for use in integration tests * Get rid of unnecessary redirect in e2e tests now that SDK is generated * Omit generated SDK code from codecoverage checks * Add flake and envrc for nix * Run Dirty and Tests CI steps with nix * Run pre-commit using nix * Remove duplicated steps from Earthfile * Fixes for linter * Add go generate to Justfile pre-commit * Still run Deploy on ubuntu-latest * Address coderabbit feedback
103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
package testserver
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/formancehq/payments/internal/connectors/httpwrapper"
|
|
"github.com/formancehq/payments/internal/models"
|
|
formance "github.com/formancehq/payments/pkg/client"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func NewStackClient(urlStr string, clientTimeout time.Duration, transport http.RoundTripper) (*formance.Formance, error) {
|
|
httpClient := &http.Client{
|
|
Timeout: clientTimeout,
|
|
Transport: transport,
|
|
}
|
|
|
|
return formance.New(
|
|
urlStr,
|
|
formance.WithClient(httpClient),
|
|
), nil
|
|
}
|
|
|
|
type Client struct {
|
|
baseUrl string
|
|
internalClient httpwrapper.Client
|
|
transport http.RoundTripper
|
|
}
|
|
|
|
func NewClient(urlStr string, clientTimeout time.Duration, transport http.RoundTripper) (*Client, error) {
|
|
config := &httpwrapper.Config{Timeout: clientTimeout}
|
|
internalClient := httpwrapper.NewClient(config)
|
|
return &Client{
|
|
baseUrl: urlStr,
|
|
transport: transport,
|
|
internalClient: internalClient,
|
|
}, nil
|
|
}
|
|
|
|
func (c *Client) wrapError(err error, method, path string, status int, errBody map[string]interface{}) error {
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("error with status %d for %s to '%s': %w, body: %+v", status, method, path, err, errBody)
|
|
}
|
|
|
|
func (c *Client) Get(ctx context.Context, path string, resBody any) error {
|
|
method := http.MethodGet
|
|
req, err := http.NewRequestWithContext(ctx, method, c.baseUrl+path, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var errBody map[string]interface{}
|
|
status, err := c.internalClient.Do(ctx, req, resBody, &errBody)
|
|
return c.wrapError(err, method, path, status, errBody)
|
|
}
|
|
|
|
func (c *Client) Do(ctx context.Context, method string, path string, body any, resBody any) error {
|
|
b, err := json.Marshal(body)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
req, err := http.NewRequestWithContext(ctx, method, c.baseUrl+path, bytes.NewReader(b))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var errBody map[string]interface{}
|
|
status, err := c.internalClient.Do(ctx, req, resBody, &errBody)
|
|
return c.wrapError(err, method, path, status, errBody)
|
|
}
|
|
|
|
func (c *Client) PollTask(ctx context.Context, t T) func(id string) func() models.Task {
|
|
return func(id string) func() models.Task {
|
|
return func() models.Task {
|
|
path := "/v3/tasks/" + id
|
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, c.baseUrl+path, nil)
|
|
require.NoError(t, err)
|
|
|
|
httpClient := &http.Client{Timeout: defaultHttpClientTimeout}
|
|
res, err := httpClient.Do(req)
|
|
require.NoError(t, err)
|
|
defer res.Body.Close()
|
|
require.Equal(t, http.StatusOK, res.StatusCode)
|
|
|
|
var expectedBody struct{ Data models.Task }
|
|
rawBody, err := io.ReadAll(res.Body)
|
|
require.NoError(t, err)
|
|
err = json.Unmarshal(rawBody, &expectedBody)
|
|
require.NoError(t, err)
|
|
return expectedBody.Data
|
|
}
|
|
}
|
|
}
|