49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package filesystem
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
. "github.com/franela/goblin"
|
|
)
|
|
|
|
func TestProgress(t *testing.T) {
|
|
g := Goblin(t)
|
|
|
|
g.Describe("Progress", func() {
|
|
g.It("properly initializes", func() {
|
|
total := int64(1000)
|
|
p := NewProgress(total)
|
|
g.Assert(p).IsNotNil()
|
|
g.Assert(p.Total()).Equal(total)
|
|
g.Assert(p.Written()).Equal(int64(0))
|
|
})
|
|
|
|
g.It("increments written when Write is called", func() {
|
|
v := []byte("hello")
|
|
p := NewProgress(1000)
|
|
_, err := p.Write(v)
|
|
g.Assert(err).IsNil()
|
|
g.Assert(p.Written()).Equal(int64(len(v)))
|
|
})
|
|
|
|
g.It("renders a progress bar", func() {
|
|
v := bytes.Repeat([]byte{' '}, 100)
|
|
p := NewProgress(1000)
|
|
_, err := p.Write(v)
|
|
g.Assert(err).IsNil()
|
|
g.Assert(p.Written()).Equal(int64(len(v)))
|
|
g.Assert(p.Progress(25)).Equal("[== ] 100 B / 1000 B")
|
|
})
|
|
|
|
g.It("renders a progress bar when written exceeds total", func() {
|
|
v := bytes.Repeat([]byte{' '}, 1001)
|
|
p := NewProgress(1000)
|
|
_, err := p.Write(v)
|
|
g.Assert(err).IsNil()
|
|
g.Assert(p.Written()).Equal(int64(len(v)))
|
|
g.Assert(p.Progress(25)).Equal("[=========================] 1001 B / 1000 B")
|
|
})
|
|
})
|
|
}
|