TestAppendPercentage
Vladimir Bauer
9 years ago
| 6 | 6 | "io/ioutil" |
| 7 | 7 | "strings" |
| 8 | 8 | "testing" |
| 9 | "time" | |
| 9 | 10 | |
| 10 | 11 | "github.com/vbauerster/mpb" |
| 11 | 12 | ) |
| 49 | 50 | t.Errorf("Expected: %s, got %s\n", want, out) |
| 50 | 51 | } |
| 51 | 52 | } |
| 53 | ||
| 54 | func TestAppendPercentage(t *testing.T) { | |
| 55 | var buf bytes.Buffer | |
| 56 | p := mpb.New().SetOut(&buf) | |
| 57 | ||
| 58 | bar := p.AddBar(100).TrimLeftSpace().TrimRightSpace(). | |
| 59 | AppendPercentage(0, 0) | |
| 60 | ||
| 61 | for i := 0; i < 100; i++ { | |
| 62 | time.Sleep(10 * time.Millisecond) | |
| 63 | bar.Incr(1) | |
| 64 | } | |
| 65 | ||
| 66 | p.Stop() | |
| 67 | ||
| 68 | bytes := removeLastRune(buf.Bytes()) | |
| 69 | out := string(bytes) | |
| 70 | out = out[strings.LastIndex(out, "]")+1:] | |
| 71 | want := "100 %" | |
| 72 | if out != want { | |
| 73 | t.Errorf("Expected: %s, got %s\n", want, out) | |
| 74 | } | |
| 75 | } | |