clone from danieldin95

pkg/libol/utils_test.go

@@ -0,0 +1,95 @@
+package libol
+
+import (
+	"bytes"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestPrettyTime(t *testing.T) {
+	var s string
+
+	s = PrettyTime(59)
+	assert.Equal(t, "0m59s", s, "be the same.")
+	s = PrettyTime(60 + 59)
+	assert.Equal(t, "1m59s", s, "be the same.")
+
+	s = PrettyTime(60*2 + 8)
+	assert.Equal(t, "2m8s", s, "be the same.")
+
+	s = PrettyTime(3600 + 1)
+	assert.Equal(t, "1h0m", s, "be the same.")
+
+	s = PrettyTime(3600 + 61)
+	assert.Equal(t, "1h1m", s, "be the same.")
+
+	s = PrettyTime(3600 + 60*59)
+	assert.Equal(t, "1h59m", s, "be the same.")
+
+	s = PrettyTime(3600*23 + 60*59)
+	assert.Equal(t, "23h59m", s, "be the same.")
+
+	s = PrettyTime(86400)
+	assert.Equal(t, "1d0h", s, "be the same.")
+
+	s = PrettyTime(86400 + 3600*5 + 59)
+	assert.Equal(t, "1d5h", s, "be the same.")
+
+	s = PrettyTime(86400 + 3600*23 + 59)
+	assert.Equal(t, "1d23h", s, "be the same.")
+}
+
+func TestPrettyBytes(t *testing.T) {
+	var s string
+
+	s = PrettyBytes(1023)
+	assert.Equal(t, "1023B", s, "be the same.")
+	s = PrettyBytes(1024 + 1023)
+	assert.Equal(t, "1.99K", s, "be the same.")
+
+	s = PrettyBytes(1024*2 + 8)
+	assert.Equal(t, "2.00K", s, "be the same.")
+
+	s = PrettyBytes(1024*2 + 1023)
+	assert.Equal(t, "2.99K", s, "be the same.")
+
+	s = PrettyBytes(1024*1024 + 1)
+	assert.Equal(t, "1.00M", s, "be the same.")
+
+	s = PrettyBytes(1024*1024 + 1024*256 + 1023)
+	assert.Equal(t, "1.25M", s, "be the same.")
+
+	s = PrettyBytes(1024*1024 + 1024*1023)
+	assert.Equal(t, "1.99M", s, "be the same.")
+
+	s = PrettyBytes(1024 * 1024 * 1024)
+	assert.Equal(t, "1.00G", s, "be the same.")
+
+	s = PrettyBytes(1024*1024*1024 + 1024*1024*5 + 59)
+	assert.Equal(t, "1.00G", s, "be the same.")
+
+	s = PrettyBytes(1024*1024*1024 + 1024*1024*512 + 59)
+	assert.Equal(t, "1.50G", s, "be the same.")
+}
+
+func TestScanPure(t *testing.T) {
+	buff := bytes.NewBuffer([]byte(`// hi`))
+	_, err := ScanAnn(buff)
+	assert.Equal(t, nil, err, "be the same.")
+	buff = bytes.NewBuffer([]byte(`// hi
+			you are`))
+	data, err := ScanAnn(buff)
+	assert.Equal(t, string(data), "\t\t\tyou are", "be the same.")
+	buff = bytes.NewBuffer([]byte(`// hi
+			you are
+			// you are
+            //`))
+	data, err = ScanAnn(buff)
+	assert.Equal(t, string(data), "\t\t\tyou are", "be the same.")
+	buff = bytes.NewBuffer([]byte(`// hi
+			you are
+			// you are
+			/`))
+	data, err = ScanAnn(buff)
+	assert.Equal(t, string(data), "\t\t\tyou are\t\t\t/", "be the same.")
+}