blob: 65641dad0e862b8b0aebef049ffe89b0a3988760 [file] [log] [blame]
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package tabwriter
import (
"io";
"os";
"testing";
)
type buffer struct {
a []byte;
}
func (b *buffer) init(n int) {
b.a = make([]byte, n)[0 : 0];
}
func (b *buffer) clear() {
b.a = b.a[0 : 0];
}
func (b *buffer) Write(buf []byte) (written int, err os.Error) {
n := len(b.a);
m := len(buf);
if n + m <= cap(b.a) {
b.a = b.a[0 : n + m];
for i := 0; i < m; i++ {
b.a[n+i] = buf[i];
}
} else {
panicln("buffer too small", n, m, cap(b.a));
}
return len(buf), nil;
}
func (b *buffer) String() string {
return string(b.a);
}
func write(t *testing.T, testname string, w *Writer, src string) {
written, err := io.WriteString(w, src);
if err != nil {
t.Errorf("--- test: %s\n--- src:\n%s\n--- write error: %v\n", testname, src, err);
}
if written != len(src) {
t.Errorf("--- test: %s\n--- src:\n%s\n--- written = %d, len(src) = %d\n", testname, src, written, len(src));
}
}
func verify(t *testing.T, testname string, w *Writer, b *buffer, src, expected string) {
err := w.Flush();
if err != nil {
t.Errorf("--- test: %s\n--- src:\n%s\n--- flush error: %v\n", testname, src, err);
}
res := b.String();
if res != expected {
t.Errorf("--- test: %s\n--- src:\n%s\n--- found:\n%s\n--- expected:\n%s\n", testname, src, res, expected)
}
}
func check(t *testing.T, testname string, tabwidth, padding int, padchar byte, flags uint, src, expected string) {
var b buffer;
b.init(1000);
var w Writer;
w.Init(&b, tabwidth, padding, padchar, flags);
// write all at once
b.clear();
write(t, testname, &w, src);
verify(t, testname, &w, &b, src, expected);
// write byte-by-byte
b.clear();
for i := 0; i < len(src); i++ {
write(t, testname, &w, src[i : i+1]);
}
verify(t, testname, &w, &b, src, expected);
// write using Fibonacci slice sizes
b.clear();
for i, d := 0, 0; i < len(src); {
write(t, testname, &w, src[i : i+d]);
i, d = i+d, d+1;
if i+d > len(src) {
d = len(src) - i;
}
}
verify(t, testname, &w, &b, src, expected);
}
type entry struct {
testname string;
tabwidth, padding int;
padchar byte;
flags uint;
src, expected string;
}
var tests = []entry {
entry{
"1",
8, 1, '.', 0,
"",
""
},
entry{
"2",
8, 1, '.', 0,
"\n\n\n",
"\n\n\n"
},
entry{
"3",
8, 1, '.', 0,
"a\nb\nc",
"a\nb\nc"
},
entry{
"4a",
8, 1, '.', 0,
"\t", // '\t' terminates an empty cell on last line - nothing to print
""
},
entry{
"4b",
8, 1, '.', AlignRight,
"\t", // '\t' terminates an empty cell on last line - nothing to print
""
},
entry{
"5",
8, 1, '.', 0,
"*\t*",
"*.......*"
},
entry{
"5b",
8, 1, '.', 0,
"*\t*\n",
"*.......*\n"
},
entry{
"5c",
8, 1, '.', 0,
"*\t*\t",
"*.......*"
},
entry{
"5d",
8, 1, '.', AlignRight,
"*\t*\t",
".......**"
},
entry{
"6",
8, 1, '.', 0,
"\t\n",
"........\n"
},
entry{
"7a",
8, 1, '.', 0,
"a) foo",
"a) foo"
},
entry{
"7b",
8, 1, ' ', 0,
"b) foo\tbar",
"b) foo bar"
},
entry{
"7c",
8, 1, '.', 0,
"c) foo\tbar\t",
"c) foo..bar"
},
entry{
"7d",
8, 1, '.', 0,
"d) foo\tbar\n",
"d) foo..bar\n"
},
entry{
"7e",
8, 1, '.', 0,
"e) foo\tbar\t\n",
"e) foo..bar.....\n"
},
entry{
"7f",
8, 1, '.', FilterHTML,
"f) f&lt;o\t<b>bar</b>\t\n",
"f) f&lt;o..<b>bar</b>.....\n"
},
entry{
"7g",
8, 1, '.', FilterHTML,
"g) f&lt;o\t<b>bar</b>\t non-terminated entity &amp",
"g) f&lt;o..<b>bar</b>..... non-terminated entity &amp"
},
entry{
"8",
8, 1, '*', 0,
"Hello, world!\n",
"Hello, world!\n"
},
entry{
"9a",
0, 0, '.', 0,
"1\t2\t3\t4\n"
"11\t222\t3333\t44444\n",
"1.2..3...4\n"
"11222333344444\n"
},
entry{
"9b",
0, 0, '.', FilterHTML,
"1\t2<!---\f--->\t3\t4\n" // \f inside HTML is ignored
"11\t222\t3333\t44444\n",
"1.2<!---\f--->..3...4\n"
"11222333344444\n"
},
entry{
"9c",
0, 0, '.', 0,
"1\t2\t3\t4\f" // \f causes a newline and flush
"11\t222\t3333\t44444\n",
"1234\n"
"11222333344444\n"
},
entry{
"10a",
5, 0, '.', 0,
"1\t2\t3\t4\n",
"1....2....3....4\n"
},
entry{
"10b",
5, 0, '.', 0,
"1\t2\t3\t4\t\n",
"1....2....3....4....\n"
},
entry{
"11",
8, 1, '.', 0,
"本\tb\tc\n"
"aa\t\u672c\u672c\u672c\tcccc\tddddd\n"
"aaa\tbbbb\n",
"本.......b.......c\n"
"aa......本本本.....cccc....ddddd\n"
"aaa.....bbbb\n"
},
entry{
"12a",
8, 1, ' ', AlignRight,
"a\tè\tc\t\n"
"aa\tèèè\tcccc\tddddd\t\n"
"aaa\tèèèè\t\n",
" a è c\n"
" aa èèè cccc ddddd\n"
" aaa èèèè\n"
},
entry{
"12b",
2, 0, ' ', 0,
"a\tb\tc\n"
"aa\tbbb\tcccc\n"
"aaa\tbbbb\n",
"a b c\n"
"aa bbbcccc\n"
"aaabbbb\n"
},
entry{
"12c",
8, 1, '_', 0,
"a\tb\tc\n"
"aa\tbbb\tcccc\n"
"aaa\tbbbb\n",
"a_______b_______c\n"
"aa______bbb_____cccc\n"
"aaa_____bbbb\n"
},
entry{
"13a",
4, 1, '-', 0,
"4444\t日本語\t22\t1\t333\n"
"999999999\t22\n"
"7\t22\n"
"\t\t\t88888888\n"
"\n"
"666666\t666666\t666666\t4444\n"
"1\t1\t999999999\t0000000000\n",
"4444------日本語-22--1---333\n"
"999999999-22\n"
"7---------22\n"
"------------------88888888\n"
"\n"
"666666-666666-666666----4444\n"
"1------1------999999999-0000000000\n"
},
entry{
"13b",
4, 3, '.', 0,
"4444\t333\t22\t1\t333\n"
"999999999\t22\n"
"7\t22\n"
"\t\t\t88888888\n"
"\n"
"666666\t666666\t666666\t4444\n"
"1\t1\t999999999\t0000000000\n",
"4444........333...22...1...333\n"
"999999999...22\n"
"7...........22\n"
"....................88888888\n"
"\n"
"666666...666666...666666......4444\n"
"1........1........999999999...0000000000\n"
},
entry{
"13c",
8, 1, '\t', FilterHTML,
"4444\t333\t22\t1\t333\n"
"999999999\t22\n"
"7\t22\n"
"\t\t\t88888888\n"
"\n"
"666666\t666666\t666666\t4444\n"
"1\t1\t<font color=red attr=日本語>999999999</font>\t0000000000\n",
"4444\t\t333\t22\t1\t333\n"
"999999999\t22\n"
"7\t\t22\n"
"\t\t\t\t88888888\n"
"\n"
"666666\t666666\t666666\t\t4444\n"
"1\t1\t<font color=red attr=日本語>999999999</font>\t0000000000\n"
},
entry{
"14",
0, 2, ' ', AlignRight,
".0\t.3\t2.4\t-5.1\t\n"
"23.0\t12345678.9\t2.4\t-989.4\t\n"
"5.1\t12.0\t2.4\t-7.0\t\n"
".0\t0.0\t332.0\t8908.0\t\n"
".0\t-.3\t456.4\t22.1\t\n"
".0\t1.2\t44.4\t-13.3\t\t",
" .0 .3 2.4 -5.1\n"
" 23.0 12345678.9 2.4 -989.4\n"
" 5.1 12.0 2.4 -7.0\n"
" .0 0.0 332.0 8908.0\n"
" .0 -.3 456.4 22.1\n"
" .0 1.2 44.4 -13.3"
},
}
func Test(t *testing.T) {
for _, e := range tests {
check(t, e.testname, e.tabwidth, e.padding, e.padchar, e.flags, e.src, e.expected);
}
}