Skip to content

Commit

Permalink
chore: add populate wallet alias test
Browse files Browse the repository at this point in the history
  • Loading branch information
freak12techno committed Jul 15, 2024
1 parent 26fe6b1 commit 57c73af
Showing 1 changed file with 61 additions and 0 deletions.
61 changes: 61 additions & 0 deletions pkg/data_fetcher/populate_wallet_alias_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
package data_fetcher

import (
aliasManagerPkg "main/pkg/alias_manager"
configPkg "main/pkg/config"
"main/pkg/config/types"
"main/pkg/fs"
loggerPkg "main/pkg/logger"
"main/pkg/metrics"
"testing"

"github.com/stretchr/testify/require"
)

func TestDataFetcherPopulateWalletAliasNotPresent(t *testing.T) {
t.Parallel()

config := &configPkg.AppConfig{
Chains: types.Chains{
{Name: "chain"},
},
Metrics: configPkg.MetricsConfig{Enabled: false},
}

filesystem := &fs.MockFs{}
logger := loggerPkg.GetNopLogger()
aliasManager := aliasManagerPkg.NewAliasManager(logger, config, filesystem)
metricsManager := metrics.NewManager(logger, config.Metrics)
dataFetcher := NewDataFetcher(logger, config, aliasManager, metricsManager)

wallet := &types.Link{Value: "address"}

dataFetcher.PopulateWalletAlias(config.Chains[0], wallet, "subscription")
require.Empty(t, wallet.Title)
}

func TestDataFetcherPopulateWalletAliasPresent(t *testing.T) {
t.Parallel()

config := &configPkg.AppConfig{
Chains: types.Chains{
{Name: "chain"},
},
Metrics: configPkg.MetricsConfig{Enabled: false},
AliasesPath: "path.toml",
}

filesystem := &fs.MockFs{}
logger := loggerPkg.GetNopLogger()
aliasManager := aliasManagerPkg.NewAliasManager(logger, config, filesystem)
metricsManager := metrics.NewManager(logger, config.Metrics)
dataFetcher := NewDataFetcher(logger, config, aliasManager, metricsManager)

err := aliasManager.Set("subscription", "chain", "address", "alias")
require.NoError(t, err)

wallet := &types.Link{Value: "address"}

dataFetcher.PopulateWalletAlias(config.Chains[0], wallet, "subscription")
require.Equal(t, "alias", wallet.Title)
}

0 comments on commit 57c73af

Please sign in to comment.