/* 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/.
 */

const KEY_UPDATE_ARCHIVE_DIR = "UpdArchD"

var gActiveUpdate;
var gDirService;
var gDirProvider;
var gOldProviders;

function run_test() {
  setupTestCommon();

  debugDump("testing that error codes set from a directory provider propagate" +
            "up to AUS.downloadUpdate() correctly (Bug 794211).");

  start_httpserver();
  setUpdateURLOverride(gURLData + gHTTPHandlerPath);
  standardInit();

  gDirProvider = new FakeDirProvider();

  let cm = Cc["@mozilla.org/categorymanager;1"].getService(Ci.nsICategoryManager);
  gOldProviders = [];
  let enumerator = cm.enumerateCategory("xpcom-directory-providers");
  while (enumerator.hasMoreElements()) {
    let entry = enumerator.getNext().QueryInterface(Ci.nsISupportsCString).data;
    let contractID = cm.getCategoryEntry("xpcom-directory-providers", entry);
    gOldProviders.push(Cc[contractID].createInstance(Ci.nsIDirectoryServiceProvider));
  }

  gDirService = Cc["@mozilla.org/file/directory_service;1"].
                getService(Ci.nsIProperties);

  gOldProviders.forEach(function(p) {
    gDirService.unregisterProvider(p);
  });
  gDirService.registerProvider(gDirProvider);

  Services.prefs.setBoolPref(PREF_APP_UPDATE_SILENT, true);
  do_execute_soon(run_test_pt1);
}

function run_test_pt1() {
  gUpdates = null;
  gUpdateCount = null;
  gCheckFunc = check_test_pt1;

  let patches = getRemotePatchString();
  let updates = getRemoteUpdateString(patches);
  gResponseBody = getRemoteUpdatesXMLString(updates);
  gUpdateChecker.checkForUpdates(updateCheckListener, true);
}

function check_test_pt1() {
  Assert.equal(gUpdateCount, 1,
               "the update count" + MSG_SHOULD_EQUAL);

  gActiveUpdate = gUpdates[0];
  Assert.ok(!!gActiveUpdate,
            "there should be an active update");

  let state = gAUS.downloadUpdate(gActiveUpdate, true);
  Assert.equal(state, STATE_NONE,
               "the update state" + MSG_SHOULD_EQUAL);
  Assert.equal(gActiveUpdate.errorCode >>> 0, Cr.NS_ERROR_FILE_TOO_BIG,
               "the update error code" + MSG_SHOULD_EQUAL);

  doTestFinish();
}

function end_test() {
  gDirService.unregisterProvider(gDirProvider);
  gOldProviders.forEach(function(p) {
    gDirService.registerProvider(p);
  });
  gActiveUpdate = null;
  gDirService = null;
  gDirProvider = null;
}

function FakeDirProvider() {}
FakeDirProvider.prototype = {
  getFile: function FP_getFile(prop, persistent) {
    if (prop == KEY_UPDATE_ARCHIVE_DIR) {
      if (gActiveUpdate) {
        gActiveUpdate.errorCode = Cr.NS_ERROR_FILE_TOO_BIG;
      }
    }
    return null;
  },
  classID: Components.ID("{f30b43a7-2bfa-4e5f-8c4f-abc7dd4ac486}"),
  QueryInterface: XPCOMUtils.generateQI([Ci.nsIDirectoryServiceProvider])
};
