Add test

Author: raymondjacobson

api/auth_middleware.go

@@ -18,7 +18,8 @@ const (
 func (app *ApiServer) recoverAuthorityFromSignatureHeaders(c *fiber.Ctx) (int32, string) {
 	message := c.Get(messageHeader)
 	signature := c.Get(signatureHeader)
-
+	fmt.Println("message", message)
+	fmt.Println("signature", signature)
 	if message == "" || signature == "" {
 		return 0, ""
 	}
@@ -60,19 +61,24 @@ func (app *ApiServer) recoverAuthorityFromSignatureHeaders(c *fiber.Ctx) (int32,
 	return userId, walletLower
 }
 
+func (app *ApiServer) getAuthedUserId(c *fiber.Ctx) int32 {
+	return int32(c.Locals("authedUserId").(int32))
+}
+
+func (app *ApiServer) getAuthedWallet(c *fiber.Ctx) string {
+	return c.Locals("authedWallet").(string)
+}
+
 // Middleware to set authedUserId and authedWallet in context
 func (app *ApiServer) authMiddleware(c *fiber.Ctx) error {
+	fmt.Println("authMiddleware")
 	userId, wallet := app.recoverAuthorityFromSignatureHeaders(c)
 	c.Locals("authedUserId", userId)
 	c.Locals("authedWallet", wallet)
 
 	return c.Next()
 }
 
-func (app *ApiServer) getAuthedUserId(c *fiber.Ctx) int32 {
-	return int32(c.Locals("authedUserId").(int32))
-}
-
 // Middleware that asserts authedUserId is valid
 func (app *ApiServer) requiresAuthMiddleware(c *fiber.Ctx) error {
 	authedUserId := app.getAuthedUserId(c)

api/auth_middleware_test.go

@@ -0,0 +1,30 @@
+package api
+
+import (
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestRecoverAuthorityFromSignatureHeaders(t *testing.T) {
+	var userId int32
+	var wallet string
+
+	// Create a dummy endpoint to test the authMiddleware
+	testApp := fiber.New()
+	testApp.Get("/", app.authMiddleware, func(c *fiber.Ctx) error {
+		userId, wallet = app.getAuthedUserId(c), app.getAuthedWallet(c)
+		return c.SendStatus(fiber.StatusOK)
+	})
+
+	req := httptest.NewRequest("GET", "/", nil)
+	req.Header.Set("Encoded-Data-Message", "signature:1744763856446")
+	req.Header.Set("Encoded-Data-Signature", "0xbb202be3a7f3a0aa22c1458ef6a3f2f8360fb86791c7b137e8562df0707825c11fa1db01096efd2abc5e6613c4d1e8d4ae1e2b993abdd555fe270c1b17bff0d21c")
+
+	_, err := testApp.Test(req, -1)
+	assert.NoError(t, err)
+	assert.Equal(t, int32(1), userId)
+	assert.Equal(t, "0x7d273271690538cf855e5b3002a0dd8c154bb060", wallet)
+}

api/fixture_test.go

@@ -15,6 +15,7 @@ var (
 		"user_id":              nil,
 		"handle":               nil,
 		"handle_lc":            nil,
+		"wallet":               nil,
 		"is_current":           true,
 		"is_verified":          false,
 		"created_at":           time.Now(),

api/testdata/user_fixtures.csv

@@ -1,5 +1,5 @@
-user_id,handle,handle_lc,is_deactivated
-1,rayjacobson,rayjacobson,f
-2,stereosteve,stereosteve,f
-3,someseller,someseller,f
-91,badguy,badguy,t
+user_id,handle,handle_lc,is_deactivated,wallet
+1,rayjacobson,rayjacobson,f,0x7d273271690538cf855e5b3002a0dd8c154bb060
+2,stereosteve,stereosteve,f,
+3,someseller,someseller,f,
+91,badguy,badguy,t,