Use decimal prefixes for counts

Fixes #5126
2021-04-07 12:10:31 +02:00 · 2021-04-07 12:10:31 +02:00 · 08a2df51be
commit 08a2df51be
parent 2925e1384c
3 changed files with 337 additions and 1 deletions
--- a/fs/countsuffix_test.go
+++ b/fs/countsuffix_test.go
@ -0,0 +1,147 @@
+package fs
+
+import (
+	"encoding/json"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+// Check it satisfies the interface
+var _ flagger = (*CountSuffix)(nil)
+
+func TestCountSuffixString(t *testing.T) {
+	for _, test := range []struct {
+		in   float64
+		want string
+	}{
+		{0, "0"},
+		{102, "102"},
+		{1000, "1k"},
+		{1000 * 1000, "1M"},
+		{1000 * 1000 * 1000, "1G"},
+		{10 * 1000 * 1000 * 1000, "10G"},
+		{10.1 * 1000 * 1000 * 1000, "10.100G"},
+		{-1, "off"},
+		{-100, "off"},
+	} {
+		ss := CountSuffix(test.in)
+		got := ss.String()
+		assert.Equal(t, test.want, got)
+	}
+}
+
+func TestCountSuffixUnit(t *testing.T) {
+	for _, test := range []struct {
+		in   float64
+		want string
+	}{
+		{0, "0 Byte"},
+		{102, "102 Byte"},
+		{1000, "1 kByte"},
+		{1000 * 1000, "1 MByte"},
+		{1000 * 1000 * 1000, "1 GByte"},
+		{10 * 1000 * 1000 * 1000, "10 GByte"},
+		{10.1 * 1000 * 1000 * 1000, "10.100 GByte"},
+		{10 * 1000 * 1000 * 1000 * 1000, "10 TByte"},
+		{10 * 1000 * 1000 * 1000 * 1000 * 1000, "10 PByte"},
+		{1 * 1000 * 1000 * 1000 * 1000 * 1000 * 1000, "1 EByte"},
+		{-1, "off"},
+		{-100, "off"},
+	} {
+		ss := CountSuffix(test.in)
+		got := ss.Unit("Byte")
+		assert.Equal(t, test.want, got)
+	}
+}
+
+func TestCountSuffixSet(t *testing.T) {
+	for _, test := range []struct {
+		in   string
+		want int64
+		err  bool
+	}{
+		{"0", 0, false},
+		{"1b", 1, false},
+		{"100B", 100, false},
+		{"0.1k", 100, false},
+		{"0.1", 100, false},
+		{"1K", 1000, false},
+		{"1k", 1000, false},
+		{"1KB", 1000, false},
+		{"1kB", 1000, false},
+		{"1kb", 1000, false},
+		{"1", 1000, false},
+		{"2.5", 1000 * 2.5, false},
+		{"1M", 1000 * 1000, false},
+		{"1MB", 1000 * 1000, false},
+		{"1.g", 1000 * 1000 * 1000, false},
+		{"10G", 10 * 1000 * 1000 * 1000, false},
+		{"10T", 10 * 1000 * 1000 * 1000 * 1000, false},
+		{"10P", 10 * 1000 * 1000 * 1000 * 1000 * 1000, false},
+		{"off", -1, false},
+		{"OFF", -1, false},
+		{"", 0, true},
+		{"1q", 0, true},
+		{"1.q", 0, true},
+		{"1q", 0, true},
+		{"-1K", 0, true},
+		{"1i", 0, true},
+		{"1iB", 0, true},
+		{"1Ki", 0, true},
+		{"1KiB", 0, true},
+	} {
+		ss := CountSuffix(0)
+		err := ss.Set(test.in)
+		if test.err {
+			require.Error(t, err, test.in)
+		} else {
+			require.NoError(t, err, test.in)
+		}
+		assert.Equal(t, test.want, int64(ss))
+	}
+}
+
+func TestCountSuffixScan(t *testing.T) {
+	var v CountSuffix
+	n, err := fmt.Sscan(" 17M ", &v)
+	require.NoError(t, err)
+	assert.Equal(t, 1, n)
+	assert.Equal(t, CountSuffix(17000000), v)
+}
+
+func TestCountSuffixUnmarshalJSON(t *testing.T) {
+	for _, test := range []struct {
+		in   string
+		want int64
+		err  bool
+	}{
+		{`"0"`, 0, false},
+		{`"102B"`, 102, false},
+		{`"1K"`, 1000, false},
+		{`"2.5"`, 1000 * 2.5, false},
+		{`"1M"`, 1000 * 1000, false},
+		{`"1.g"`, 1000 * 1000 * 1000, false},
+		{`"10G"`, 10 * 1000 * 1000 * 1000, false},
+		{`"off"`, -1, false},
+		{`""`, 0, true},
+		{`"1q"`, 0, true},
+		{`"-1K"`, 0, true},
+		{`0`, 0, false},
+		{`102`, 102, false},
+		{`1000`, 1000, false},
+		{`1000000000`, 1000000000, false},
+		{`1.1.1`, 0, true},
+	} {
+		var ss CountSuffix
+		err := json.Unmarshal([]byte(test.in), &ss)
+		if test.err {
+			require.Error(t, err, test.in)
+		} else {
+			require.NoError(t, err, test.in)
+		}
+		assert.Equal(t, test.want, int64(ss))
+	}
+}