diff options
Diffstat (limited to 'server/test')
| -rw-r--r-- | server/test/guest_test.go | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/server/test/guest_test.go b/server/test/guest_test.go index b7d1aae..09476d1 100644 --- a/server/test/guest_test.go +++ b/server/test/guest_test.go @@ -7,7 +7,6 @@ import ( "log" "net/http" "net/http/httptest" - "os" "strings" "testing" @@ -17,14 +16,16 @@ import ( ) var ( - user = os.Getenv("USER") - password = os.Getenv("PASS") + user = "postgres" + password = "pass" host = "localhost" port = "5432" database = "wedding" ) func TestUpdateRSVP(test *testing.T) { + test.Setenv("GUEST_KEY", "/Users/michael/secrets/guest.pem") + test.Setenv("ADMIN_KEY", "/Users/michael/secrets/admin.pem") databasePool, err := pgxpool.New(context.Background(), fmt.Sprintf("postgres://%s:%s@%s:%s/%s", user, password, host, port, database)) if err != nil { @@ -93,6 +94,8 @@ func addPartyMember(guestHandler *guest.GuestHandler, token string, test *testin } func logInAdmin(adminHandler *admin.AdminHandler, test *testing.T) admin.Login { + test.Setenv("GUEST_KEY", "/Users/michael/secrets/guest.pem") + test.Setenv("ADMIN_KEY", "/Users/michael/secrets/admin.pem") response := httptest.NewRecorder() loginRequest, err := http.NewRequest(http.MethodPost, fmt.Sprintf("http://%s:8080/api/admin/login", host), strings.NewReader(getCredentials())) @@ -158,8 +161,7 @@ func getName() string { } func getCredentials() string { - return fmt.Sprintf("{ \"username\": \"mhunteman\", \"password\": \"%s\" }", - os.Getenv("PASS")) + return "{ \"username\": \"mhunteman\", \"password\": \"pass\" }" } func getUpdatedGuest() string { @@ -176,6 +178,8 @@ func getEmptyGuest() string { } func TestAddGuest(test *testing.T) { + test.Setenv("GUEST_KEY", "/Users/michael/secrets/guest.pem") + test.Setenv("ADMIN_KEY", "/Users/michael/secrets/admin.pem") databasePool, err := pgxpool.New(context.Background(), fmt.Sprintf("postgres://%s:%s@%s:%s/%s", user, password, host, port, database)) if err != nil { @@ -230,6 +234,8 @@ func postGuest(guestHandler *guest.GuestHandler, token string, test *testing.T) } func TestInvalidGuest(test *testing.T) { + test.Setenv("GUEST_KEY", "/Users/michael/secrets/guest.pem") + test.Setenv("ADMIN_KEY", "/Users/michael/secrets/admin.pem") databasePool, err := pgxpool.New(context.Background(), fmt.Sprintf("postgres://%s:%s@%s:%s/%s", user, password, host, port, database)) if err != nil { |