remove dependency of assert

This commit is contained in:
Yasuhiro Matsumoto 2017-03-05 21:33:03 +09:00
parent 0147a481cf
commit f694bb773c
1 changed files with 36 additions and 14 deletions

View File

@ -10,8 +10,6 @@ import (
"fmt"
"os"
"testing"
"github.com/stretchr/testify/assert"
)
type testModule struct {
@ -29,13 +27,27 @@ type testVTabCursor struct {
}
func (m testModule) Create(c *SQLiteConn, args []string) (VTab, error) {
assert.True(m.t, len(args) == 6, "six arguments expected")
assert.Equal(m.t, "test", args[0], "module name")
assert.Equal(m.t, "main", args[1], "db name")
assert.Equal(m.t, "vtab", args[2], "table name")
assert.Equal(m.t, "'1'", args[3], "first arg")
assert.Equal(m.t, "2", args[4], "second arg")
assert.Equal(m.t, "three", args[5], "third arg")
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
@ -116,17 +128,27 @@ func TestCreateModule(t *testing.T) {
},
})
db, err := sql.Open("sqlite3_TestCreateModule", tempFilename)
assert.Nil(t, err, "could not open db")
if err != nil {
t.Fatalf("could not open db: %v", err)
}
_, err = db.Exec("CREATE VIRTUAL TABLE vtab USING test('1', 2, three)")
assert.Nil(t, err, "could not create vtable")
if err != nil {
t.Fatal("could not create vtable: %v", err)
}
var i, value int
rows, err := db.Query("SELECT rowid, * FROM vtab WHERE test = '3'")
assert.Nil(t, err, "couldn't select from virtual table")
if err != nil {
t.Fatalf("couldn't select from virtual table: %v", err)
}
for rows.Next() {
rows.Scan(&i, &value)
assert.Equal(t, intarray[i], value)
if intarray[i] != value {
t.Fatalf("want %v but %v", intarray[i], value)
}
}
_, err = db.Exec("DROP TABLE vtab")
assert.Nil(t, err, "couldn't drop virtual table")
if err != nil {
t.Fatalf("couldn't drop virtual table: %v", err)
}
}