2eab570d54
- Implement tests for AuthMiddleware to handle cookie and token authentication. - Create tests for the Importer and Manager in the Caddy package. - Enhance AuthService tests with password change and token validation scenarios. - Introduce tests for CertificateService to validate certificate listing and expiry. - Expand LogService tests to cover log querying and pagination. - Add NotificationService tests for creating, listing, and marking notifications as read. - Implement ProxyHostService tests for CRUD operations and unique domain validation. - Create RemoteServerService tests for CRUD operations. - Add UpdateService tests to mock GitHub API responses for version checking. - Introduce UptimeService tests to check host availability and notifications for down hosts.
104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
package handlers_test
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/Wikid82/CaddyProxyManagerPlus/backend/internal/api/handlers"
|
|
"github.com/Wikid82/CaddyProxyManagerPlus/backend/internal/models"
|
|
)
|
|
|
|
func setupRemoteServerTest_New(t *testing.T) (*gin.Engine, *handlers.RemoteServerHandler) {
|
|
db := setupTestDB()
|
|
// Ensure RemoteServer table exists
|
|
db.AutoMigrate(&models.RemoteServer{})
|
|
|
|
handler := handlers.NewRemoteServerHandler(db)
|
|
|
|
r := gin.Default()
|
|
api := r.Group("/api/v1")
|
|
servers := api.Group("/remote-servers")
|
|
servers.GET("", handler.List)
|
|
servers.POST("", handler.Create)
|
|
servers.GET("/:uuid", handler.Get)
|
|
servers.PUT("/:uuid", handler.Update)
|
|
servers.DELETE("/:uuid", handler.Delete)
|
|
servers.POST("/test-connection", handler.TestConnection)
|
|
|
|
return r, handler
|
|
}
|
|
|
|
func TestRemoteServerHandler_FullCRUD(t *testing.T) {
|
|
r, _ := setupRemoteServerTest_New(t)
|
|
|
|
// Create
|
|
rs := models.RemoteServer{
|
|
Name: "Test Server CRUD",
|
|
Host: "192.168.1.100",
|
|
Port: 22,
|
|
Provider: "manual",
|
|
}
|
|
body, _ := json.Marshal(rs)
|
|
req, _ := http.NewRequest("POST", "/api/v1/remote-servers", bytes.NewBuffer(body))
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusCreated, w.Code)
|
|
|
|
var created models.RemoteServer
|
|
err := json.Unmarshal(w.Body.Bytes(), &created)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, rs.Name, created.Name)
|
|
assert.NotEmpty(t, created.UUID)
|
|
|
|
// List
|
|
req, _ = http.NewRequest("GET", "/api/v1/remote-servers", nil)
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
// Get
|
|
req, _ = http.NewRequest("GET", "/api/v1/remote-servers/"+created.UUID, nil)
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
// Update
|
|
created.Name = "Updated Server CRUD"
|
|
body, _ = json.Marshal(created)
|
|
req, _ = http.NewRequest("PUT", "/api/v1/remote-servers/"+created.UUID, bytes.NewBuffer(body))
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusOK, w.Code)
|
|
|
|
// Delete
|
|
req, _ = http.NewRequest("DELETE", "/api/v1/remote-servers/"+created.UUID, nil)
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusNoContent, w.Code)
|
|
|
|
// Create - Invalid JSON
|
|
req, _ = http.NewRequest("POST", "/api/v1/remote-servers", bytes.NewBuffer([]byte("invalid json")))
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusBadRequest, w.Code)
|
|
|
|
// Update - Not Found
|
|
req, _ = http.NewRequest("PUT", "/api/v1/remote-servers/non-existent-uuid", bytes.NewBuffer(body))
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusNotFound, w.Code)
|
|
|
|
// Delete - Not Found
|
|
req, _ = http.NewRequest("DELETE", "/api/v1/remote-servers/non-existent-uuid", nil)
|
|
w = httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
assert.Equal(t, http.StatusNotFound, w.Code)
|
|
}
|