Adding tests for adding votes

server/services/services_test.go

@@ -13,6 +13,85 @@ import (
 
 const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890-=`~!@#$%^&*()_+[]\\;',./{}|:\"<>?"
 
+var new_members = []struct{
+	email string
+	member_name string
+}{
+	{"test1@mail.me", "test1"},
+	{"test2@mail.me", "test2"},
+	{"test3@mail.me", "test3"},
+	{"test4@mail.me", "test4"},
+	{"test5@mail.me", "test5"},
+	{"test6@mail.me", "test6"},
+	{"test7@mail.me", "test7"},
+	{"test100@mail.me", "test100"},
+	{"test101@mail.me", "test101"},
+	{"test102@mail.me", "test102"},
+	{"test103@mail.me", "test103"},
+	{"test104@mail.me", "test104"},
+	{"test105@mail.me", "test105"},
+}
+
+var new_polls = []struct{
+	question string
+	member_yes_votes int64
+	member_no_votes int64
+	non_member_yes_votes int64
+	non_member_no_votes int64
+}{
+	{"ques1", 1, 2, 3, 4},
+	{"ques2", 3, 2, 4, 5},
+	{"ques3", 4, 3, 6, 5},
+}
+
+var new_voters = []struct{
+	poll_id int64
+	voter_email string
+}{
+	{1, "test1@mail.me"},
+	{1, "test2@mail.me"},
+	{1, "test3@mail.me"},
+	{1, "test10@mail.me"},
+	{1, "test11@mail.me"},
+	{1, "test12@mail.me"},
+	{1, "test13@mail.me"},
+	{1, "test14@mail.me"},
+	{1, "test15@mail.me"},
+	{1, "test16@mail.me"},
+	{2, "test1@mail.me"},
+	{2, "test2@mail.me"},
+	{2, "test3@mail.me"},
+	{2, "test4@mail.me"},
+	{2, "test5@mail.me"},
+	{2, "test10@mail.me"},
+	{2, "test11@mail.me"},
+	{2, "test12@mail.me"},
+	{2, "test13@mail.me"},
+	{2, "test14@mail.me"},
+	{2, "test15@mail.me"},
+	{2, "test16@mail.me"},
+	{2, "test17@mail.me"},
+	{2, "test18@mail.me"},
+	{3, "test1@mail.me"},
+	{3, "test2@mail.me"},
+	{3, "test3@mail.me"},
+	{3, "test4@mail.me"},
+	{3, "test5@mail.me"},
+	{3, "test6@mail.me"},
+	{3, "test7@mail.me"},
+	{3, "test10@mail.me"},
+	{3, "test11@mail.me"},
+	{3, "test12@mail.me"},
+	{3, "test13@mail.me"},
+	{3, "test14@mail.me"},
+	{3, "test15@mail.me"},
+	{3, "test16@mail.me"},
+	{3, "test17@mail.me"},
+	{3, "test18@mail.me"},
+	{3, "test19@mail.me"},
+	{3, "test20@mail.me"},
+}
+
 func RandString(length int) string {
 	rand_bytes := make([]byte, length)
 	for rand_index := range length {
@@ -21,6 +100,44 @@ func RandString(length int) string {
 	return string(rand_bytes)
 }
 
+func PreLoadDB() error {
+	db_conn, err := db.Connect()
+	if err != nil {
+		return err
+	}
+	defer db.Close()
+
+	// Insert members
+	for i := range new_members {
+		_, err := db_conn.Exec(`INSERT INTO members (email, member_name) VALUES (?, ?)`, new_members[i].email, new_members[i].member_name)
+		if err != nil {
+			return err
+		}
+	}
+
+	// Insert polls
+	for i := range new_polls {
+		result, err := db_conn.Exec(`INSERT INTO polls (question, member_yes_votes, member_no_votes, non_member_yes_votes, non_member_no_votes, expires_at) VALUES (?, ?, ?, ?, ?, ?)`, new_polls[i].question, new_polls[i].member_yes_votes, new_polls[i].member_no_votes, new_polls[i].non_member_yes_votes, new_polls[i].non_member_no_votes, time.Now().Add(time.Hour * 10).Format("2006-01-02 15:04:05"))
+		if err != nil {
+			return err
+		}
+		_, err = result.LastInsertId()
+		if err != nil {
+			return err
+		}
+	}
+
+	// Insert voters
+	for i := range new_voters {
+		_, err := db_conn.Exec(`INSERT INTO voters (poll_id, voter_email) VALUES (?, ?)`, new_voters[i].poll_id, new_voters[i].voter_email)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
 func TestCreatePoll(t *testing.T) {
 	parameters := []struct{
 		question string
@@ -214,3 +331,66 @@ func TestGetCreatePollByQuestion(t *testing.T) {
 	}
 }
 
+func TestSetVote(t *testing.T) {
+    // Preload the database with members, polls, and voters
+    tmp_db, err := os.CreateTemp("", "vote_test.*.db")
+    if err != nil {
+        t.Fatalf("Failed to create temporary database: %v", err)
+    }
+    defer os.Remove(tmp_db.Name())
+
+    init_conf := &config.Config{
+        DBPath: string(tmp_db.Name()),
+    }
+    config.SetConfig(init_conf)
+
+    err = PreLoadDB()
+    if err != nil {
+        t.Fatalf("Failed to preload database: %v", err)
+    }
+
+    // Add a non-member vote
+    random_email := RandString(10) + "@mail.me"
+    vote := &models.Vote{
+        PollId: 1,
+        Email: random_email,
+        Vote: true,
+    }
+    err = SetVote(vote)
+    if err != nil {
+        t.Fatalf("Failed to set non-member vote: %v", err)
+    }
+
+    // Add a member vote
+    member_email := "test100@mail.me"
+    vote = &models.Vote{
+        PollId: 1,
+        Email: member_email,
+        Vote: true,
+    }
+    err = SetVote(vote)
+    if err != nil {
+        t.Fatalf("Failed to set member vote: %v", err)
+    }
+
+    // Verify the votes were added correctly
+    voters, err := models.GetVoters(1) // Use GetVoters from models
+    if err != nil {
+        t.Fatalf("Failed to get voters: %v", err)
+    }
+
+    expected_non_member_votes := 4 + 1 // Original non-member votes + new non-member vote
+    expected_member_votes := 3 + 1    // Original member votes + new member vote
+
+    for _, voter := range voters {
+        if voter.Email == random_email && voter.YesVote {
+            expected_non_member_votes--
+        } else if voter.Email == member_email && voter.YesVote {
+            expected_member_votes--
+        }
+    }
+
+    if expected_non_member_votes != 5 || expected_member_votes != 4 {
+        t.Errorf("Expected %d non-member votes and %d member votes, but got %d non-member votes and %d member votes", 4+1, 3+1, expected_non_member_votes, expected_member_votes)
+    }
+}