84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package satellitedb
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"storj.io/storj/internal/testcontext"
|
|
)
|
|
|
|
func TestDatabase(t *testing.T) {
|
|
ctx := testcontext.New(t)
|
|
defer ctx.Cleanup()
|
|
|
|
t.Run("BeginTx return err when db is nil", func(t *testing.T) {
|
|
db := &ConsoleDB{
|
|
db: nil,
|
|
}
|
|
|
|
transaction, err := db.BeginTx(ctx)
|
|
assert.Error(t, err)
|
|
assert.Nil(t, transaction)
|
|
})
|
|
|
|
t.Run("BeginTx and Commit success", func(t *testing.T) {
|
|
db, err := NewConsoleDB("sqlite3", "file::memory:?mode=memory&cache=shared")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
transaction, err := db.BeginTx(ctx)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, transaction)
|
|
|
|
err = transaction.Commit()
|
|
assert.NoError(t, err)
|
|
|
|
err = db.Close()
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("BeginTx and Rollback success", func(t *testing.T) {
|
|
db, err := NewConsoleDB("sqlite3", "file::memory:?mode=memory&cache=shared")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
transaction, err := db.BeginTx(ctx)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, transaction)
|
|
|
|
err = transaction.Rollback()
|
|
assert.NoError(t, err)
|
|
|
|
err = db.Close()
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
t.Run("Commit fails", func(t *testing.T) {
|
|
transaction := &DBTx{
|
|
ConsoleDB: &ConsoleDB{
|
|
tx: nil,
|
|
},
|
|
}
|
|
|
|
err := transaction.Commit()
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
t.Run("Rollback fails", func(t *testing.T) {
|
|
transaction := &DBTx{
|
|
ConsoleDB: &ConsoleDB{
|
|
tx: nil,
|
|
},
|
|
}
|
|
|
|
err := transaction.Rollback()
|
|
assert.Error(t, err)
|
|
})
|
|
}
|