akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

lnwallet/btcwallet: add compile time check for BlockChainIO interface

Browse Source
This commit is contained in:
Johan T. Halseth 2018-08-24 15:30:23 +02:00
parent c4415f0400
commit beb5d14ed9
No known key found for this signature in database
GPG Key ID: 15BAADA29DA20D26
4 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
contractcourt/channel_arbitrator_test.go
View File

@ -124,6 +124,8 @@ func (b *mockArbitratorLog) WipeHistory() error {
type mockChainIO struct{} type mockChainIO struct{}
var _ lnwallet.BlockChainIO = (*mockChainIO)(nil)
func (*mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) { func (*mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) {
return nil, 0, nil return nil, 0, nil
} }

3
lnwallet/btcwallet/btcwallet.go
View File

@ -67,8 +67,9 @@ type BtcWallet struct {
} }
// A compile time check to ensure that BtcWallet implements the // A compile time check to ensure that BtcWallet implements the
// WalletController interface. // WalletController and BlockChainIO interfaces.
var _ lnwallet.WalletController = (*BtcWallet)(nil) var _ lnwallet.WalletController = (*BtcWallet)(nil)
var _ lnwallet.BlockChainIO = (*BtcWallet)(nil)
// New returns a new fully initialized instance of BtcWallet given a valid // New returns a new fully initialized instance of BtcWallet given a valid
// configuration struct. // configuration struct.

2
mock.go
View File

@ -205,6 +205,8 @@ type mockChainIO struct {
bestHeight int32 bestHeight int32
} }
var _ lnwallet.BlockChainIO = (*mockChainIO)(nil)
func (m *mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) { func (m *mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) {
return activeNetParams.GenesisHash, m.bestHeight, nil return activeNetParams.GenesisHash, m.bestHeight, nil
} }

3
sweep/test_utils.go
View File

@ -10,6 +10,7 @@ import (
"github.com/btcsuite/btcd/wire" "github.com/btcsuite/btcd/wire"
"github.com/lightningnetwork/lnd/chainntnfs" "github.com/lightningnetwork/lnd/chainntnfs"
"github.com/lightningnetwork/lnd/input" "github.com/lightningnetwork/lnd/input"
"github.com/lightningnetwork/lnd/lnwallet"
) )
var ( var (
@ -237,6 +238,8 @@ func (m *MockNotifier) RegisterSpendNtfn(outpoint *wire.OutPoint,
type mockChainIO struct{} type mockChainIO struct{}
var _ lnwallet.BlockChainIO = (*mockChainIO)(nil)
func (m *mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) { func (m *mockChainIO) GetBestBlock() (*chainhash.Hash, int32, error) {
return nil, mockChainIOHeight, nil return nil, mockChainIOHeight, nil
} }