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

channeldb: export now function

Browse Source 
To allow mocking when testing other packages.
This commit is contained in:
Joost Jager 2019-11-15 12:18:06 +01:00
parent 1371e5affc
commit e234f88b82
No known key found for this signature in database
GPG Key ID: A61B9D4C393C59C7
3 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
channeldb/db.go
View File

@ -137,7 +137,8 @@ type DB struct {
*bbolt.DB *bbolt.DB
dbPath string dbPath string
graph *ChannelGraph graph *ChannelGraph
now func() time.Time
Now func() time.Time
} }
// Open opens an existing channeldb. Any necessary schemas migrations due to // Open opens an existing channeldb. Any necessary schemas migrations due to
@ -171,7 +172,7 @@ func Open(dbPath string, modifiers ...OptionModifier) (*DB, error) {
chanDB := &DB{ chanDB := &DB{
DB: bdb, DB: bdb,
dbPath: dbPath, dbPath: dbPath,
now: time.Now, Now: time.Now,
} }
chanDB.graph = newChannelGraph( chanDB.graph = newChannelGraph(
chanDB, opts.RejectCacheSize, opts.ChannelCacheSize, chanDB, opts.RejectCacheSize, opts.ChannelCacheSize,

2
channeldb/invoice_test.go
View File

@ -337,7 +337,7 @@ func TestDuplicateSettleInvoice(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("unable to make test db: %v", err) t.Fatalf("unable to make test db: %v", err)
} }
db.now = func() time.Time { return time.Unix(1, 0) } db.Now = func() time.Time { return time.Unix(1, 0) }
// We'll start out by creating an invoice and writing it to the DB. // We'll start out by creating an invoice and writing it to the DB.
amt := lnwire.NewMSatFromSatoshis(1000) amt := lnwire.NewMSatFromSatoshis(1000)

2
channeldb/invoices.go
View File

@ -1244,7 +1244,7 @@ func (d *DB) updateInvoice(hash lntypes.Hash, invoices, settleIndex *bbolt.Bucke
// Update invoice state. // Update invoice state.
invoice.State = update.State invoice.State = update.State
now := d.now() now := d.Now()
// Update htlc set. // Update htlc set.
for key, htlcUpdate := range update.Htlcs { for key, htlcUpdate := range update.Htlcs {