routing: refactor attempt makers to return pointers
Thus adding following unit tests can be a bit easier.
This commit is contained in:
2 files changed
+58
-20
No files matched your search
@@ -3483,7 +3483,7 @@ func TestSendToRouteSkipTempErrSuccess(t *testing.T) {
|
||||
)
|
||||
|
||||
preimage := lntypes.Preimage{1}
|
||||
testAttempt := makeSettledAttempt(int(payAmt), 0, preimage)
|
||||
testAttempt := makeSettledAttempt(t, int(payAmt), preimage)
|
||||
|
||||
node, err := createTestNode()
|
||||
require.NoError(t, err)
|
||||
@@ -3521,7 +3521,7 @@ func TestSendToRouteSkipTempErrSuccess(t *testing.T) {
|
||||
controlTower.On("RegisterAttempt", payHash, mock.Anything).Return(nil)
|
||||
controlTower.On("SettleAttempt",
|
||||
payHash, mock.Anything, mock.Anything,
|
||||
).Return(&testAttempt, nil)
|
||||
).Return(testAttempt, nil)
|
||||
|
||||
payer.On("SendHTLC",
|
||||
mock.Anything, mock.Anything, mock.Anything,
|
||||
@@ -3550,7 +3550,7 @@ func TestSendToRouteSkipTempErrSuccess(t *testing.T) {
|
||||
// Expect a successful send to route.
|
||||
attempt, err := router.SendToRouteSkipTempErr(payHash, rt)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, &testAttempt, attempt)
|
||||
require.Equal(t, testAttempt, attempt)
|
||||
|
||||
// Assert the above methods are called as expected.
|
||||
controlTower.AssertExpectations(t)
|
||||
@@ -3563,11 +3563,11 @@ func TestSendToRouteSkipTempErrSuccess(t *testing.T) {
|
||||
// cause the payment to be failed.
|
||||
func TestSendToRouteSkipTempErrTempFailure(t *testing.T) {
|
||||
var (
|
||||
payHash lntypes.Hash
|
||||
payAmt = lnwire.MilliSatoshi(10000)
|
||||
testAttempt = &channeldb.HTLCAttempt{}
|
||||
payHash lntypes.Hash
|
||||
payAmt = lnwire.MilliSatoshi(10000)
|
||||
)
|
||||
|
||||
testAttempt := makeFailedAttempt(t, int(payAmt))
|
||||
node, err := createTestNode()
|
||||
require.NoError(t, err)
|
||||
|
||||
@@ -3648,7 +3648,7 @@ func TestSendToRouteSkipTempErrPermanentFailure(t *testing.T) {
|
||||
payAmt = lnwire.MilliSatoshi(10000)
|
||||
)
|
||||
|
||||
testAttempt := makeFailedAttempt(int(payAmt), 0)
|
||||
testAttempt := makeFailedAttempt(t, int(payAmt))
|
||||
node, err := createTestNode()
|
||||
require.NoError(t, err)
|
||||
|
||||
@@ -3733,7 +3733,7 @@ func TestSendToRouteTempFailure(t *testing.T) {
|
||||
payAmt = lnwire.MilliSatoshi(10000)
|
||||
)
|
||||
|
||||
testAttempt := makeFailedAttempt(int(payAmt), 0)
|
||||
testAttempt := makeFailedAttempt(t, int(payAmt))
|
||||
node, err := createTestNode()
|
||||
require.NoError(t, err)
|
||||
|
||||
|
||||
Reference in new issue
Block a user