// TestAddDuplicateDriver ensures that adding a duplicate driver does not // overwrite an existing one. func TestAddDuplicateDriver(t *testing.T) { supportedDrivers := database.SupportedDrivers() if len(supportedDrivers) == 0 { t.Errorf("no backends to test") return } dbType := supportedDrivers[0] // bogusCreateDB is a function which acts as a bogus create and open // driver function and intentionally returns a failure that can be // detected if the interface allows a duplicate driver to overwrite an // existing one. bogusCreateDB := func(args ...interface{}) (database.DB, error) { return nil, fmt.Errorf("duplicate driver allowed for database "+ "type [%v]", dbType) } // Create a driver that tries to replace an existing one. Set its // create and open functions to a function that causes a test failure if // they are invoked. driver := database.Driver{ DbType: dbType, Create: bogusCreateDB, Open: bogusCreateDB, } testName := "duplicate driver registration" err := database.RegisterDriver(driver) if !checkDbError(t, testName, err, database.ErrDbTypeRegistered) { return } }
import ( "errors" "fmt" "os" "path/filepath" "github.com/btcsuite/btcd/chaincfg" database "github.com/btcsuite/btcd/database2" _ "github.com/btcsuite/btcd/database2/ffldb" "github.com/btcsuite/btcd/wire" "github.com/btcsuite/btcutil" ) var ( btcdHomeDir = btcutil.AppDataDir("btcd", false) knownDbTypes = database.SupportedDrivers() activeNetParams = &chaincfg.MainNetParams // Default global config. cfg = &config{ DataDir: filepath.Join(btcdHomeDir, "data"), DbType: "ffldb", } ) // config defines the global configuration options. type config struct { DataDir string `short:"b" long:"datadir" description:"Location of the btcd data directory"` DbType string `long:"dbtype" description:"Database backend to use for the Block Chain"` TestNet3 bool `long:"testnet" description:"Use the test network"` RegressionTest bool `long:"regtest" description:"Use the regression test network"`