sweep: add sweeper test

sweep/fee_estimator_mock_test.go

@@ -0,0 +1,62 @@
+package sweep
+
+import (
+	"github.com/lightningnetwork/lnd/lnwallet"
+	"sync"
+)
+
+// mockFeeEstimator implements a mock fee estimator. It closely resembles
+// lnwallet.StaticFeeEstimator with the addition that fees can be changed for
+// testing purposes in a thread safe manner.
+type mockFeeEstimator struct {
+	feePerKW lnwallet.SatPerKWeight
+
+	relayFee lnwallet.SatPerKWeight
+
+	lock sync.Mutex
+}
+
+func newMockFeeEstimator(feePerKW,
+	relayFee lnwallet.SatPerKWeight) *mockFeeEstimator {
+
+	return &mockFeeEstimator{
+		feePerKW: feePerKW,
+		relayFee: relayFee,
+	}
+}
+
+func (e *mockFeeEstimator) updateFees(feePerKW,
+	relayFee lnwallet.SatPerKWeight) {
+
+	e.lock.Lock()
+	defer e.lock.Unlock()
+
+	e.feePerKW = feePerKW
+	e.relayFee = relayFee
+}
+
+func (e *mockFeeEstimator) EstimateFeePerKW(numBlocks uint32) (
+	lnwallet.SatPerKWeight, error) {
+
+	e.lock.Lock()
+	defer e.lock.Unlock()
+
+	return e.feePerKW, nil
+}
+
+func (e *mockFeeEstimator) RelayFeePerKW() lnwallet.SatPerKWeight {
+	e.lock.Lock()
+	defer e.lock.Unlock()
+
+	return e.relayFee
+}
+
+func (e *mockFeeEstimator) Start() error {
+	return nil
+}
+
+func (e *mockFeeEstimator) Stop() error {
+	return nil
+}
+
+var _ lnwallet.FeeEstimator = (*mockFeeEstimator)(nil)