mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 01:30:25 +08:00
32 lines
601 B
Go
32 lines
601 B
Go
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()
|
|
}
|