Refactoring draw test
Vladimir Bauer
8 years ago
| 179 | 179 | |
| 180 | 180 | prependWs := newWidthSyncer(nil, 1, 0) |
| 181 | 181 | appendWs := newWidthSyncer(nil, 1, 0) |
| 182 | var temp bytes.Buffer | |
| 182 | var tmpBuf bytes.Buffer | |
| 183 | 183 | for termWidth, cases := range testSuite { |
| 184 | 184 | for name, tc := range cases { |
| 185 | 185 | s := newTestState() |
| 189 | 189 | if tc.barRefill != nil { |
| 190 | 190 | s.refill = tc.barRefill |
| 191 | 191 | } |
| 192 | temp.Reset() | |
| 193 | temp.ReadFrom(s.draw(termWidth, prependWs, appendWs)) | |
| 194 | got := temp.String() | |
| 195 | if got != tc.want+"\n" { | |
| 196 | t.Errorf("termWidth %d; %s: want: %s %d, got: %s %d\n", termWidth, name, tc.want+"\n", len(tc.want), got, len(got)) | |
| 192 | tmpBuf.Reset() | |
| 193 | tmpBuf.ReadFrom(s.draw(termWidth, prependWs, appendWs)) | |
| 194 | got := tmpBuf.String() | |
| 195 | want := tc.want + "\n" | |
| 196 | if got != want { | |
| 197 | t.Errorf("termWidth %d; %s: want: %s %d, got: %s %d\n", termWidth, name, want, len(want), got, len(got)) | |
| 197 | 198 | } |
| 198 | 199 | } |
| 199 | 200 | } |