|
4 | 4 |
|
5 | 5 | import com.fasterxml.jackson.dataformat.csv.*;
|
6 | 6 |
|
7 |
| -import com.google.common.collect.ImmutableMap; |
8 |
| - |
9 |
| -import org.junit.Test; |
10 |
| - |
11 |
| -import java.lang.String; |
12 |
| - |
13 |
| -// [Issue#33] |
| 7 | +// [dataformats-text#33] |
14 | 8 | public class TestWriterWithMissingValues extends ModuleTestBase
|
15 | 9 | {
|
16 | 10 | private final CsvSchema SCHEMA = new CsvSchema.Builder()
|
17 | 11 | .addColumn("timestamp", CsvSchema.ColumnType.STRING)
|
18 | 12 | .addColumn("value", CsvSchema.ColumnType.NUMBER)
|
19 | 13 | .addColumn("id", CsvSchema.ColumnType.STRING)
|
20 | 14 | .build();
|
21 |
| - final ObjectWriter WRITER = new CsvMapper().writer().with(SCHEMA); |
22 |
| - |
23 |
| - @Test |
| 15 | + private final CsvMapper MAPPER = mapperForCsv(); |
| 16 | + final ObjectWriter WRITER = MAPPER.writer().with(SCHEMA); |
| 17 | + |
24 | 18 | public void testWrite_NoNulls() {
|
25 | 19 | final String csv = WRITER.writeValueAsString(
|
26 |
| - ImmutableMap.of("timestamp", "2014-03-10T23:32:47+00:00", |
| 20 | + mapOf("timestamp", "2014-03-10T23:32:47+00:00", |
27 | 21 | "value", 42, "id", "hello"));
|
28 | 22 |
|
29 | 23 | assertEquals("\"2014-03-10T23:32:47+00:00\",42,hello\n", csv);
|
30 | 24 | }
|
31 | 25 |
|
32 |
| - @Test |
33 | 26 | public void testWrite_NullFirstColumn() {
|
34 | 27 | final String csv = WRITER.writeValueAsString(
|
35 |
| - ImmutableMap.of("value", 42, "id", "hello")); |
| 28 | + mapOf("value", 42, "id", "hello")); |
36 | 29 | assertEquals(",42,hello\n", csv);
|
37 | 30 | }
|
38 | 31 |
|
39 |
| - @Test |
40 | 32 | public void testWrite_NullSecondColumn() {
|
41 | 33 | final String csv = WRITER.writeValueAsString(
|
42 |
| - ImmutableMap.of("timestamp", "2014-03-10T23:32:47+00:00", |
| 34 | + mapOf("timestamp", "2014-03-10T23:32:47+00:00", |
43 | 35 | "id", "hello"));
|
44 | 36 |
|
45 | 37 | assertEquals("\"2014-03-10T23:32:47+00:00\",,hello\n", csv);
|
46 | 38 | }
|
47 | 39 |
|
48 |
| - @Test |
49 | 40 | public void testWrite_NullThirdColumn()
|
50 | 41 | {
|
51 |
| - CsvMapper mapper = new CsvMapper(); |
52 |
| - assertFalse(mapper.tokenStreamFactory().isEnabled(CsvGenerator.Feature.OMIT_MISSING_TAIL_COLUMNS)); |
53 |
| - String csv = mapper.writer(SCHEMA).writeValueAsString( |
54 |
| - ImmutableMap.of("timestamp", "2014-03-10T23:32:47+00:00", |
| 42 | + assertFalse(MAPPER.tokenStreamFactory().isEnabled(CsvGenerator.Feature.OMIT_MISSING_TAIL_COLUMNS)); |
| 43 | + String csv = MAPPER.writer(SCHEMA).writeValueAsString( |
| 44 | + mapOf("timestamp", "2014-03-10T23:32:47+00:00", |
55 | 45 | "value", 42));
|
56 | 46 |
|
57 | 47 | assertEquals("\"2014-03-10T23:32:47+00:00\",42,\n", csv);
|
58 |
| - ObjectWriter w = mapper.writer().with(CsvGenerator.Feature.OMIT_MISSING_TAIL_COLUMNS); |
| 48 | + ObjectWriter w = MAPPER.writer().with(CsvGenerator.Feature.OMIT_MISSING_TAIL_COLUMNS); |
59 | 49 | csv = w.with(SCHEMA).writeValueAsString(
|
60 |
| - ImmutableMap.of("timestamp", "2014-03-10T23:32:47+00:00", |
| 50 | + mapOf("timestamp", "2014-03-10T23:32:47+00:00", |
61 | 51 | "value", 42));
|
62 | 52 | assertEquals("\"2014-03-10T23:32:47+00:00\",42\n", csv);
|
63 | 53 | }
|
|
0 commit comments