synapsecns/sanguine

package db_test

import (
    "github.com/shopspring/decimal"
    "github.com/synapsecns/sanguine/services/rfq/api/db"
)

func (d *DBSuite) TestGetQuotesByDestChainAndToken() {
    d.RunOnAllDBs(func(testDB db.APIDB) {
        // Arrange: Create and insert a quote
        expectedQuote := &db.Quote{
            // Initialize fields like ID, DestChainID, DestTokenAddr, etc.
            OriginChainID:   1,
            OriginTokenAddr: "0x3f5CE5FBFe3E9af3971dD833D26bA9b5C936f0bE",
            DestChainID:     42161,
            DestTokenAddr:   "0x3f5CE5FBFe3E9af3971dD833D26bA9b5C936f0bE",
            DestAmount:      decimal.NewFromInt(1000),
            MaxOriginAmount: decimal.NewFromInt(1000).Div(decimal.NewFromFloat(0.01)),
            FixedFee:        decimal.NewFromFloat(1),
        }
        err := testDB.UpsertQuote(d.GetTestContext(), expectedQuote)
        d.Require().NoError(err)

        // Act: Retrieve quotes by DestChainID and DestTokenAddr
        quotes, err := testDB.GetQuotesByDestChainAndToken(d.GetTestContext(), expectedQuote.DestChainID, expectedQuote.DestTokenAddr)
        d.Require().NoError(err)

        // Assert: Check if the retrieved quotes match the inserted quote
        d.Len(quotes, 1)
        d.Equal(expectedQuote.OriginChainID, quotes[0].OriginChainID)
        d.Equal(expectedQuote.OriginTokenAddr, quotes[0].OriginTokenAddr)
        d.Equal(expectedQuote.DestChainID, quotes[0].DestChainID)
        d.Equal(expectedQuote.DestTokenAddr, quotes[0].DestTokenAddr)
        // TODO: decimal comparison
        // d.Equal(expectedQuote.DestAmount, quotes[0].DestAmount)
        // d.Equal(expectedQuote.Price, quotes[0].Price)
        // d.Equal(expectedQuote.MaxOriginAmount, quotes[0].MaxOriginAmount)
        d.NotEqual(quotes[0].UpdatedAt, nil)
        // Continue asserting other fields
    })
}

func (d *DBSuite) TestUpsertQuote() {
    d.RunOnAllDBs(func(testDB db.APIDB) {
        // Arrange: Create a quote
        quote := &db.Quote{
            // Initialize fields like ID, DestChainID, DestTokenAddr, etc.
            OriginChainID:   1,
            OriginTokenAddr: "0x3f5CE5FBFe3E9af3971dD833D26bA9b5C936f0bE",
            DestChainID:     42161,
            DestTokenAddr:   "0x3f5CE5FBFe3E9af3971dD833D26bA9b5C936f0bE",
            DestAmount:      decimal.NewFromInt(1000),
            MaxOriginAmount: decimal.NewFromInt(1000).Div(decimal.NewFromFloat(0.01)),
            FixedFee:        decimal.NewFromFloat(1),
        }

        // Act & Assert: Insert new quote
        err := testDB.UpsertQuote(d.GetTestContext(), quote)
        d.Require().NoError(err)

        // Retrieve to verify insertion
        retrievedQuotes, err := testDB.GetQuotesByDestChainAndToken(d.GetTestContext(), quote.DestChainID, quote.DestTokenAddr)
        d.Require().NoError(err)
        d.Len(retrievedQuotes, 1)
        // Assert other fields if necessary

        // Act & Assert: Update the existing quote
        quote.FixedFee = decimal.NewFromFloat(2)
        err = testDB.UpsertQuote(d.GetTestContext(), quote)
        d.Require().NoError(err)

        // Retrieve to verify update
        updatedQuotes, err := testDB.GetQuotesByDestChainAndToken(d.GetTestContext(), quote.DestChainID, quote.DestTokenAddr)
        d.Require().NoError(err)
        d.Len(updatedQuotes, 1)
        d.Equal(quote.FixedFee, updatedQuotes[0].FixedFee)
        // Assert other fields if necessary
    })
}