contrib/gofiber/fiber.v2: improve UserContext test in Middleware for …

…span creation
DataDog · Jan 15, 2025 · bac65b3 · bac65b3
1 parent 1fdf56d
commit bac65b3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/contrib/gofiber/fiber.v2/fiber_test.go b/contrib/gofiber/fiber.v2/fiber_test.go
@@ -198,8 +198,9 @@ func TestUserContext(t *testing.T) {
 		assert.NotEmpty(c.UserContext())
 
 		// checks that the user context still has the information provided before using the middleware
-		_, ok := c.UserContext().Value(fooKey).(string)
+		foo, ok := c.UserContext().Value(fooKey).(string)
 		assert.True(ok)
+		assert.Equal(foo, "bar")
 
 		span, _ := tracer.StartSpanFromContext(c.UserContext(), "http.request")
 		defer span.Finish()