Hunter0x7c7
2022-08-11 b8230139fb40edea387617b6accd8371e37eda58
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package units_test
 
import (
    "testing"
 
    "github.com/v2fly/v2ray-core/v5/common/units"
)
 
func TestByteSizes(t *testing.T) {
    size := units.ByteSize(0)
    assertSizeString(t, size, "0")
    size++
    assertSizeValue(t,
        assertSizeString(t, size, "1.00B"),
        size,
    )
    size <<= 10
    assertSizeValue(t,
        assertSizeString(t, size, "1.00KB"),
        size,
    )
    size <<= 10
    assertSizeValue(t,
        assertSizeString(t, size, "1.00MB"),
        size,
    )
    size <<= 10
    assertSizeValue(t,
        assertSizeString(t, size, "1.00GB"),
        size,
    )
    size <<= 10
    assertSizeValue(t,
        assertSizeString(t, size, "1.00TB"),
        size,
    )
    size <<= 10
    assertSizeValue(t,
        assertSizeString(t, size, "1.00PB"),
        size,
    )
    size <<= 10
    assertSizeValue(t,
        assertSizeString(t, size, "1.00EB"),
        size,
    )
}
 
func assertSizeValue(t *testing.T, size string, expected units.ByteSize) {
    actual := units.ByteSize(0)
    err := actual.Parse(size)
    if err != nil {
        t.Error(err)
    }
    if actual != expected {
        t.Errorf("expect %s, but got %s", expected, actual)
    }
}
 
func assertSizeString(t *testing.T, size units.ByteSize, expected string) string {
    actual := size.String()
    if actual != expected {
        t.Errorf("expect %s, but got %s", expected, actual)
    }
    return expected
}