Codebase list golang-github-vbauerster-mpb / 8d080a5 draw_test.go
8d080a5

Tree @8d080a5 (Download .tar.gz)

draw_test.go @8d080a5 — raw · history · blame 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
package mpb

import (
	"bytes"
	"testing"
)

func TestFillBar(t *testing.T) {
	tests := []struct {
		termWidth int
		barWidth  int
		total     int64
		current   int64
		barRefill *refill
		want      string
	}{
		{
			termWidth: 2,
			barWidth:  100,
			want:      "[]",
		},
		{
			termWidth: 3,
			barWidth:  100,
			total:     100,
			current:   20,
			want:      "[-]",
		},
		{
			termWidth: 5,
			barWidth:  100,
			total:     100,
			current:   20,
			want:      "[>--]",
		},
		{
			termWidth: 6,
			barWidth:  100,
			total:     100,
			current:   20,
			want:      "[>---]",
		},
		{
			termWidth: 20,
			barWidth:  100,
			total:     100,
			current:   20,
			want:      "[===>--------------]",
		},
		{
			termWidth: 50,
			barWidth:  100,
			total:     100,
			current:   20,
			want:      "[=========>--------------------------------------]",
		},
		{
			termWidth: 100,
			barWidth:  100,
			total:     100,
			current:   0,
			want:      "[--------------------------------------------------------------------------------------------------]",
		},
		{
			termWidth: 100,
			barWidth:  100,
			total:     100,
			current:   1,
			want:      "[>-------------------------------------------------------------------------------------------------]",
		},
		{
			termWidth: 100,
			barWidth:  100,
			total:     100,
			current:   40,
			want:      "[======================================>-----------------------------------------------------------]",
		},
		{
			termWidth: 100,
			barWidth:  100,
			total:     100,
			current:   40,
			barRefill: &refill{'+', 32},
			want:      "[+++++++++++++++++++++++++++++++=======>-----------------------------------------------------------]",
		},
		{
			termWidth: 100,
			barWidth:  100,
			total:     100,
			current:   99,
			want:      "[================================================================================================>-]",
		},
		{
			termWidth: 100,
			barWidth:  100,
			total:     100,
			current:   100,
			want:      "[==================================================================================================]",
		},
	}

	prependWs := newWidthSync(nil, 1, 0)
	appendWs := newWidthSync(nil, 1, 0)
	for _, test := range tests {
		s := newTestState()
		s.width = test.barWidth
		s.total = test.total
		s.current = test.current
		if test.barRefill != nil {
			s.refill = test.barRefill
		}
		s.draw(test.termWidth, prependWs, appendWs)
		got := s.bufB.String()
		if got != test.want {
			t.Errorf("Want: %q, Got: %q\n", test.want, got)
		}
	}
}

func newTestState() *bState {
	s := &bState{
		trimLeftSpace:  true,
		trimRightSpace: true,
		bufP:           new(bytes.Buffer),
		bufB:           new(bytes.Buffer),
		bufA:           new(bytes.Buffer),
	}
	s.updateFormat("[=>-]")
	return s
}