Files
Prism/internal/mihomoapi/traffic_test.go
T
rose_cat707 c26b06f1cc
CI / test (push) Successful in 1m11s
CI / docker (push) Failing after 17s
feat: Prism HTTP/SOCKS5 代理网关及管理台
Go 控制面(SQLite + REST API)嵌入 Mihomo 数据面,Vue 3 多语言 Web 管理台。
含订阅/节点/规则/出站/监控/日志、OpenAPI、API Key、Gitea Actions CI 与开源文档;
CI 使用 ubuntu-latest runner,Go/Docker 步骤走国内镜像与 shell,避免 GitHub 下载超时。

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-06-23 14:07:40 +08:00

36 lines
855 B
Go

package mihomoapi
import (
"context"
"encoding/json"
"net/http"
"net/http/httptest"
"testing"
"time"
)
func TestTrafficRatesReadsFirstFrame(t *testing.T) {
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
flusher, _ := w.(http.Flusher)
_ = json.NewEncoder(w).Encode(TrafficRates{Up: 100, Down: 200})
flusher.Flush()
time.Sleep(2 * time.Second)
_ = json.NewEncoder(w).Encode(TrafficRates{Up: 300, Down: 400})
flusher.Flush()
}))
defer srv.Close()
client := New(srv.URL, "")
ctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)
defer cancel()
rates, err := client.TrafficRates(ctx)
if err != nil {
t.Fatalf("TrafficRates: %v", err)
}
if rates.Up != 100 || rates.Down != 200 {
t.Fatalf("got %+v", rates)
}
}