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) } }