mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
2a0ebd16fa
- 修复 ClickHouse Batch Writer 与 RiskControl LogWriter 优雅停机，确保退出前队列数据正确刷盘。 - 修复 OpenFlare 系统配置参数全局变量并发读写的 Data Race 冲突，在读取配置时引入读锁保护。 - 修复 upload 模块的 API 错误响应格式，使用 response.Abort* 代替原始的 c.AbortWithStatus 与 c.JSON，保证全局信封格式统一。 - 修复 pkg/utils/network 的 isPrivateIPv4 以使用标准库 net.IP.IsPrivate() 校验，修复 format 的 Bytes2Size 边界，修改大小单位因子变量为只读常量。 - 重构 upload 文件服务与路由器测试，使用 t.TempDir() 代替硬编码的 uploads 相对路径写入和删除，解决测试目录文件污染问题。
50 lines
927 B
Go
50 lines
927 B
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestBytes2Size(t *testing.T) {
|
|
tests := []struct {
|
|
input int64
|
|
expected string
|
|
}{
|
|
{0, "0 B"},
|
|
{500, "500 B"},
|
|
{1023, "1023 B"},
|
|
{1024, "1 KB"},
|
|
{2048, "2 KB"},
|
|
{1024 * 1024, "1 MB"},
|
|
{1024 * 1024 * 1024, "1.00 GB"},
|
|
{1024 * 1024 * 1024 * 2, "2.00 GB"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
result := Bytes2Size(tt.input)
|
|
if result != tt.expected {
|
|
t.Errorf("Bytes2Size(%d) = %q, expected %q", tt.input, result, tt.expected)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSeconds2Time(t *testing.T) {
|
|
tests := []struct {
|
|
input int
|
|
expected string
|
|
}{
|
|
{0, "0 秒"},
|
|
{30, "30 秒"},
|
|
{60, "1 分钟 0 秒"},
|
|
{125, "2 分钟 5 秒"},
|
|
{3600, "1 小时 0 秒"},
|
|
{86400, "1 天 0 秒"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
result := Seconds2Time(tt.input)
|
|
if result != tt.expected {
|
|
t.Errorf("Seconds2Time(%d) = %q, expected %q", tt.input, result, tt.expected)
|
|
}
|
|
}
|
|
}
|