diff options
Diffstat (limited to 'tests/golite.go')
-rw-r--r-- | tests/golite.go | 578 |
1 files changed, 0 insertions, 578 deletions
diff --git a/tests/golite.go b/tests/golite.go index 1cb9436..682cead 100644 --- a/tests/golite.go +++ b/tests/golite.go @@ -1842,298 +1842,6 @@ func TestUnlockNotifyDeadlock(t *testing.T) { wg.Wait() } -type testModule struct { - t *testing.T - intarray []int -} - -type testVTab struct { - intarray []int -} - -type testVTabCursor struct { - vTab *testVTab - index int -} - -func (m testModule) Create(c *SQLiteConn, args []string) (VTab, error) { - if len(args) != 6 { - m.t.Fatal("six arguments expected") - } - if args[0] != "test" { - m.t.Fatal("module name") - } - if args[1] != "main" { - m.t.Fatal("db name") - } - if args[2] != "vtab" { - m.t.Fatal("table name") - } - if args[3] != "'1'" { - m.t.Fatal("first arg") - } - if args[4] != "2" { - m.t.Fatal("second arg") - } - if args[5] != "three" { - m.t.Fatal("third argsecond arg") - } - err := c.DeclareVTab("CREATE TABLE x(test TEXT)") - if err != nil { - return nil, err - } - return &testVTab{m.intarray}, nil -} - -func (m testModule) Connect(c *SQLiteConn, args []string) (VTab, error) { - return m.Create(c, args) -} - -func (m testModule) DestroyModule() {} - -func (v *testVTab) BestIndex(cst []InfoConstraint, ob []InfoOrderBy) (*IndexResult, error) { - used := make([]bool, 0, len(cst)) - for range cst { - used = append(used, false) - } - return &IndexResult{ - Used: used, - IdxNum: 0, - IdxStr: "test-index", - AlreadyOrdered: true, - EstimatedCost: 100, - EstimatedRows: 200, - }, nil -} - -func (v *testVTab) Disconnect() error { - return nil -} - -func (v *testVTab) Destroy() error { - return nil -} - -func (v *testVTab) Open() (VTabCursor, error) { - return &testVTabCursor{v, 0}, nil -} - -func (vc *testVTabCursor) Close() error { - return nil -} - -func (vc *testVTabCursor) Filter(idxNum int, idxStr string, vals []any) error { - vc.index = 0 - return nil -} - -func (vc *testVTabCursor) Next() error { - vc.index++ - return nil -} - -func (vc *testVTabCursor) EOF() bool { - return vc.index >= len(vc.vTab.intarray) -} - -func (vc *testVTabCursor) Column(c *SQLiteContext, col int) error { - if col != 0 { - return fmt.Errorf("column index out of bounds: %d", col) - } - c.ResultInt(vc.vTab.intarray[vc.index]) - return nil -} - -func (vc *testVTabCursor) Rowid() (int64, error) { - return int64(vc.index), nil -} - -func TestCreateModule(t *testing.T) { - intarray := []int{1, 2, 3} - sql.Register("sqlite3_TestCreateModule", &SQLiteDriver{ - ConnectHook: func(conn *SQLiteConn) error { - return conn.CreateModule("test", testModule{t, intarray}) - }, - }) - db, err := sql.Open("sqlite3_TestCreateModule", ":memory:") - if err != nil { - t.Fatalf("could not open db: %v", err) - } - _, err = db.Exec("CREATE VIRTUAL TABLE vtab USING test('1', 2, three)") - if err != nil { - t.Fatalf("could not create vtable: %v", err) - } - - var i, value int - rows, err := db.Query("SELECT rowid, * FROM vtab WHERE test = '3'") - if err != nil { - t.Fatalf("couldn't select from virtual table: %v", err) - } - for rows.Next() { - rows.Scan(&i, &value) - if intarray[i] != value { - t.Fatalf("want %v but %v", intarray[i], value) - } - } - - _, err = db.Exec("DROP TABLE vtab") - if err != nil { - t.Fatalf("couldn't drop virtual table: %v", err) - } -} - -func TestVUpdate(t *testing.T) { - // create module - updateMod := &vtabUpdateModule{t, make(map[string]*vtabUpdateTable)} - - // register module - sql.Register("sqlite3_TestVUpdate", &SQLiteDriver{ - ConnectHook: func(conn *SQLiteConn) error { - return conn.CreateModule("updatetest", updateMod) - }, - }) - - // connect - db, err := sql.Open("sqlite3_TestVUpdate", ":memory:") - if err != nil { - t.Fatalf("could not open db: %v", err) - } - - // create test table - _, err = db.Exec(`CREATE VIRTUAL TABLE vt USING updatetest(f1 integer, f2 text, f3 text)`) - if err != nil { - t.Fatalf("could not create updatetest vtable vt, got: %v", err) - } - - // check that table is defined properly - if len(updateMod.tables) != 1 { - t.Fatalf("expected exactly 1 table to exist, got: %d", len(updateMod.tables)) - } - if _, ok := updateMod.tables["vt"]; !ok { - t.Fatalf("expected table `vt` to exist in tables") - } - - // check nothing in updatetest - rows, err := db.Query(`select * from vt`) - if err != nil { - t.Fatalf("could not query vt, got: %v", err) - } - i, err := getRowCount(rows) - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - if i != 0 { - t.Fatalf("expected no rows in vt, got: %d", i) - } - - _, err = db.Exec(`delete from vt where f1 = 'yes'`) - if err != nil { - t.Fatalf("expected error on delete, got nil") - } - - // test bad column name - _, err = db.Exec(`insert into vt (f4) values('a')`) - if err == nil { - t.Fatalf("expected error on insert, got nil") - } - - // insert to vt - res, err := db.Exec(`insert into vt (f1, f2, f3) values (115, 'b', 'c'), (116, 'd', 'e')`) - if err != nil { - t.Fatalf("expected no error on insert, got: %v", err) - } - n, err := res.RowsAffected() - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - if n != 2 { - t.Fatalf("expected 1 row affected, got: %d", n) - } - - // check vt table - vt := updateMod.tables["vt"] - if len(vt.data) != 2 { - t.Fatalf("expected table vt to have exactly 2 rows, got: %d", len(vt.data)) - } - if !reflect.DeepEqual(vt.data[0], []any{int64(115), "b", "c"}) { - t.Fatalf("expected table vt entry 0 to be [115 b c], instead: %v", vt.data[0]) - } - if !reflect.DeepEqual(vt.data[1], []any{int64(116), "d", "e"}) { - t.Fatalf("expected table vt entry 1 to be [116 d e], instead: %v", vt.data[1]) - } - - // query vt - var f1 int - var f2, f3 string - err = db.QueryRow(`select * from vt where f1 = 115`).Scan(&f1, &f2, &f3) - if err != nil { - t.Fatalf("expected no error on vt query, got: %v", err) - } - - // check column values - if f1 != 115 || f2 != "b" || f3 != "c" { - t.Errorf("expected f1==115, f2==b, f3==c, got: %d, %q, %q", f1, f2, f3) - } - - // update vt - res, err = db.Exec(`update vt set f1=117, f2='f' where f3='e'`) - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - n, err = res.RowsAffected() - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - if n != 1 { - t.Fatalf("expected exactly one row updated, got: %d", n) - } - - // check vt table - if len(vt.data) != 2 { - t.Fatalf("expected table vt to have exactly 2 rows, got: %d", len(vt.data)) - } - if !reflect.DeepEqual(vt.data[0], []any{int64(115), "b", "c"}) { - t.Fatalf("expected table vt entry 0 to be [115 b c], instead: %v", vt.data[0]) - } - if !reflect.DeepEqual(vt.data[1], []any{int64(117), "f", "e"}) { - t.Fatalf("expected table vt entry 1 to be [117 f e], instead: %v", vt.data[1]) - } - - // delete from vt - res, err = db.Exec(`delete from vt where f1 = 117`) - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - n, err = res.RowsAffected() - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - if n != 1 { - t.Fatalf("expected exactly one row deleted, got: %d", n) - } - - // check vt table - if len(vt.data) != 1 { - t.Fatalf("expected table vt to have exactly 1 row, got: %d", len(vt.data)) - } - if !reflect.DeepEqual(vt.data[0], []any{int64(115), "b", "c"}) { - t.Fatalf("expected table vt entry 0 to be [115 b c], instead: %v", vt.data[0]) - } - - // check updatetest has 1 result - rows, err = db.Query(`select * from vt`) - if err != nil { - t.Fatalf("could not query vt, got: %v", err) - } - i, err = getRowCount(rows) - if err != nil { - t.Fatalf("expected no error, got: %v", err) - } - if i != 1 { - t.Fatalf("expected 1 row in vt, got: %d", i) - } -} - func getRowCount(rows *sql.Rows) (int, error) { var i int for rows.Next() { @@ -2142,289 +1850,6 @@ func getRowCount(rows *sql.Rows) (int, error) { return i, nil } -type vtabUpdateModule struct { - t *testing.T - tables map[string]*vtabUpdateTable -} - -func (m *vtabUpdateModule) Create(c *SQLiteConn, args []string) (VTab, error) { - if len(args) < 2 { - return nil, errors.New("must declare at least one column") - } - - // get database name, table name, and column declarations ... - dbname, tname, decls := args[1], args[2], args[3:] - - // extract column names + types from parameters declarations - cols, typs := make([]string, len(decls)), make([]string, len(decls)) - for i := 0; i < len(decls); i++ { - n, typ := decls[i], "" - if j := strings.IndexAny(n, " \t\n"); j != -1 { - typ, n = strings.TrimSpace(n[j+1:]), n[:j] - } - cols[i], typs[i] = n, typ - } - - // declare table - err := c.DeclareVTab(fmt.Sprintf(`CREATE TABLE "%s"."%s" (%s)`, dbname, tname, strings.Join(decls, ","))) - if err != nil { - return nil, err - } - - // create table - vtab := &vtabUpdateTable{m.t, dbname, tname, cols, typs, make([][]any, 0)} - m.tables[tname] = vtab - return vtab, nil -} - -func (m *vtabUpdateModule) Connect(c *SQLiteConn, args []string) (VTab, error) { - return m.Create(c, args) -} - -func (m *vtabUpdateModule) DestroyModule() {} - -type vtabUpdateTable struct { - t *testing.T - db string - name string - cols []string - typs []string - data [][]any -} - -func (t *vtabUpdateTable) Open() (VTabCursor, error) { - return &vtabUpdateCursor{t, 0}, nil -} - -func (t *vtabUpdateTable) BestIndex(cst []InfoConstraint, ob []InfoOrderBy) (*IndexResult, error) { - return &IndexResult{Used: make([]bool, len(cst))}, nil -} - -func (t *vtabUpdateTable) Disconnect() error { - return nil -} - -func (t *vtabUpdateTable) Destroy() error { - return nil -} - -func (t *vtabUpdateTable) Insert(id any, vals []any) (int64, error) { - var i int64 - if id == nil { - i, t.data = int64(len(t.data)), append(t.data, vals) - return i, nil - } - - var ok bool - i, ok = id.(int64) - if !ok { - return 0, fmt.Errorf("id is invalid type: %T", id) - } - - t.data[i] = vals - - return i, nil -} - -func (t *vtabUpdateTable) Update(id any, vals []any) error { - i, ok := id.(int64) - if !ok { - return fmt.Errorf("id is invalid type: %T", id) - } - - if int(i) >= len(t.data) || i < 0 { - return fmt.Errorf("invalid row id %d", i) - } - - t.data[int(i)] = vals - - return nil -} - -func (t *vtabUpdateTable) Delete(id any) error { - i, ok := id.(int64) - if !ok { - return fmt.Errorf("id is invalid type: %T", id) - } - - if int(i) >= len(t.data) || i < 0 { - return fmt.Errorf("invalid row id %d", i) - } - - t.data = append(t.data[:i], t.data[i+1:]...) - - return nil -} - -type vtabUpdateCursor struct { - t *vtabUpdateTable - i int -} - -func (c *vtabUpdateCursor) Column(ctxt *SQLiteContext, col int) error { - switch x := c.t.data[c.i][col].(type) { - case []byte: - ctxt.ResultBlob(x) - case bool: - ctxt.ResultBool(x) - case float64: - ctxt.ResultDouble(x) - case int: - ctxt.ResultInt(x) - case int64: - ctxt.ResultInt64(x) - case nil: - ctxt.ResultNull() - case string: - ctxt.ResultText(x) - default: - ctxt.ResultText(fmt.Sprintf("%v", x)) - } - - return nil -} - -func (c *vtabUpdateCursor) Filter(ixNum int, ixName string, vals []any) error { - return nil -} - -func (c *vtabUpdateCursor) Next() error { - c.i++ - return nil -} - -func (c *vtabUpdateCursor) EOF() bool { - return c.i >= len(c.t.data) -} - -func (c *vtabUpdateCursor) Rowid() (int64, error) { - return int64(c.i), nil -} - -func (c *vtabUpdateCursor) Close() error { - return nil -} - -type testModuleEponymousOnly struct { - t *testing.T - intarray []int -} - -type testVTabEponymousOnly struct { - intarray []int -} - -type testVTabCursorEponymousOnly struct { - vTab *testVTabEponymousOnly - index int -} - -func (m testModuleEponymousOnly) EponymousOnlyModule() {} - -func (m testModuleEponymousOnly) Create(c *SQLiteConn, args []string) (VTab, error) { - err := c.DeclareVTab("CREATE TABLE x(test INT)") - if err != nil { - return nil, err - } - return &testVTabEponymousOnly{m.intarray}, nil -} - -func (m testModuleEponymousOnly) Connect(c *SQLiteConn, args []string) (VTab, error) { - return m.Create(c, args) -} - -func (m testModuleEponymousOnly) DestroyModule() {} - -func (v *testVTabEponymousOnly) BestIndex(cst []InfoConstraint, ob []InfoOrderBy) (*IndexResult, error) { - used := make([]bool, 0, len(cst)) - for range cst { - used = append(used, false) - } - return &IndexResult{ - Used: used, - IdxNum: 0, - IdxStr: "test-index", - AlreadyOrdered: true, - EstimatedCost: 100, - EstimatedRows: 200, - }, nil -} - -func (v *testVTabEponymousOnly) Disconnect() error { - return nil -} - -func (v *testVTabEponymousOnly) Destroy() error { - return nil -} - -func (v *testVTabEponymousOnly) Open() (VTabCursor, error) { - return &testVTabCursorEponymousOnly{v, 0}, nil -} - -func (vc *testVTabCursorEponymousOnly) Close() error { - return nil -} - -func (vc *testVTabCursorEponymousOnly) Filter(idxNum int, idxStr string, vals []any) error { - vc.index = 0 - return nil -} - -func (vc *testVTabCursorEponymousOnly) Next() error { - vc.index++ - return nil -} - -func (vc *testVTabCursorEponymousOnly) EOF() bool { - return vc.index >= len(vc.vTab.intarray) -} - -func (vc *testVTabCursorEponymousOnly) Column(c *SQLiteContext, col int) error { - if col != 0 { - return fmt.Errorf("column index out of bounds: %d", col) - } - c.ResultInt(vc.vTab.intarray[vc.index]) - return nil -} - -func (vc *testVTabCursorEponymousOnly) Rowid() (int64, error) { - return int64(vc.index), nil -} - -func TestCreateModuleEponymousOnly(t *testing.T) { - intarray := []int{1, 2, 3} - sql.Register("sqlite3_TestCreateModuleEponymousOnly", &SQLiteDriver{ - ConnectHook: func(conn *SQLiteConn) error { - return conn.CreateModule("test", testModuleEponymousOnly{t, intarray}) - }, - }) - db, err := sql.Open("sqlite3_TestCreateModuleEponymousOnly", ":memory:") - if err != nil { - t.Fatalf("could not open db: %v", err) - } - - var i, value int - rows, err := db.Query("SELECT rowid, * FROM test") - if err != nil { - t.Fatalf("couldn't select from virtual table: %v", err) - } - for rows.Next() { - err := rows.Scan(&i, &value) - if err != nil { - t.Fatal(err) - } - if intarray[i] != value { - t.Fatalf("want %v but %v", intarray[i], value) - } - } - - _, err = db.Exec("DROP TABLE test") - if err != nil { - t.Logf("couldn't drop virtual table: %v", err) - } -} - func TempFilename(t testing.TB) string { f, err := ioutil.TempFile("", "go-sqlite3-test-") if err != nil { @@ -4873,9 +4298,6 @@ func MainTest() { // { "TestUnlockNotify", TestUnlockNotify }, // { "TestUnlockNotifyMany", TestUnlockNotifyMany }, // { "TestUnlockNotifyDeadlock", TestUnlockNotifyDeadlock }, - { "TestCreateModule", TestCreateModule }, - { "TestVUpdate", TestVUpdate }, - { "TestCreateModuleEponymousOnly", TestCreateModuleEponymousOnly }, { "TestOpen", TestOpen }, { "TestOpenWithVFS", TestOpenWithVFS }, { "TestOpenNoCreate", TestOpenNoCreate }, |