You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
601 B
Go
32 lines
601 B
Go
4 years ago
|
package filex
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"gopkg.in/cheggaaa/pb.v1"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestProgressScanner(t *testing.T) {
|
||
|
|
const text = "hello, world"
|
||
|
|
bar := pb.New(100)
|
||
|
|
var builder strings.Builder
|
||
|
|
builder.WriteString(text)
|
||
|
|
scanner := NewProgressScanner(&mockedScanner{builder: &builder}, bar)
|
||
|
|
assert.True(t, scanner.Scan())
|
||
|
|
assert.Equal(t, text, scanner.Text())
|
||
|
|
}
|
||
|
|
|
||
|
|
type mockedScanner struct {
|
||
|
|
builder *strings.Builder
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockedScanner) Scan() bool {
|
||
|
|
return s.builder.Len() > 0
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *mockedScanner) Text() string {
|
||
|
|
return s.builder.String()
|
||
|
|
}
|