You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.4 KiB
55 lines
1.4 KiB
package sweep |
|
|
|
import ( |
|
"github.com/btcsuite/btcd/chaincfg/chainhash" |
|
"github.com/btcsuite/btcd/wire" |
|
) |
|
|
|
// MockSweeperStore is a mock implementation of sweeper store. This type is |
|
// exported, because it is currently used in nursery tests too. |
|
type MockSweeperStore struct { |
|
lastTx *wire.MsgTx |
|
ourTxes map[chainhash.Hash]struct{} |
|
} |
|
|
|
// NewMockSweeperStore returns a new instance. |
|
func NewMockSweeperStore() *MockSweeperStore { |
|
return &MockSweeperStore{ |
|
ourTxes: make(map[chainhash.Hash]struct{}), |
|
} |
|
} |
|
|
|
// IsOurTx determines whether a tx is published by us, based on its |
|
// hash. |
|
func (s *MockSweeperStore) IsOurTx(hash chainhash.Hash) (bool, error) { |
|
_, ok := s.ourTxes[hash] |
|
return ok, nil |
|
} |
|
|
|
// NotifyPublishTx signals that we are about to publish a tx. |
|
func (s *MockSweeperStore) NotifyPublishTx(tx *wire.MsgTx) error { |
|
txHash := tx.TxHash() |
|
s.ourTxes[txHash] = struct{}{} |
|
s.lastTx = tx |
|
|
|
return nil |
|
} |
|
|
|
// GetLastPublishedTx returns the last tx that we called NotifyPublishTx |
|
// for. |
|
func (s *MockSweeperStore) GetLastPublishedTx() (*wire.MsgTx, error) { |
|
return s.lastTx, nil |
|
} |
|
|
|
// ListSweeps lists all the sweeps we have successfully published. |
|
func (s *MockSweeperStore) ListSweeps() ([]chainhash.Hash, error) { |
|
var txns []chainhash.Hash |
|
for tx := range s.ourTxes { |
|
txns = append(txns, tx) |
|
} |
|
|
|
return txns, nil |
|
} |
|
|
|
// Compile-time constraint to ensure MockSweeperStore implements SweeperStore. |
|
var _ SweeperStore = (*MockSweeperStore)(nil)
|
|
|