Skip to content

Commit f7f8019

Browse files
committed
UPD: TestAuthCreateDatabase
* Renamed test * Implemented 'exists' test * WIP mattn#580
1 parent 183e7d6 commit f7f8019

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

sqlite3_opt_userauth_test.go

+10-9
Original file line numberDiff line numberDiff line change
@@ -9,31 +9,32 @@ package sqlite3
99

1010
import (
1111
"database/sql"
12-
"fmt"
12+
"os"
1313
"testing"
1414
)
1515

16-
func TestCreateAuthDatabase(t *testing.T) {
16+
func TestAuthCreateDatabase(t *testing.T) {
1717
tempFilename := TempFilename(t)
18-
fmt.Println(tempFilename) // debug
19-
//defer os.Remove(tempFilename) // Disable for debug
18+
defer os.Remove(tempFilename)
2019

2120
db, err := sql.Open("sqlite3", "file:"+tempFilename+"?_auth&_auth_user=admin&_auth_pass=admin")
2221
if err != nil {
2322
t.Fatal("Failed to open database:", err)
2423
}
2524
defer db.Close()
2625

27-
var i int64
28-
err = db.QueryRow("SELECT count(type) FROM sqlite_master WHERE type='table' AND name='sqlite_user';").Scan(&i)
29-
if err != nil {
26+
// Ping database
27+
if err := db.Ping(); err != nil {
3028
t.Fatal(err)
3129
}
32-
t.Logf("sqlite_user exists: %d", i)
3330

34-
_, err = db.Exec("SELECT auth_user_add('test', 'test', false);", nil)
31+
var exists bool
32+
err = db.QueryRow("select count(type) from sqlite_master WHERE type='table' and name='sqlite_user';").Scan(&exists)
3533
if err != nil {
3634
t.Fatal(err)
3735
}
3836

37+
if !exists {
38+
t.Fatal("failed to enable User Authentication")
39+
}
3940
}

0 commit comments

Comments
 (0)