diff --git a/csv_writer_test.go b/csv_writer_test.go index f7746d1..a18442c 100644 --- a/csv_writer_test.go +++ b/csv_writer_test.go @@ -37,6 +37,36 @@ func TestCsvWriterWriteRows(t *testing.T) { } } +func TestCsvWriterWriteRowsToDir(t *testing.T) { + fw := NewTestFileWriter() + writer := NewCsvWriter(fw, "", "/tmp/some_dir", ";") + rows1 := make([]*map[string]interface{}, 0) + rows1 = append(rows1, &map[string]interface{}{ + "id": 1, + "name": "Name 1", + }) + rows2 := make([]*map[string]interface{}, 0) + rows2 = append(rows2, &map[string]interface{}{ + "id": 1, + "value": 5, + }) + + writer.WriteRows("some_table1", []string{"id", "name"}, rows1) + writer.WriteRows("some_table2", []string{"id", "value"}, rows2) + result1 := fw.getContents("/tmp/some_dir/some_table1.csv") + result2 := fw.getContents("/tmp/some_dir/some_table2.csv") + expected1 := "\"id\";\"name\"\r\n" + expected1 += "1;\"Name 1\"\r\n" + if expected1 != result1 { + t.Errorf("Expected:\n%sGot:\n%s", expected1, result1) + } + expected2 := "\"id\";\"value\"\r\n" + expected2 += "1;5\r\n" + if expected2 != result2 { + t.Errorf("Expected:\n%sGot:\n%s", expected2, result2) + } +} + func TestCsvWriterWriteDDL(t *testing.T) { fw := NewTestFileWriter() writer := NewCsvWriter(fw, "result.csv", "", ",")