Rewrite to use embed.FS and sqlx.
This commit is contained in:
62
simplesql_test/simplesql_test.go
Normal file
62
simplesql_test/simplesql_test.go
Normal file
@ -0,0 +1,62 @@
|
||||
package simplesql_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"embed"
|
||||
"testing"
|
||||
|
||||
ssql "git.simplesystems.tech/bjj-gym-management/simplesql"
|
||||
"github.com/jmoiron/sqlx"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
//go:embed sqlq
|
||||
var sqlQueries embed.FS
|
||||
|
||||
type user struct {
|
||||
Username string `db:"username"`
|
||||
}
|
||||
|
||||
func TestNew(t *testing.T) {
|
||||
// Create in-memory db for test
|
||||
db, err := sqlx.Open("sqlite3", "file:temp.db?mode=memory")
|
||||
if err != nil {
|
||||
t.Fatalf("opening db: %v", err)
|
||||
}
|
||||
defer db.Close()
|
||||
|
||||
// Create table
|
||||
if _, err := db.Exec("CREATE TABLE users(id INT, username VARCHAR)"); err != nil {
|
||||
t.Fatalf("creating table: %v", err)
|
||||
}
|
||||
|
||||
// Add users
|
||||
if _, err := db.Exec("INSERT INTO users (id, username) VALUES (1, 'u1'), (2, 'u2')"); err != nil {
|
||||
t.Fatalf("adding users: %v", err)
|
||||
}
|
||||
|
||||
// Prepare statements
|
||||
stmts, err := ssql.Prepare(context.TODO(), db, sqlQueries)
|
||||
if err != nil {
|
||||
t.Fatalf("preparing statements: %v", err)
|
||||
}
|
||||
defer stmts.Close()
|
||||
|
||||
if len(stmts) != 1 {
|
||||
t.Errorf("expecting 1 statement, but got %d", len(stmts))
|
||||
}
|
||||
|
||||
if _, ok := stmts["get_user"]; !ok {
|
||||
t.Error("expecting get_user to be in stmts, but it is not")
|
||||
}
|
||||
|
||||
// Get a user from the table, using the prepared statement
|
||||
var res user
|
||||
if err := stmts["get_user"].Get(&res, map[string]interface{}{"user_id": "2"}); err != nil {
|
||||
t.Fatalf("getting users: %v", err)
|
||||
}
|
||||
|
||||
if res.Username != "u2" {
|
||||
t.Errorf("expecting u2, but got %s", res.Username)
|
||||
}
|
||||
}
|
2
simplesql_test/sqlq/get_user.sql
Normal file
2
simplesql_test/sqlq/get_user.sql
Normal file
@ -0,0 +1,2 @@
|
||||
-- Can use SQL comments here!
|
||||
SELECT username FROM users WHERE id = :user_id;
|
Reference in New Issue
Block a user