summaryrefslogtreecommitdiffstats
path: root/netwerk/test/unit/test_safeoutputstream.js
blob: 3e41fcbb0266cdb688a9d6a4c42d9101c4e959f5 (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
/* -*- indent-tabs-mode: nil; js-indent-level: 4 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

function write_atomic(file, str) {
    var stream = Cc["@mozilla.org/network/atomic-file-output-stream;1"]
                   .createInstance(Ci.nsIFileOutputStream);
    stream.init(file, -1, -1, 0);
    do {
        var written = stream.write(str, str.length);
        if (written == str.length)
            break;
        str = str.substring(written);
    } while (1);
    stream.QueryInterface(Ci.nsISafeOutputStream).finish();
    stream.close();
}

function write(file, str) {
    var stream = Cc["@mozilla.org/network/safe-file-output-stream;1"]
                   .createInstance(Ci.nsIFileOutputStream);
    stream.init(file, -1, -1, 0);
    do {
        var written = stream.write(str, str.length);
        if (written == str.length)
            break;
        str = str.substring(written);
    } while (1);
    stream.QueryInterface(Ci.nsISafeOutputStream).finish();
    stream.close();
}

function checkFile(file, str) {
    var stream = Cc["@mozilla.org/network/file-input-stream;1"]
                   .createInstance(Ci.nsIFileInputStream);
    stream.init(file, -1, -1, 0);

    var scriptStream = Cc["@mozilla.org/scriptableinputstream;1"]
                         .createInstance(Ci.nsIScriptableInputStream);
    scriptStream.init(stream);

    do_check_eq(scriptStream.read(scriptStream.available()), str);
    scriptStream.close();
}

function run_test()
{
    var filename = "\u0913";
    var file = Cc["@mozilla.org/file/directory_service;1"]
                 .getService(Ci.nsIProperties)
                 .get("TmpD", Ci.nsIFile);
    file.append(filename);

    write(file, "First write");
    checkFile(file, "First write");

    write(file, "Second write");
    checkFile(file, "Second write");

    write_atomic(file, "First write: Atomic");
    checkFile(file, "First write: Atomic");

    write_atomic(file, "Second write: Atomic");
    checkFile(file, "Second write: Atomic");
}