53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
|
package sqliteclidriver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSqliteCliDriver(t *testing.T) {
|
||
|
|
db, err := sql.Open("sqliteclidriver", ":memory:")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = db.Exec(`CREATE TABLE my_test_table ( id INTEGER PRIMARY KEY, extra TEXT NOT NULL );`)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
_, err = db.Exec(`INSERT INTO my_test_table (id, extra) VALUES (1337, "abcdef"), (9001, "whoop");`)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
res, err := db.Query(`SELECT * FROM my_test_table ORDER BY id ASC;`)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
cols, err := res.Columns()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.EqualValues(t, cols, []string{"id", "extra"})
|
||
|
|
|
||
|
|
var rowCount int = 0
|
||
|
|
|
||
|
|
for res.Next() {
|
||
|
|
rowCount++
|
||
|
|
|
||
|
|
var idVal int
|
||
|
|
var extraVal string
|
||
|
|
err = res.Scan(&idVal, &extraVal)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
switch rowCount {
|
||
|
|
case 1:
|
||
|
|
require.EqualValues(t, 1337, idVal)
|
||
|
|
require.EqualValues(t, "abcdef", extraVal)
|
||
|
|
|
||
|
|
case 2:
|
||
|
|
require.EqualValues(t, 9001, idVal)
|
||
|
|
require.EqualValues(t, "whoop", extraVal)
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, rowCount, 2)
|
||
|
|
}
|