mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-04 21:17:47 +03:00
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package utils
|
|
|
|
import "testing"
|
|
|
|
func TestStringToBps(t *testing.T) {
|
|
type args struct {
|
|
s string
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want uint64
|
|
wantErr bool
|
|
}{
|
|
{"bps", args{"800 bps"}, 100, false},
|
|
{"kbps", args{"800 kbps"}, 100_000, false},
|
|
{"mbps", args{"800 mbps"}, 100_000_000, false},
|
|
{"gbps", args{"800 gbps"}, 100_000_000_000, false},
|
|
{"tbps", args{"800 tbps"}, 100_000_000_000_000, false},
|
|
{"mbps simp", args{"100m"}, 12_500_000, false},
|
|
{"gbps simp upper", args{"2G"}, 250_000_000, false},
|
|
{"invalid 1", args{"damn"}, 0, true},
|
|
{"invalid 2", args{"6444"}, 0, true},
|
|
{"invalid 3", args{"5.4 mbps"}, 0, true},
|
|
{"invalid 4", args{"kbps"}, 0, true},
|
|
{"invalid 5", args{"1234 5678 gbps"}, 0, true},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := StringToBps(tt.args.s)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("StringToBps() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("StringToBps() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|