45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
// Copyright 2011 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 csv
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
var writeTests = []struct {
|
|
Input [][]string
|
|
Output string
|
|
UseCRLF bool
|
|
}{
|
|
{Input: [][]string{{"abc"}}, Output: "abc\n"},
|
|
{Input: [][]string{{"abc"}}, Output: "abc\r\n", UseCRLF: true},
|
|
{Input: [][]string{{`"abc"`}}, Output: `"""abc"""` + "\n"},
|
|
{Input: [][]string{{`a"b`}}, Output: `"a""b"` + "\n"},
|
|
{Input: [][]string{{`"a"b"`}}, Output: `"""a""b"""` + "\n"},
|
|
{Input: [][]string{{" abc"}}, Output: `" abc"` + "\n"},
|
|
{Input: [][]string{{"abc,def"}}, Output: `"abc,def"` + "\n"},
|
|
{Input: [][]string{{"abc", "def"}}, Output: "abc,def\n"},
|
|
{Input: [][]string{{"abc"}, {"def"}}, Output: "abc\ndef\n"},
|
|
{Input: [][]string{{"abc\ndef"}}, Output: "\"abc\ndef\"\n"},
|
|
{Input: [][]string{{"abc\ndef"}}, Output: "\"abc\r\ndef\"\r\n", UseCRLF: true},
|
|
}
|
|
|
|
func TestWrite(t *testing.T) {
|
|
for n, tt := range writeTests {
|
|
b := &bytes.Buffer{}
|
|
f := NewWriter(b)
|
|
f.UseCRLF = tt.UseCRLF
|
|
err := f.WriteAll(tt.Input)
|
|
if err != nil {
|
|
t.Errorf("Unexpected error: %s\n", err)
|
|
}
|
|
out := b.String()
|
|
if out != tt.Output {
|
|
t.Errorf("#%d: out=%q want %q", n, out, tt.Output)
|
|
}
|
|
}
|
|
}
|