synapsecns/sanguine

package client_test

import (
    "github.com/synapsecns/sanguine/services/rfq/api/model"
)

// TODO: @aurelius tese tests make a lot less sesnes w/ a composite index

func (c *ClientSuite) TestPutAndGetQuote() {
    req := model.PutRelayerQuoteRequest{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
        DestAmount:      "100",
        MaxOriginAmount: "200",
        FixedFee:        "10",
    }

    err := c.client.PutQuote(c.GetTestContext(), &req)
    c.Require().NoError(err)

    quotes, err := c.client.GetAllQuotes(c.GetTestContext())
    c.Require().NoError(err)

    expectedResp := model.GetQuoteResponse{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
        DestAmount:      "100",
        MaxOriginAmount: "200",
        FixedFee:        "10",
        RelayerAddr:     c.testWallet.Address().String(),
        UpdatedAt:       quotes[0].UpdatedAt,
    }
    c.Len(quotes, 1)
    c.Equal(expectedResp, *quotes[0])
}

func (c *ClientSuite) TestPutAndGetBulkQuotes() {
    req := model.PutBulkQuotesRequest{
        Quotes: []model.PutRelayerQuoteRequest{
            {
                OriginChainID:   1,
                OriginTokenAddr: "0xOriginTokenAddr",
                DestChainID:     42161,
                DestTokenAddr:   "0xDestTokenAddr",
                DestAmount:      "100",
                MaxOriginAmount: "200",
                FixedFee:        "10",
            },
            {
                OriginChainID:   42161,
                OriginTokenAddr: "0xOriginTokenAddr",
                DestChainID:     1,
                DestTokenAddr:   "0xDestTokenAddr",
                DestAmount:      "100",
                MaxOriginAmount: "200",
                FixedFee:        "10",
            },
        },
    }

    err := c.client.PutBulkQuotes(c.GetTestContext(), &req)
    c.Require().NoError(err)

    quotes, err := c.client.GetAllQuotes(c.GetTestContext())
    c.Require().NoError(err)

    expectedResp := []model.GetQuoteResponse{
        {
            OriginChainID:   1,
            OriginTokenAddr: "0xOriginTokenAddr",
            DestChainID:     42161,
            DestTokenAddr:   "0xDestTokenAddr",
            DestAmount:      "100",
            MaxOriginAmount: "200",
            FixedFee:        "10",
            RelayerAddr:     c.testWallet.Address().String(),
            UpdatedAt:       quotes[0].UpdatedAt,
        },
        {
            OriginChainID:   42161,
            OriginTokenAddr: "0xOriginTokenAddr",
            DestChainID:     1,
            DestTokenAddr:   "0xDestTokenAddr",
            DestAmount:      "100",
            MaxOriginAmount: "200",
            FixedFee:        "10",
            RelayerAddr:     c.testWallet.Address().String(),
            UpdatedAt:       quotes[0].UpdatedAt,
        },
    }
    c.Len(quotes, 2)
    c.Equal(expectedResp[0], *quotes[0])
    c.Equal(expectedResp[1], *quotes[1])
}

func (c *ClientSuite) TestGetSpecificQuote() {
    req := model.PutRelayerQuoteRequest{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
        DestAmount:      "100",
        MaxOriginAmount: "200",
        FixedFee:        "10",
    }

    err := c.client.PutQuote(c.GetTestContext(), &req)
    c.Require().NoError(err)

    quotes, err := c.client.GetSpecificQuote(c.GetTestContext(), &model.GetQuoteSpecificRequest{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
    })
    c.Require().NoError(err)

    c.Len(quotes, 1)
    expectedResp := model.GetQuoteResponse{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
        DestAmount:      "100",
        MaxOriginAmount: "200",
        FixedFee:        "10",
        RelayerAddr:     c.testWallet.Address().String(),
        UpdatedAt:       quotes[0].UpdatedAt,
    }
    c.Equal(expectedResp, *quotes[0])
}

func (c *ClientSuite) TestGetQuoteByRelayerAddress() {
    req := model.PutRelayerQuoteRequest{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
        DestAmount:      "100",
        MaxOriginAmount: "200",
        FixedFee:        "10",
    }

    err := c.client.PutQuote(c.GetTestContext(), &req)
    c.Require().NoError(err)

    relayerAddr := c.testWallet.Address().Hex()
    quotes, err := c.client.GetQuoteByRelayerAddress(c.GetTestContext(), relayerAddr)
    c.Require().NoError(err)

    c.Len(quotes, 1)
    expectedResp := model.GetQuoteResponse{
        OriginChainID:   1,
        OriginTokenAddr: "0xOriginTokenAddr",
        DestChainID:     42161,
        DestTokenAddr:   "0xDestTokenAddr",
        DestAmount:      "100",
        MaxOriginAmount: "200",
        FixedFee:        "10",
        RelayerAddr:     c.testWallet.Address().String(),
        UpdatedAt:       quotes[0].UpdatedAt,
    }
    c.Equal(expectedResp, *quotes[0])
}