summaryrefslogtreecommitdiffstats
path: root/netwerk/test/unit/test_safeoutputstream.js
blob: 4925394ce4364fa78ab9e7a195942aa49b9faf54 (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
/* -*- 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/. */
"use strict";

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);

  Assert.equal(scriptStream.read(scriptStream.available()), str);
  scriptStream.close();
}

function run_test() {
  var filename = "\u0913";
  var file = Services.dirsvc.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");
}