multi: address linter issues

This commit is contained in:
Elle Mouton
2023-11-08 11:40:20 +02:00
parent 84cdcd6847
commit 0e82293e45
8 changed files with 82 additions and 56 deletions

View File

@@ -16,9 +16,14 @@ import (
)
type hopHintsConfigMock struct {
t *testing.T
mock.Mock
}
func newHopHintsConfigMock(t *testing.T) *hopHintsConfigMock {
return &hopHintsConfigMock{t: t}
}
// IsPublicNode mocks node public state lookup.
func (h *hopHintsConfigMock) IsPublicNode(pubKey [33]byte) (bool, error) {
args := h.Mock.Called(pubKey)
@@ -65,9 +70,14 @@ func (h *hopHintsConfigMock) FetchChannelEdgesByID(chanID uint64) (
return nil, nil, nil, err
}
edgeInfo := args.Get(0).(*models.ChannelEdgeInfo)
policy1 := args.Get(1).(*models.ChannelEdgePolicy)
policy2 := args.Get(2).(*models.ChannelEdgePolicy)
edgeInfo, ok := args.Get(0).(*models.ChannelEdgeInfo)
require.True(h.t, ok)
policy1, ok := args.Get(1).(*models.ChannelEdgePolicy)
require.True(h.t, ok)
policy2, ok := args.Get(2).(*models.ChannelEdgePolicy)
require.True(h.t, ok)
return edgeInfo, policy1, policy2, err
}
@@ -429,7 +439,7 @@ func TestShouldIncludeChannel(t *testing.T) {
t.Parallel()
// Create mock and prime it for the test case.
mock := &hopHintsConfigMock{}
mock := newHopHintsConfigMock(t)
if tc.setupMock != nil {
tc.setupMock(mock)
}
@@ -862,7 +872,7 @@ func TestPopulateHopHints(t *testing.T) {
t.Parallel()
// Create mock and prime it for the test case.
mock := &hopHintsConfigMock{}
mock := newHopHintsConfigMock(t)
if tc.setupMock != nil {
tc.setupMock(mock)
}