mirror of
https://github.com/chaisql/chai.git
synced 2026-04-22 23:07:05 +08:00
Asdine El Hrychy
116 lines
3.6 KiB
Go
116 lines
3.6 KiB
Go
package statement_test
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
|
|
_ "github.com/chaisql/chai"
|
|
errs "github.com/chaisql/chai/internal/errors"
|
|
"github.com/chaisql/chai/internal/testutil"
|
|
"github.com/cockroachdb/errors"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDropTable(t *testing.T) {
|
|
db, err := sql.Open("chai", ":memory:")
|
|
require.NoError(t, err)
|
|
defer db.Close()
|
|
|
|
_, err = db.Exec("CREATE TABLE test1(pk INT PRIMARY KEY, a INT UNIQUE); CREATE TABLE test2(pk INT PRIMARY KEY, a INT); CREATE TABLE test3(pk INT PRIMARY KEY, a INT)")
|
|
require.NoError(t, err)
|
|
|
|
_, err = db.Exec("DROP TABLE test1")
|
|
require.NoError(t, err)
|
|
|
|
_, err = db.Exec("DROP TABLE IF EXISTS test1")
|
|
require.NoError(t, err)
|
|
|
|
// Dropping a table that doesn't exist without "IF EXISTS"
|
|
// should return an error.
|
|
_, err = db.Exec("DROP TABLE test1")
|
|
require.Error(t, err)
|
|
|
|
// Assert that no other table has been dropped.
|
|
rows, err := db.Query("SELECT name FROM __chai_catalog WHERE type = 'table'")
|
|
require.NoError(t, err)
|
|
var tables []string
|
|
for rows.Next() {
|
|
var name string
|
|
err := rows.Scan(&name)
|
|
require.NoError(t, err)
|
|
tables = append(tables, name)
|
|
}
|
|
require.NoError(t, rows.Err())
|
|
require.NoError(t, rows.Close())
|
|
require.Equal(t, []string{"__chai_catalog", "__chai_sequence", "test2", "test3"}, tables)
|
|
|
|
// Assert the unique index test1_a_idx, created upon the creation of the table,
|
|
// has been dropped as well.
|
|
err = db.QueryRow("SELECT 1 FROM __chai_catalog WHERE name = 'test1_a_idx'").Scan(new(int))
|
|
require.Error(t, err)
|
|
|
|
// Assert the rowid sequence test1_seq, created upon the creation of the table,
|
|
// has been dropped as well.
|
|
err = db.QueryRow("SELECT 1 FROM __chai_catalog WHERE name = 'test1_seq'").Scan(new(int))
|
|
require.Error(t, err)
|
|
err = db.QueryRow("SELECT 1 FROM __chai_sequence WHERE name = 'test1_seq'").Scan(new(int))
|
|
require.Error(t, err)
|
|
|
|
// Dropping a read-only table should fail.
|
|
_, err = db.Exec("DROP TABLE __chai_catalog")
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestDropIndex(t *testing.T) {
|
|
db, tx, cleanup := testutil.NewTestTx(t)
|
|
defer cleanup()
|
|
|
|
testutil.MustExec(t, db, tx, `
|
|
CREATE TABLE test1(pk int primary key, foo text, bar int unique); CREATE INDEX idx_test1_foo ON test1(foo);
|
|
CREATE TABLE test2(pk int primary key, bar text); CREATE INDEX idx_test2_bar ON test2(bar);
|
|
`)
|
|
|
|
testutil.MustExec(t, db, tx, "DROP INDEX idx_test2_bar")
|
|
|
|
// Assert that the good index has been dropped.
|
|
indexes := tx.Catalog.ListIndexes("")
|
|
require.Len(t, indexes, 2)
|
|
require.Equal(t, "idx_test1_foo", indexes[0])
|
|
require.Equal(t, "test1_bar_idx", indexes[1])
|
|
|
|
// Dropping a non existing index with IF EXISTS should not fail.
|
|
err := testutil.Exec(db, tx, "DROP INDEX IF EXISTS unknown")
|
|
require.NoError(t, err)
|
|
|
|
// Dropping an index created with a table constraint should fail.
|
|
err = testutil.Exec(db, tx, "DROP INDEX test1_bar_idx")
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestDropSequence(t *testing.T) {
|
|
db, tx, cleanup := testutil.NewTestTx(t)
|
|
defer cleanup()
|
|
|
|
testutil.MustExec(t, db, tx, `
|
|
CREATE TABLE test1(foo int primary key, bar int unique);
|
|
CREATE SEQUENCE seq1;
|
|
CREATE SEQUENCE seq2;
|
|
`)
|
|
|
|
testutil.MustExec(t, db, tx, "DROP SEQUENCE seq1")
|
|
|
|
// Assert that the good index has been dropped.
|
|
_, err := tx.Catalog.GetSequence("seq1")
|
|
require.IsType(t, &errs.NotFoundError{}, errors.Unwrap(err))
|
|
_, err = tx.Catalog.GetSequence("seq2")
|
|
require.NoError(t, err)
|
|
|
|
// Dropping a non existing sequence with IF EXISTS should not fail.
|
|
err = testutil.Exec(db, tx, "DROP SEQUENCE IF EXISTS unknown")
|
|
require.NoError(t, err)
|
|
|
|
// Dropping a sequence created with a table constraint should fail.
|
|
err = testutil.Exec(db, tx, "DROP SEQUENCE test1_seq")
|
|
require.Error(t, err)
|
|
}
|