michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: function write_atomic(file, str) {
michael@0:     var stream = Cc["@mozilla.org/network/atomic-file-output-stream;1"]
michael@0:                    .createInstance(Ci.nsIFileOutputStream);
michael@0:     stream.init(file, -1, -1, 0);
michael@0:     do {
michael@0:         var written = stream.write(str, str.length);
michael@0:         if (written == str.length)
michael@0:             break;
michael@0:         str = str.substring(written);
michael@0:     } while (1);
michael@0:     stream.QueryInterface(Ci.nsISafeOutputStream).finish();
michael@0:     stream.close();
michael@0: }
michael@0: 
michael@0: function write(file, str) {
michael@0:     var stream = Cc["@mozilla.org/network/safe-file-output-stream;1"]
michael@0:                    .createInstance(Ci.nsIFileOutputStream);
michael@0:     stream.init(file, -1, -1, 0);
michael@0:     do {
michael@0:         var written = stream.write(str, str.length);
michael@0:         if (written == str.length)
michael@0:             break;
michael@0:         str = str.substring(written);
michael@0:     } while (1);
michael@0:     stream.QueryInterface(Ci.nsISafeOutputStream).finish();
michael@0:     stream.close();
michael@0: }
michael@0: 
michael@0: function checkFile(file, str) {
michael@0:     var stream = Cc["@mozilla.org/network/file-input-stream;1"]
michael@0:                    .createInstance(Ci.nsIFileInputStream);
michael@0:     stream.init(file, -1, -1, 0);
michael@0: 
michael@0:     var scriptStream = Cc["@mozilla.org/scriptableinputstream;1"]
michael@0:                          .createInstance(Ci.nsIScriptableInputStream);
michael@0:     scriptStream.init(stream);
michael@0: 
michael@0:     do_check_eq(scriptStream.read(scriptStream.available()), str);
michael@0:     scriptStream.close();
michael@0: }
michael@0: 
michael@0: function run_test()
michael@0: {
michael@0:     var filename = "\u0913";
michael@0:     var file = Cc["@mozilla.org/file/directory_service;1"]
michael@0:                  .getService(Ci.nsIProperties)
michael@0:                  .get("TmpD", Ci.nsIFile);
michael@0:     file.append(filename);
michael@0: 
michael@0:     write(file, "First write");
michael@0:     checkFile(file, "First write");
michael@0: 
michael@0:     write(file, "Second write");
michael@0:     checkFile(file, "Second write");
michael@0: 
michael@0:     write_atomic(file, "First write: Atomic");
michael@0:     checkFile(file, "First write: Atomic");
michael@0: 
michael@0:     write_atomic(file, "Second write: Atomic");
michael@0:     checkFile(file, "Second write: Atomic");
michael@0: }