From 64099e47712d948eaf55b4f26dc6d5b7b424307c Mon Sep 17 00:00:00 2001
From: Michael Hunteman <michael@huntm.net>
Date: Sat, 31 Aug 2024 13:33:44 -0700
Subject: Fix tests

---
 server/test/guest_test.go | 38 +++++++++++++++++++++-----------------
 1 file changed, 21 insertions(+), 17 deletions(-)

(limited to 'server/test')

diff --git a/server/test/guest_test.go b/server/test/guest_test.go
index 26bd143..7a74443 100644
--- a/server/test/guest_test.go
+++ b/server/test/guest_test.go
@@ -36,12 +36,14 @@ func TestUpdateRSVP(test *testing.T) {
 	adminStore := admin.NewStore(databasePool)
 	adminHandler := admin.NewAdminHandler(adminStore, guestStore)
 
+	adminLogin := logInAdmin(adminHandler, test)
+	adminToken := adminLogin.Token
+	postGuest(guestHandler, adminToken, test)
+
 	guestLogin := logInGuest(guestHandler, test)
 	guestToken := guestLogin.Token
-	addPartyGuest(guestHandler, guestToken, test)
+	addPartyMember(guestHandler, guestToken, test)
 
-	adminLogin := logInAdmin(adminHandler, test)
-	adminToken := adminLogin.Token
 	guests := getGuests(guestHandler, adminToken, test)
 	guest := findGuest(guests)
 	assertEquals(test, guest.Attendance, "accept")
@@ -51,13 +53,14 @@ func TestUpdateRSVP(test *testing.T) {
 	assertEquals(test, guest.PartyList[0].FirstName, "Madison")
 	assertEquals(test, guest.PartyList[0].LastName, "Rossitto")
 
-	deletePartyGuest(guestHandler, guestToken, test)
+	deletePartyMember(guestHandler, guestToken, test)
 	guests = getGuests(guestHandler, adminToken, test)
 	guest = findGuest(guests)
 	assertEquals(test, guest.Attendance, "")
 	assertEquals(test, guest.Email, "")
 	assertEquals(test, guest.Message, "")
 	assertEquals(test, guest.PartySize, 1)
+	deleteGuest(guestHandler, adminToken, test)
 }
 
 func logInGuest(guestHandler *guest.GuestHandler, test *testing.T) guest.LoginResponse {
@@ -76,10 +79,10 @@ func logInGuest(guestHandler *guest.GuestHandler, test *testing.T) guest.LoginRe
 	return login
 }
 
-func addPartyGuest(guestHandler *guest.GuestHandler, token string, test *testing.T) {
+func addPartyMember(guestHandler *guest.GuestHandler, token string, test *testing.T) {
 	response := httptest.NewRecorder()
 	putRequest, err := http.NewRequest(http.MethodPut, "http://localhost:8080/guests/1",
-		strings.NewReader(getGuestWithParty()))
+		strings.NewReader(getUpdatedGuest()))
 	if err != nil {
 		test.Error(err)
 	}
@@ -120,10 +123,10 @@ func getGuests(guestHandler *guest.GuestHandler, token string, test *testing.T)
 	return guests
 }
 
-func deletePartyGuest(guestHandler *guest.GuestHandler, token string, test *testing.T) {
+func deletePartyMember(guestHandler *guest.GuestHandler, token string, test *testing.T) {
 	response := httptest.NewRecorder()
 	putRequest, err := http.NewRequest(http.MethodPut, "http://localhost:8080/guests/1",
-		strings.NewReader(getGuestWithoutParty()))
+		strings.NewReader(getEmptyGuest()))
 	if err != nil {
 		test.Error(err)
 	}
@@ -156,13 +159,13 @@ func getCredentials() string {
 	return "{ \"username\": \"mhunteman\", \"password\": \"password\"}"
 }
 
-func getGuestWithParty() string {
+func getUpdatedGuest() string {
 	return `{"id":1,"firstName":"Michael","lastName":"Hunteman", "attendance":"accept",
 		"email":"mhunteman@cox.net","message":"We'll be there!","partySize":2,
 		"partyList":[{"firstName":"Madison","lastName":"Rossitto"}]}`
 }
 
-func getGuestWithoutParty() string {
+func getEmptyGuest() string {
 	return `{"id":1,"firstName":"Michael","lastName":"Hunteman","attendance":"",
 		"email":"","message":"","partySize":1,"partyList":[]}`
 }
@@ -181,18 +184,19 @@ func TestAddGuest(test *testing.T) {
 
 	adminLogin := logInAdmin(adminHandler, test)
 	adminToken := adminLogin.Token
+	deleteGuest(guestHandler, adminToken, test)
 	guests := getGuests(guestHandler, adminToken, test)
 	assertEquals(test, findGuest(guests), findGuest([]guest.Guest{}))
 	postGuest(guestHandler, adminToken, test)
 
 	guests = getGuests(guestHandler, adminToken, test)
 	guest := findGuest(guests)
-	assertEquals(test, guest.Attendance, "accept")
-	assertEquals(test, guest.Email, "mhunteman@cox.net")
-	assertEquals(test, guest.Message, "We'll be there!")
-	assertEquals(test, guest.PartySize, 2)
-	assertEquals(test, guest.PartyList[0].FirstName, "Madison")
-	assertEquals(test, guest.PartyList[0].LastName, "Rossitto")
+	assertEquals(test, guest.FirstName, "Michael")
+	assertEquals(test, guest.LastName, "Hunteman")
+	assertEquals(test, guest.Attendance, "")
+	assertEquals(test, guest.Email, "")
+	assertEquals(test, guest.Message, "")
+	assertEquals(test, guest.PartySize, 1)
 	deleteGuest(guestHandler, adminToken, test)
 }
 
@@ -210,7 +214,7 @@ func deleteGuest(guestHandler *guest.GuestHandler, token string, test *testing.T
 func postGuest(guestHandler *guest.GuestHandler, token string, test *testing.T) {
 	response := httptest.NewRecorder()
 	putRequest, err := http.NewRequest(http.MethodPost, "http://localhost:8080/guests/",
-		strings.NewReader(getGuestWithParty()))
+		strings.NewReader(getEmptyGuest()))
 	if err != nil {
 		test.Error(err)
 	}
-- 
cgit v1.2.3