summaryrefslogtreecommitdiff
path: root/jstests/tool/csvexport1.js
blob: afea559b2b0786381a96f14e2e3844853fae8609 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
// csvexport1.js

t = new ToolTest("csvexport1");

c = t.startDB("foo");

assert.eq(0, c.count(), "setup1");

objId = ObjectId();

c.insert({a: new NumberInt(1), b: objId, c: [1, 2, 3], d: {a: "hello", b: "world"}, e: '-'});
c.insert({a: -2.0, c: MinKey, d: "Then he said, \"Hello World!\"", e: new NumberLong(3)});
c.insert({
    a: new BinData(0, "1234"),
    b: ISODate("2009-08-27T12:34:56.789"),
    c: new Timestamp(1234, 9876),
    d: /foo*\"bar\"/i,
    e: function foo() {
        print("Hello World!");
    }
});

assert.eq(3, c.count(), "setup2");

t.runTool("export", "--out", t.extFile, "-d", t.baseName, "-c", "foo", "--csv", "-f", "a,b,c,d,e");

c.drop();

assert.eq(0, c.count(), "after drop");

t.runTool(
    "import", "--file", t.extFile, "-d", t.baseName, "-c", "foo", "--type", "csv", "--headerline");

assert.soon(3 + " == c.count()", "after import");

// Note: Exporting and Importing to/from CSV is not designed to be round-trippable
expected = [];
expected.push({
    a: 1,
    b: "ObjectId(" + objId.valueOf() + ")",
    c: [1, 2, 3],
    d: {"a": "hello", "b": "world"},
    e: "-"
});
expected.push({a: -2.0, b: "", c: "$MinKey", d: "Then he said, \"Hello World!\"", e: 3});
// "t" should be 1234, but the shell interprets the first field of timestamps as milliseconds while
// they are stored as seconds.  See SERVER-7718.
expected.push({
    a: "D76DF8",
    b: "2009-08-27T12:34:56.789Z",
    c: {"$timestamp": {"t": 1234, "i": 9876}},
    d: "/foo*\\\"bar\\\"/i",
    e: tojson(function foo() {
        print("Hello World!");
    })
});

actual = [];
actual.push(c.find({a: 1}).toArray()[0]);
actual.push(c.find({a: -2.0}).toArray()[0]);
actual.push(c.find({a: "D76DF8"}).toArray()[0]);

for (i = 0; i < expected.length; i++) {
    delete actual[i]._id;
    assert.eq(Object.keys(expected[i]).length, Object.keys(actual[i]).length);
    keys = Object.keys(expected[i]);
    for (var j = 0; j < keys.length; j++) {
        expectedVal = expected[i][keys[j]];
        if ((typeof expectedVal) == "object") {
            // For fields which contain arrays or objects, they have been
            // exported as JSON - parse the JSON in the output and verify
            // that it matches the original document's value
            assert.docEq(expectedVal, JSON.parse(actual[i][keys[j]]), "CSV export " + i);
        } else {
            // Otherwise just compare the values directly
            assert.eq(expectedVal, actual[i][keys[j]], "CSV export " + i);
        }
    }
}

t.stop();