synapsecns/sanguine

package db_test

import (
    "github.com/brianvoe/gofakeit/v6"
    "github.com/synapsecns/sanguine/contrib/screener-api/db"
    "gorm.io/gorm"
)

func (d *DBSuite) TestBlacklist() {
    d.RunOnAllDBs(func(testDB db.DB) {
        testAddress := gofakeit.BitcoinAddress()

        blacklistBody := db.BlacklistedAddress{
            Type:    "create",
            ID:      "testId",
            Address: testAddress,
            Network: "bitcoin",
            Tag:     "testTag",
            Remark:  "testRemark",
        }

        // blacklist the address
        err := testDB.PutBlacklistedAddress(d.GetTestContext(), blacklistBody)
        d.Require().NoError(err)
        blacklistedAddress, err := testDB.GetBlacklistedAddress(d.GetTestContext(), blacklistBody.Address)
        d.Require().NoError(err)
        d.Require().NotNil(blacklistedAddress)

        // update the address
        blacklistBody.Type = "update"
        blacklistBody.Remark = "testRemarkUpdated"
        err = testDB.UpdateBlacklistedAddress(d.GetTestContext(), blacklistBody.ID, blacklistBody)
        d.Require().NoError(err)

        // check to make sure it updated
        blacklistedAddress, err = testDB.GetBlacklistedAddress(d.GetTestContext(), blacklistBody.Address)
        d.Require().NoError(err)
        d.Require().NotNil(blacklistedAddress)
        d.Require().Equal("testRemarkUpdated", blacklistedAddress.Remark)

        // check for non blacklisted address
        res, err := testDB.GetBlacklistedAddress(d.GetTestContext(), gofakeit.BitcoinAddress())
        d.Require().EqualError(err, gorm.ErrRecordNotFound.Error())
        d.Require().Nil(res)

        // delete it
        err = testDB.DeleteBlacklistedAddress(d.GetTestContext(), blacklistBody.ID)
        d.Require().NoError(err)

        // delete nonexistent
        err = testDB.DeleteBlacklistedAddress(d.GetTestContext(), "NonexistentId")
        d.Require().NoError(err)
    })
}