diff --git a/interface_test.go b/interface_test.go index 66ea7926..f223375d 100644 --- a/interface_test.go +++ b/interface_test.go @@ -10,6 +10,14 @@ import ( "testing" ) +var ( + // ignoreDbTypes are types which should be ignored when running tests + // that iterate all supported DB types. This allows some tests to add + // bogus drivers for testing purposes while still allowing other tests + // to easily iterate all supported drivers. + ignoreDbTypes = map[string]bool{"createopenfail": true} +) + // testNewestShaEmpty ensures the NewestSha returns the values expected by // the interface contract. func testNewestShaEmpty(t *testing.T, db btcdb.Db) { @@ -157,3 +165,79 @@ func TestCreateOpenUnsupported(t *testing.T) { return } } + +// testInterface tests performs tests for the various interfaces of btcdb which +// require state in the database for the given database type. +func testInterface(t *testing.T, dbType string) { + db, teardown, err := setupDB(dbType, "interface") + if err != nil { + t.Errorf("Failed to create test database %v", err) + return + } + defer teardown() + + // Load up a bunch of test blocks. + blocks, err := loadBlocks(t) + if err != nil { + t.Errorf("Unable to load blocks from test data %v: %v", + blockDataFile, err) + return + } + + t.Logf("Loaded %d blocks", len(blocks)) + for height := int64(1); height < int64(len(blocks)); height++ { + block := blocks[height] + + // Ensure there are no errors inserting each block into the + // database. + newHeight, err := db.InsertBlock(block) + if err != nil { + t.Errorf("InsertBlock: failed to insert block %v err %v", + height, err) + return + } + if newHeight != height { + t.Errorf("InsertBlock: height mismatch got: %v, want: %v", + newHeight, height) + return + } + } + + // TODO(davec): Need to figure out how to handle the special checks + // required for the duplicate transactions allowed by blocks 91842 and + // 91880 on the main network due to the old miner + Satoshi client bug. + + // TODO(davec): Add tests for the following functions: + /* + Close() + DropAfterBlockBySha(*btcwire.ShaHash) (err error) + ExistsSha(sha *btcwire.ShaHash) (exists bool) + FetchBlockBySha(sha *btcwire.ShaHash) (blk *btcutil.Block, err error) + FetchBlockShaByHeight(height int64) (sha *btcwire.ShaHash, err error) + FetchHeightRange(startHeight, endHeight int64) (rshalist []btcwire.ShaHash, err error) + ExistsTxSha(sha *btcwire.ShaHash) (exists bool) + FetchTxBySha(txsha *btcwire.ShaHash) ([]*TxListReply, error) + FetchTxByShaList(txShaList []*btcwire.ShaHash) []*TxListReply + FetchUnSpentTxByShaList(txShaList []*btcwire.ShaHash) []*TxListReply + - InsertBlock(block *btcutil.Block) (height int64, err error) + InvalidateBlockCache() + InvalidateCache() + InvalidateTxCache() + NewIterateBlocks() (pbi BlockIterator, err error) + NewestSha() (sha *btcwire.ShaHash, height int64, err error) + RollbackClose() + SetDBInsertMode(InsertMode) + Sync() + */ +} + +// TestInterface performs tests for the various interfaces of btcdb which +// require state in the database for each supported database type (those loaded +// in common_test.go that is). +func TestInterface(t *testing.T) { + for _, dbType := range btcdb.SupportedDBs() { + if _, exists := ignoreDbTypes[dbType]; !exists { + testInterface(t, dbType) + } + } +}