We add this constructor for an AuxHtlcDescriptor that allows setting some of the internal fields. This is useful for testing purposes for code external to this package that may need to extensively test the AuxHtlcView.
32 lines
1.1 KiB
Go
32 lines
1.1 KiB
Go
package lnwallet
|
|
|
|
import (
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
)
|
|
|
|
// NewTestAuxHtlcDescriptor creates an AuxHtlcDescriptor for testing purposes.
|
|
// This function allows tests to create descriptors with specific commit heights
|
|
// and entry types, which are normally unexported fields.
|
|
func NewTestAuxHtlcDescriptor(chanID lnwire.ChannelID,
|
|
rHash PaymentHash, timeout uint32,
|
|
amount lnwire.MilliSatoshi, htlcIndex, parentIndex uint64,
|
|
entryType uint8, customRecords lnwire.CustomRecords,
|
|
addHeightLocal, addHeightRemote, removeHeightLocal,
|
|
removeHeightRemote uint64) AuxHtlcDescriptor {
|
|
|
|
return AuxHtlcDescriptor{
|
|
ChanID: chanID,
|
|
RHash: rHash,
|
|
Timeout: timeout,
|
|
Amount: amount,
|
|
HtlcIndex: htlcIndex,
|
|
ParentIndex: parentIndex,
|
|
EntryType: updateType(entryType),
|
|
CustomRecords: customRecords,
|
|
addCommitHeightLocal: addHeightLocal,
|
|
addCommitHeightRemote: addHeightRemote,
|
|
removeCommitHeightLocal: removeHeightLocal,
|
|
removeCommitHeightRemote: removeHeightRemote,
|
|
}
|
|
}
|