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" "fmt"
"os" "os"
"testing" "testing"
"github.com/stretchr/testify/assert"
) )
type testModule struct { type testModule struct {
@ -29,13 +27,27 @@ type testVTabCursor struct {
} }
func (m testModule) Create(c *SQLiteConn, args []string) (VTab, error) { func (m testModule) Create(c *SQLiteConn, args []string) (VTab, error) {
assert.True(m.t, len(args) == 6, "six arguments expected") if len(args) != 6 {
assert.Equal(m.t, "test", args[0], "module name") m.t.Fatal("six arguments expected")
assert.Equal(m.t, "main", args[1], "db name") }
assert.Equal(m.t, "vtab", args[2], "table name") if args[0] != "test" {
assert.Equal(m.t, "'1'", args[3], "first arg") m.t.Fatal("module name")
assert.Equal(m.t, "2", args[4], "second arg") }
assert.Equal(m.t, "three", args[5], "third arg") 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)") err := c.DeclareVTab("CREATE TABLE x(test TEXT)")
if err != nil { if err != nil {
return nil, err return nil, err
@ -116,17 +128,27 @@ func TestCreateModule(t *testing.T) {
}, },
}) })
db, err := sql.Open("sqlite3_TestCreateModule", tempFilename) 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)") _, 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 var i, value int
rows, err := db.Query("SELECT rowid, * FROM vtab WHERE test = '3'") 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() { for rows.Next() {
rows.Scan(&i, &value) 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") _, 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)
}
} }