From 3e2efae4ced3e836c7f1f7c617705c8c5be2c006 Mon Sep 17 00:00:00 2001
From: paul <paul@halvo.me>
Date: Tue, 4 Nov 2025 14:41:49 -0500
Subject: [PATCH] Adding tests

---
 server/db/db.go         | 15 ++++++---------
 server/db/db_format.sql | 15 +++++++--------
 server/db/db_test.go    | 24 ++++++++++++++++++++++++
 3 files changed, 37 insertions(+), 17 deletions(-)
 create mode 100644 server/db/db_test.go

diff --git a/server/db/db.go b/server/db/db.go
index 69321c4..60c1ccc 100644
--- a/server/db/db.go
+++ b/server/db/db.go
@@ -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()
 	}
-}
\ No newline at end of file
+}
diff --git a/server/db/db_format.sql b/server/db/db_format.sql
index a72260a..1721e07 100644
--- a/server/db/db_format.sql
+++ b/server/db/db_format.sql
@@ -1,5 +1,5 @@
-CREATE TABLE IF NOT EXISTS poll (
-    id UNSIGNED INT NOT NULL AUTOINCREMENT,
+CREATE TABLE IF NOT EXISTS polls (
+    id INTEGER PRIMARY KEY,
     question TEXT NOT NULL,
     member_yes_votes UNSIGNED INT NOT NULL,
     member_no_votes UNSIGNED INT NOT NULL,
@@ -7,19 +7,18 @@ CREATE TABLE IF NOT EXISTS poll (
     non_member_no_votes UNSIGNED INT NOT NULL,
     created_at DATETIME,
     updated_at DATETIME,
-    expires_at DATETIME,
-    PRIMARY KEY (id)
-)
+    expires_at DATETIME
+);
 
 CREATE TABLE IF NOT EXISTS voters (
     poll_id UNSIGNED INT NOT NULL,
     voter_email TEXT NOT NULL,
-    FOREIGN KEY (poll_id) poll(id),
+    FOREIGN KEY (poll_id) REFERENCES polls(id),
     PRIMARY KEY (poll_id, voter_email)
-)
+);
 
 CREATE TABLE IF NOT EXISTS members (
     email TEXT NOT NULL,
     member_name TEXT,
     PRIMARY KEY (email)
-)
+);
diff --git a/server/db/db_test.go b/server/db/db_test.go
new file mode 100644
index 0000000..ef57785
--- /dev/null
+++ b/server/db/db_test.go
@@ -0,0 +1,24 @@
+package db
+
+import (
+	"os"
+	"testing"
+)
+
+func TestConnect(t *testing.T) {
+	tmp_db, err := os.CreateTemp("", "vote_test.*.db")
+	tmp_db_name := tmp_db.Name()
+	tmp_db.Close()
+
+	defer os.Remove(tmp_db_name)
+
+	if err != nil {
+		t.Errorf(`Failed to create temporary db: %v`, err)
+	}
+
+	if err := Connect(tmp_db_name); err != nil {
+		t.Errorf(`Failed to create the database at %s: %v`, tmp_db_name, err)
+	}
+
+	defer Close()
+}
\ No newline at end of file