84 lines
2.7 KiB
Go
84 lines
2.7 KiB
Go
package controllers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Smail/config"
|
|
appcontext "git.pcux.cn/xuans/Smail/utils/context"
|
|
)
|
|
|
|
func TestSetupReturnsJSONForUnsupportedOperation(t *testing.T) {
|
|
request := httptest.NewRequest("POST", "/api/v1/setup", strings.NewReader(`{"step":"unknown","action":"unknown"}`))
|
|
response := httptest.NewRecorder()
|
|
|
|
Setup(&appcontext.Context{}, response, request)
|
|
|
|
if response.Code != 200 {
|
|
t.Fatalf("setup status = %d, want 200", response.Code)
|
|
}
|
|
if contentType := response.Header().Get("Content-Type"); !strings.HasPrefix(contentType, "application/json") {
|
|
t.Fatalf("setup content type = %q, want JSON", contentType)
|
|
}
|
|
var payload struct {
|
|
ErrorNo int `json:"errorNo"`
|
|
ErrorMsg string `json:"errorMsg"`
|
|
}
|
|
if err := json.Unmarshal(response.Body.Bytes(), &payload); err != nil {
|
|
t.Fatalf("decode setup response: %v", err)
|
|
}
|
|
if payload.ErrorNo != 100 || payload.ErrorMsg != "不支持的初始化操作" {
|
|
t.Fatalf("unexpected setup response: %+v", payload)
|
|
}
|
|
}
|
|
|
|
func TestBuildSetupSSLResponseUsesConfiguredProxyPorts(t *testing.T) {
|
|
cfg := &config.Config{HttpPort: 9001}
|
|
response := buildSetupSSLResponse(
|
|
config.SSLTypeUser,
|
|
cfg,
|
|
config.HTTPConfig{
|
|
Port: 9001,
|
|
HTTPSPort: 9002,
|
|
TLSMode: config.HTTPSTLSModeProxy,
|
|
HTTPSEnabled: false,
|
|
},
|
|
"example.com",
|
|
true,
|
|
)
|
|
|
|
if response.HTTPPort != 9001 || response.HTTPSPort != 9002 {
|
|
t.Fatalf("unexpected configured ports: http=%d https=%d", response.HTTPPort, response.HTTPSPort)
|
|
}
|
|
if response.TLSMode != config.HTTPSTLSModeProxy || response.HTTPSEnabled {
|
|
t.Fatalf("unexpected proxy settings: mode=%q enabled=%v", response.TLSMode, response.HTTPSEnabled)
|
|
}
|
|
if response.WebDomain != "example.com" || !response.Ready {
|
|
t.Fatalf("unexpected setup response identity: domain=%q ready=%v", response.WebDomain, response.Ready)
|
|
}
|
|
}
|
|
|
|
func TestBuildSetupSSLResponseSeparatesPublicPort(t *testing.T) {
|
|
response := buildSetupSSLResponse("1", nil, config.HTTPConfig{
|
|
Port: 8001, HTTPSPort: 8002, HTTPSEnabled: true, PublicHTTPSPort: 443,
|
|
}, "mail.example.com", true)
|
|
if response.HTTPSPort != 8002 || response.HTTPPort != 8001 || response.PublicHTTPSPort != 443 {
|
|
t.Fatalf("初始化响应混淆公网与内部端口：%+v", response)
|
|
}
|
|
}
|
|
|
|
func TestBuildSetupSSLResponseDefaultsHTTPSPort(t *testing.T) {
|
|
response := buildSetupSSLResponse(
|
|
config.SSLTypeAutoHTTP,
|
|
&config.Config{HttpPort: 9001},
|
|
config.HTTPConfig{Port: 9001},
|
|
"mail.example.com",
|
|
false,
|
|
)
|
|
if response.HTTPPort != 9001 || response.HTTPSPort != 443 {
|
|
t.Fatalf("unexpected default ports: http=%d https=%d", response.HTTPPort, response.HTTPSPort)
|
|
}
|
|
}
|