Adding tests

This commit is contained in:
2025-11-04 14:41:49 -05:00
parent 13e0efb67c
commit 3e2efae4ce
3 changed files with 37 additions and 17 deletions

View File

@@ -2,31 +2,28 @@ package db
import (
"database/sql"
"fmt"
"io/ioutil"
"go-sjles-pta-vote/config"
"os"
_ "github.com/glebarez/go-sqlite"
)
var db *sql.DB
func Connect() error {
func Connect(db_path string) error {
var err error
db, err = sql.Open("sqlite", dbConfig.DBPath)
db, err = sql.Open("sqlite", db_path)
if err != nil {
return err
}
sql_create, err := ioutil.ReadFile("./db_format.sql")
sql_create, err := os.ReadFile("./db_format.sql")
if err != nil {
return err
}
_, err = db.Exec(sql_create)
_, err = db.Exec(string(sql_create))
return err
}
@@ -35,4 +32,4 @@ func Close() {
if db != nil {
_ = db.Close()
}
}
}