mirror of
https://github.com/ncruces/go-sqlite3.git
synced 2026-01-12 05:59:14 +00:00
136 lines
2.8 KiB
Go
136 lines
2.8 KiB
Go
package unicode
|
||
|
||
import (
|
||
"errors"
|
||
"testing"
|
||
|
||
"github.com/ncruces/go-sqlite3"
|
||
_ "github.com/ncruces/go-sqlite3/embed"
|
||
)
|
||
|
||
func TestRegister(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
db, err := sqlite3.Open(":memory:")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer db.Close()
|
||
|
||
exec := func(fn string) string {
|
||
stmt, _, err := db.Prepare(`SELECT ` + fn)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer stmt.Close()
|
||
|
||
if stmt.Step() {
|
||
return stmt.ColumnText(0)
|
||
}
|
||
t.Fatal(stmt.Err())
|
||
return ""
|
||
}
|
||
|
||
Register(db)
|
||
|
||
tests := []struct {
|
||
test string
|
||
want string
|
||
}{
|
||
{`upper('hello')`, "HELLO"},
|
||
{`lower('HELLO')`, "hello"},
|
||
{`upper('привет')`, "ПРИВЕТ"},
|
||
{`lower('ПРИВЕТ')`, "привет"},
|
||
{`upper('istanbul')`, "ISTANBUL"},
|
||
{`upper('istanbul', 'tr-TR')`, "İSTANBUL"},
|
||
{`lower('Dünyanın İlk Borsası', 'tr-TR')`, "dünyanın ilk borsası"},
|
||
{`upper('Dünyanın İlk Borsası', 'tr-TR')`, "DÜNYANIN İLK BORSASI"},
|
||
{`'Hello' REGEXP 'ell'`, "1"},
|
||
{`'Hello' REGEXP 'el.'`, "1"},
|
||
{`'Hello' LIKE 'hel_'`, "0"},
|
||
{`'Hello' LIKE 'hel%'`, "1"},
|
||
{`'Hello' LIKE 'h_llo'`, "1"},
|
||
{`'Hello' LIKE 'hello'`, "1"},
|
||
{`'Привет' LIKE 'ПРИВЕТ'`, "1"},
|
||
{`'100%' LIKE '100|%' ESCAPE '|'`, "1"},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.test, func(t *testing.T) {
|
||
if got := exec(tt.test); got != tt.want {
|
||
t.Errorf("exec(%q) = %q, want %q", tt.test, got, tt.want)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestRegister_error(t *testing.T) {
|
||
t.Parallel()
|
||
|
||
db, err := sqlite3.Open(":memory:")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer db.Close()
|
||
|
||
Register(db)
|
||
|
||
err = db.Exec(`SELECT upper('hello', 'enUS')`)
|
||
if err == nil {
|
||
t.Error("want error")
|
||
}
|
||
if !errors.Is(err, sqlite3.ERROR) {
|
||
t.Errorf("got %v, want sqlite3.ERROR", err)
|
||
}
|
||
|
||
err = db.Exec(`SELECT lower('hello', 'enUS')`)
|
||
if err == nil {
|
||
t.Error("want error")
|
||
}
|
||
if !errors.Is(err, sqlite3.ERROR) {
|
||
t.Errorf("got %v, want sqlite3.ERROR", err)
|
||
}
|
||
|
||
err = db.Exec(`SELECT 'hello' REGEXP '\'`)
|
||
if err == nil {
|
||
t.Error("want error")
|
||
}
|
||
if !errors.Is(err, sqlite3.ERROR) {
|
||
t.Errorf("got %v, want sqlite3.ERROR", err)
|
||
}
|
||
|
||
err = db.Exec(`SELECT 'hello' LIKE 'HELLO' ESCAPE '\\' `)
|
||
if err == nil {
|
||
t.Error("want error")
|
||
}
|
||
if !errors.Is(err, sqlite3.ERROR) {
|
||
t.Errorf("got %v, want sqlite3.ERROR", err)
|
||
}
|
||
}
|
||
|
||
func Test_like2regex(t *testing.T) {
|
||
const prefix = `(?is)\A`
|
||
const sufix = `\z`
|
||
tests := []struct {
|
||
pattern string
|
||
escape rune
|
||
want string
|
||
}{
|
||
{`a`, -1, `a`},
|
||
{`a.`, -1, `a\.`},
|
||
{`a%`, -1, `a.*`},
|
||
{`a\`, -1, `a\\`},
|
||
{`a_b`, -1, `a.b`},
|
||
{`a|b`, '|', `ab`},
|
||
{`a|_`, '|', `a_`},
|
||
}
|
||
for _, tt := range tests {
|
||
t.Run(tt.pattern, func(t *testing.T) {
|
||
want := prefix + tt.want + sufix
|
||
if got := like2regex(tt.pattern, tt.escape); got != want {
|
||
t.Errorf("like2regex() = %q, want %q", got, want)
|
||
}
|
||
})
|
||
}
|
||
}
|