From 08308c81eda04bdb4eeb833bc5b34101537db40d Mon Sep 17 00:00:00 2001
From: yyforyongyu <yy2452@columbia.edu>
Date: Wed, 12 Aug 2020 03:49:44 +0800
Subject: [PATCH] walletdb: update tests to use db timeout param

---
 walletdb/db_test.go      | 7 ++++++-
 walletdb/example_test.go | 6 +++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/walletdb/db_test.go b/walletdb/db_test.go
index 30c4341..9e8760a 100644
--- a/walletdb/db_test.go
+++ b/walletdb/db_test.go
@@ -10,6 +10,7 @@ import (
 	"os"
 	"path/filepath"
 	"testing"
+	"time"
 
 	"github.com/btcsuite/btcwallet/walletdb"
 	_ "github.com/btcsuite/btcwallet/walletdb/bdb"
@@ -21,6 +22,10 @@ var (
 	// bogus drivers for testing purposes while still allowing other tests
 	// to easily iterate all supported drivers.
 	ignoreDbTypes = map[string]bool{"createopenfail": true}
+
+	// defaultDBTimeout specifies the timeout value when opening the wallet
+	// database.
+	defaultDBTimeout = 10 * time.Second
 )
 
 // TestAddDuplicateDriver ensures that adding a duplicate driver does not
@@ -64,7 +69,7 @@ func TestAddDuplicateDriver(t *testing.T) {
 	defer os.Remove(tempDir)
 
 	dbPath := filepath.Join(tempDir, "db")
-	db, err := walletdb.Create(dbType, dbPath, true)
+	db, err := walletdb.Create(dbType, dbPath, true, defaultDBTimeout)
 	if err != nil {
 		t.Errorf("failed to create database: %v", err)
 		return
diff --git a/walletdb/example_test.go b/walletdb/example_test.go
index 4cb5134..c9640ab 100644
--- a/walletdb/example_test.go
+++ b/walletdb/example_test.go
@@ -28,7 +28,7 @@ func ExampleCreate() {
 	// this, but it's done here in the example to ensure the example cleans
 	// up after itself.
 	dbPath := filepath.Join(os.TempDir(), "examplecreate.db")
-	db, err := walletdb.Create("bdb", dbPath, true)
+	db, err := walletdb.Create("bdb", dbPath, true, defaultDBTimeout)
 	if err != nil {
 		fmt.Println(err)
 		return
@@ -47,7 +47,7 @@ var exampleNum = 0
 func exampleLoadDB() (walletdb.DB, func(), error) {
 	dbName := fmt.Sprintf("exampleload%d.db", exampleNum)
 	dbPath := filepath.Join(os.TempDir(), dbName)
-	db, err := walletdb.Create("bdb", dbPath, true)
+	db, err := walletdb.Create("bdb", dbPath, true, defaultDBTimeout)
 	if err != nil {
 		return nil, nil, err
 	}
@@ -111,7 +111,7 @@ func Example_basicUsage() {
 	// this, but it's done here in the example to ensure the example cleans
 	// up after itself.
 	dbPath := filepath.Join(os.TempDir(), "exampleusage.db")
-	db, err := walletdb.Create("bdb", dbPath, true)
+	db, err := walletdb.Create("bdb", dbPath, true, defaultDBTimeout)
 	if err != nil {
 		fmt.Println(err)
 		return