Skip to content

Commit

Permalink
test fixes
Browse files Browse the repository at this point in the history
Signed-off-by: Dave Lee <dave@gray101.com>
  • Loading branch information
dave-gray101 committed Jun 11, 2024
1 parent 5b24181 commit 7bfc96d
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions middleware/keyauth/keyauth_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,10 @@ func TestPanicOnInvalidConfiguration(t *testing.T) {
})
// We shouldn't even make it this far, but these next two lines prevent authMiddleware from being an unused variable.
app := fiber.New()
defer app.Shutdown()
defer func() { // testing panics, defer block to ensure cleanup
err := app.Shutdown()
require.NoError(t, err)
}()
app.Use(authMiddleware)
})
}
Expand Down Expand Up @@ -207,8 +210,11 @@ func TestMultipleKeyLookup(t *testing.T) {
// construct a second request without proper key
req, err = http.NewRequestWithContext(context.Background(), fiber.MethodGet, "/foo", nil)
require.NoError(t, err)
_, err = app.Test(req, -1)
require.Error(t, err)
res, err = app.Test(req, -1)
require.NoError(t, err)
errBody, err := io.ReadAll(res.Body)
require.NoError(t, err)
require.Equal(t, ErrMissingOrMalformedAPIKey.Error(), string(errBody))
}

func Test_MultipleKeyAuth(t *testing.T) {
Expand Down

0 comments on commit 7bfc96d

Please sign in to comment.