chore: update test

.github/workflows/go.yml

@@ -5,24 +5,23 @@ name: Go
 
 on:
   push:
-    branches: [ "main" ]
+    branches: ["main"]
   pull_request:
-    branches: [ "main" ]
+    branches: ["main"]
 
 jobs:
-
   build:
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v3
+      - uses: actions/checkout@v3
 
-    - name: Set up Go
-      uses: actions/setup-go@v4
-      with:
-        go-version: '1.20'
+      - name: Set up Go
+        uses: actions/setup-go@v4
+        with:
+          go-version: "1.20"
 
-    - name: Build
-      run: go build -v ./...
+      - name: Build
+        run: go build -v ./...
 
-    - name: Test
-      run: go test -v ./...
+      - name: Test
+        run: go test -v ./...

internal/api/utils.go

@@ -10,7 +10,12 @@ type JSONErrorResponse struct {
 	Message string `json:"message"`
 }
 
+// GetBearerTokenFromHeaders retrieves the bearer token from the headers of a fiber.Ctx object.
 func GetBearerTokenFromHeaders(c *fiber.Ctx) string {
+	if c == nil {
+		return ""
+	}
+
 	authHeader := c.Get("Authorization")
 	if authHeader == "" {
 		return ""

internal/api/utils_test.go

@@ -0,0 +1,52 @@
+package api
+
+import (
+	"testing"
+
+	"github.com/gofiber/fiber/v2"
+	"github.com/valyala/fasthttp"
+)
+
+func TestGetBearerTokenFromHeaders(t *testing.T) {
+	t.Parallel()
+	app := fiber.New()
+
+	// Test with nil context
+	t.Run("Nil Context", func(t *testing.T) {
+		token := GetBearerTokenFromHeaders(nil)
+		if token != "" {
+			t.Errorf("Expected empty token for nil context, but got %s", token)
+		}
+	})
+
+	// Test with missing Authorization header
+	t.Run("Missing Authorization Header", func(t *testing.T) {
+		ctx := app.AcquireCtx(&fasthttp.RequestCtx{})
+		token := GetBearerTokenFromHeaders(ctx)
+		if token != "" {
+			t.Errorf("Expected empty token for missing Authorization header, but got %s", token)
+		}
+	})
+
+	// Test with improperly formatted Authorization header
+	t.Run("Improperly Formatted Authorization Header", func(t *testing.T) {
+		ctx := app.AcquireCtx(&fasthttp.RequestCtx{})
+		// ctx.Set("Authorization", "InvalidHeaderFormat")
+		ctx.Request().Header.Set("Authorization", "InvalidHeaderFormat")
+		token := GetBearerTokenFromHeaders(ctx)
+		if token != "" {
+			t.Errorf("Expected empty token for improperly formatted Authorization header, but got %s", token)
+		}
+	})
+
+	// Test with valid Authorization header
+	t.Run("Valid Authorization Header", func(t *testing.T) {
+		ctx := app.AcquireCtx(&fasthttp.RequestCtx{})
+		ctx.Request().Header.Set("Authorization", "Bearer YourValidTokenHere")
+		token := GetBearerTokenFromHeaders(ctx)
+		expectedToken := "YourValidTokenHere"
+		if token != expectedToken {
+			t.Errorf("Expected token: %s, but got: %s", expectedToken, token)
+		}
+	})
+}