mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-06-29 23:41:12 +08:00
5f6ebc97c6
### What problem does this PR solve? As pic shows ### Type of change - [x] New Feature (non-breaking change which adds functionality)
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
package router
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"ragflow/internal/handler"
|
|
)
|
|
|
|
func TestConnectorRoutesDoNotConflictWithOAuthCallbacks(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
engine := gin.New()
|
|
engine.GET("/api/v1/connectors/gmail/oauth/web/callback", func(c *gin.Context) {
|
|
c.String(http.StatusOK, "gmail")
|
|
})
|
|
engine.GET("/api/v1/connectors/google-drive/oauth/web/callback", func(c *gin.Context) {
|
|
c.String(http.StatusOK, "google-drive")
|
|
})
|
|
|
|
connectors := engine.Group("/api/v1/connectors")
|
|
connectors.GET("/:connector_id", func(c *gin.Context) {
|
|
c.String(http.StatusOK, c.Param("connector_id"))
|
|
})
|
|
connectors.GET("/:connector_id/logs", func(c *gin.Context) {
|
|
c.String(http.StatusOK, c.Param("connector_id"))
|
|
})
|
|
|
|
tests := []struct {
|
|
name string
|
|
path string
|
|
want string
|
|
}{
|
|
{
|
|
name: "gmail callback",
|
|
path: "/api/v1/connectors/gmail/oauth/web/callback",
|
|
want: "gmail",
|
|
},
|
|
{
|
|
name: "google drive callback",
|
|
path: "/api/v1/connectors/google-drive/oauth/web/callback",
|
|
want: "google-drive",
|
|
},
|
|
{
|
|
name: "connector detail",
|
|
path: "/api/v1/connectors/connector-1",
|
|
want: "connector-1",
|
|
},
|
|
{
|
|
name: "connector logs",
|
|
path: "/api/v1/connectors/connector-1/logs",
|
|
want: "connector-1",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
resp := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, tt.path, nil)
|
|
engine.ServeHTTP(resp, req)
|
|
|
|
if resp.Code != http.StatusOK {
|
|
t.Fatalf("status=%d body=%s", resp.Code, resp.Body.String())
|
|
}
|
|
if resp.Body.String() != tt.want {
|
|
t.Fatalf("body=%q want=%q", resp.Body.String(), tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRouterSetupRegistersUpdateDatasetRoute(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
engine := gin.New()
|
|
r := &Router{
|
|
authHandler: handler.NewAuthHandler(),
|
|
datasetsHandler: handler.NewDatasetsHandler(nil, nil),
|
|
}
|
|
r.Setup(engine)
|
|
|
|
resp := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodPut, "/api/v1/datasets/dataset-1", nil)
|
|
engine.ServeHTTP(resp, req)
|
|
|
|
if resp.Code == http.StatusNotFound {
|
|
t.Fatalf("PUT /api/v1/datasets/:dataset_id returned 404; UpdateDataset route is not registered")
|
|
}
|
|
if resp.Code != http.StatusUnauthorized {
|
|
t.Fatalf("status=%d body=%s; want auth middleware to handle registered UpdateDataset route", resp.Code, resp.Body.String())
|
|
}
|
|
}
|