chore: fix failing tests

Real-Dev-Squad · Oct 4, 2024 · 8f58692 · 8f58692
1 parent 5a5b49b
commit 8f58692
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/tests/integration/url_test.go b/tests/integration/url_test.go
@@ -15,6 +15,12 @@ import (
 func (suite *AppTestSuite) TestCreateTinyURLSuccess() {
 	// Setup the router and route for creating a tiny URL
 	router := gin.Default()
+
+	router.Use(func(ctx *gin.Context) {
+		ctx.Set("userID", int64(1))
+		ctx.Next()
+	})
+
 	router.POST("/v1/tinyurl", func(ctx *gin.Context) {
 		controller.CreateTinyURL(ctx, suite.db)
 	})
@@ -81,14 +87,19 @@ func (suite *AppTestSuite) TestCreateTinyURLEmptyOriginalURL() {
 // TestCreateTinyURLCustomShortURL tests the creation of a tiny URL with a custom short URL and expects a successful response.
 func (suite *AppTestSuite) TestCreateTinyURLCustomShortURL() {
 	router := gin.Default()
+
+	router.Use(func(ctx *gin.Context) {
+		ctx.Set("userID", int64(1))
+		ctx.Next()
+	})
+
 	router.POST("/v1/tinyurl", func(ctx *gin.Context) {
 		controller.CreateTinyURL(ctx, suite.db)
 	})
 
 	requestBody := map[string]interface{}{
 		"OriginalUrl": "https://example.com",
 		"ShortUrl":    "short",
-		"UserId":      1,
 	}
 	requestJSON, _ := json.Marshal(requestBody)
 	req, _ := http.NewRequest("POST", "/v1/tinyurl", bytes.NewBuffer(requestJSON))
@@ -102,14 +113,19 @@ func (suite *AppTestSuite) TestCreateTinyURLCustomShortURL() {
 
 func (suite *AppTestSuite) TestCreateTinyURLCustomShortURLExists() {
 	router := gin.Default()
+
+	router.Use(func(ctx *gin.Context) {
+		ctx.Set("userID", int64(1))
+		ctx.Next()
+	})
+
 	router.POST("/v1/tinyurl", func(ctx *gin.Context) {
 		controller.CreateTinyURL(ctx, suite.db)
 	})
 
 	requestBody := map[string]interface{}{
 		"OriginalUrl": "https://rds.com",
 		"ShortUrl":    "37fff",
-		"UserId":      1,
 	}
 	requestJSON, _ := json.Marshal(requestBody)
 	req, _ := http.NewRequest("POST", "/v1/tinyurl", bytes.NewBuffer(requestJSON))
@@ -123,6 +139,12 @@ func (suite *AppTestSuite) TestCreateTinyURLCustomShortURLExists() {
 
 func (suite *AppTestSuite) TestCreateTinyURLExistingOriginalURL() {
     router := gin.Default()
+
+	router.Use(func(ctx *gin.Context) {
+		ctx.Set("userID", int64(1))
+		ctx.Next()
+	})
+
     router.POST("/v1/tinyurl", func(ctx *gin.Context) {
         controller.CreateTinyURL(ctx, suite.db)
     })
@@ -131,7 +153,6 @@ func (suite *AppTestSuite) TestCreateTinyURLExistingOriginalURL() {
 
     requestBody := map[string]interface{}{
         "OriginalUrl": existingOriginalURL,
-        "UserId":      1,
     }
     requestJSON, _ := json.Marshal(requestBody)
     req, _ := http.NewRequest("POST", "/v1/tinyurl", bytes.NewBuffer(requestJSON))