Adding tests
This commit is contained in:
@@ -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()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user