toolkit/mozapps/extensions/test/browser/browser_hotfix.js

changeset 0
6474c204b198
     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/toolkit/mozapps/extensions/test/browser/browser_hotfix.js	Wed Dec 31 06:09:35 2014 +0100
     1.3 @@ -0,0 +1,190 @@
     1.4 +/* Any copyright is dedicated to the Public Domain.
     1.5 + * http://creativecommons.org/publicdomain/zero/1.0/
     1.6 + */
     1.7 +
     1.8 +const PREF_EM_HOTFIX_ID                = "extensions.hotfix.id";
     1.9 +const PREF_EM_HOTFIX_LASTVERSION       = "extensions.hotfix.lastVersion";
    1.10 +const PREF_EM_HOTFIX_URL               = "extensions.hotfix.url";
    1.11 +const PREF_EM_HOTFIX_CERTS             = "extensions.hotfix.certs.";
    1.12 +const PREF_EM_CERT_CHECKATTRIBUTES     = "extensions.hotfix.cert.checkAttributes";
    1.13 +
    1.14 +const PREF_INSTALL_REQUIREBUILTINCERTS = "extensions.install.requireBuiltInCerts";
    1.15 +const PREF_UPDATE_REQUIREBUILTINCERTS  = "extensions.update.requireBuiltInCerts";
    1.16 +
    1.17 +const PREF_APP_UPDATE_ENABLED          = "app.update.enabled";
    1.18 +const PREF_APP_UPDATE_URL              = "app.update.url";
    1.19 +
    1.20 +const HOTFIX_ID = "hotfix@tests.mozilla.org";
    1.21 +
    1.22 +var gNextTest;
    1.23 +
    1.24 +var SuccessfulInstallListener = {
    1.25 +  onDownloadCancelled: function(aInstall) {
    1.26 +    ok(false, "Should not have seen the download cancelled");
    1.27 +    is(aInstall.addon.id, HOTFIX_ID, "Should have seen the right add-on");
    1.28 +
    1.29 +    AddonManager.removeInstallListener(this);
    1.30 +    gNextTest();
    1.31 +  },
    1.32 +
    1.33 +  onInstallEnded: function(aInstall) {
    1.34 +    ok(true, "Should have seen the install complete");
    1.35 +    is(aInstall.addon.id, HOTFIX_ID, "Should have installed the right add-on");
    1.36 +
    1.37 +    AddonManager.removeInstallListener(this);
    1.38 +    aInstall.addon.uninstall();
    1.39 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_LASTVERSION);
    1.40 +    gNextTest();
    1.41 +  }
    1.42 +}
    1.43 +
    1.44 +var FailedInstallListener = {
    1.45 +  onDownloadCancelled: function(aInstall) {
    1.46 +    ok(true, "Should have seen the download cancelled");
    1.47 +    is(aInstall.addon.id, HOTFIX_ID, "Should have seen the right add-on");
    1.48 +
    1.49 +    AddonManager.removeInstallListener(this);
    1.50 +    gNextTest();
    1.51 +  },
    1.52 +
    1.53 +  onInstallEnded: function(aInstall) {
    1.54 +    ok(false, "Should not have seen the install complete");
    1.55 +    is(aInstall.addon.id, HOTFIX_ID, "Should have installed the right add-on");
    1.56 +
    1.57 +    AddonManager.removeInstallListener(this);
    1.58 +    aInstall.addon.uninstall();
    1.59 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_LASTVERSION);
    1.60 +    gNextTest();
    1.61 +  }
    1.62 +}
    1.63 +
    1.64 +function test() {
    1.65 +  waitForExplicitFinish();
    1.66 +
    1.67 +  var oldAusUrl = Services.prefs.getDefaultBranch(null).getCharPref(PREF_APP_UPDATE_URL);
    1.68 +  Services.prefs.getDefaultBranch(null).setCharPref(PREF_APP_UPDATE_URL, TESTROOT + "ausdummy.xml");
    1.69 +  Services.prefs.setBoolPref(PREF_APP_UPDATE_ENABLED, true);
    1.70 +  Services.prefs.setBoolPref(PREF_INSTALL_REQUIREBUILTINCERTS, false);
    1.71 +  Services.prefs.setBoolPref(PREF_UPDATE_REQUIREBUILTINCERTS, false);
    1.72 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_ID, HOTFIX_ID);
    1.73 +  var oldURL = Services.prefs.getCharPref(PREF_EM_HOTFIX_URL);
    1.74 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_URL, TESTROOT + "signed_hotfix.rdf");
    1.75 +
    1.76 +  registerCleanupFunction(function() {
    1.77 +    Services.prefs.setBoolPref(PREF_APP_UPDATE_ENABLED, false);
    1.78 +    Services.prefs.getDefaultBranch(null).setCharPref(PREF_APP_UPDATE_URL, oldAusUrl);
    1.79 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_ID);
    1.80 +    Services.prefs.setCharPref(PREF_EM_HOTFIX_URL, oldURL);
    1.81 +    Services.prefs.clearUserPref(PREF_INSTALL_REQUIREBUILTINCERTS);
    1.82 +    Services.prefs.clearUserPref(PREF_UPDATE_REQUIREBUILTINCERTS);
    1.83 +
    1.84 +    Services.prefs.clearUserPref(PREF_EM_CERT_CHECKATTRIBUTES);
    1.85 +    var prefs = Services.prefs.getChildList(PREF_EM_HOTFIX_CERTS);
    1.86 +    prefs.forEach(Services.prefs.clearUserPref);
    1.87 +  });
    1.88 +
    1.89 +  run_next_test();
    1.90 +}
    1.91 +
    1.92 +function end_test() {
    1.93 +  finish();
    1.94 +}
    1.95 +
    1.96 +add_test(function check_no_cert_checks() {
    1.97 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, false);
    1.98 +  AddonManager.addInstallListener(SuccessfulInstallListener);
    1.99 +
   1.100 +  gNextTest = run_next_test;
   1.101 +
   1.102 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.103 +});
   1.104 +
   1.105 +add_test(function check_wrong_cert_fingerprint() {
   1.106 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true);
   1.107 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "foo");
   1.108 +
   1.109 +  AddonManager.addInstallListener(FailedInstallListener);
   1.110 +
   1.111 +  gNextTest = function() {
   1.112 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint");
   1.113 +
   1.114 +    run_next_test();
   1.115 +  };
   1.116 +
   1.117 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.118 +});
   1.119 +
   1.120 +add_test(function check_right_cert_fingerprint() {
   1.121 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true);
   1.122 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2");
   1.123 +
   1.124 +  AddonManager.addInstallListener(SuccessfulInstallListener);
   1.125 +
   1.126 +  gNextTest = function() {
   1.127 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint");
   1.128 +
   1.129 +    run_next_test();
   1.130 +  };
   1.131 +
   1.132 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.133 +});
   1.134 +
   1.135 +add_test(function check_multi_cert_fingerprint_1() {
   1.136 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true);
   1.137 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2");
   1.138 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint", "foo");
   1.139 +
   1.140 +  AddonManager.addInstallListener(SuccessfulInstallListener);
   1.141 +
   1.142 +  gNextTest = function() {
   1.143 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint");
   1.144 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint");
   1.145 +
   1.146 +    run_next_test();
   1.147 +  };
   1.148 +
   1.149 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.150 +});
   1.151 +
   1.152 +add_test(function check_multi_cert_fingerprint_2() {
   1.153 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true);
   1.154 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "foo");
   1.155 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2");
   1.156 +
   1.157 +  AddonManager.addInstallListener(SuccessfulInstallListener);
   1.158 +
   1.159 +  gNextTest = function() {
   1.160 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint");
   1.161 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "2.sha1Fingerprint");
   1.162 +
   1.163 +    run_next_test();
   1.164 +  };
   1.165 +
   1.166 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.167 +});
   1.168 +
   1.169 +add_test(function check_no_cert_no_checks() {
   1.170 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, false);
   1.171 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_URL, TESTROOT + "unsigned_hotfix.rdf");
   1.172 +
   1.173 +  AddonManager.addInstallListener(SuccessfulInstallListener);
   1.174 +
   1.175 +  gNextTest = run_next_test;
   1.176 +
   1.177 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.178 +});
   1.179 +
   1.180 +add_test(function check_no_cert_cert_fingerprint_check() {
   1.181 +  Services.prefs.setBoolPref(PREF_EM_CERT_CHECKATTRIBUTES, true);
   1.182 +  Services.prefs.setCharPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint", "3E:B9:4E:07:12:FE:3C:01:41:46:13:46:FC:84:52:1A:8C:BE:1D:A2");
   1.183 +
   1.184 +  AddonManager.addInstallListener(FailedInstallListener);
   1.185 +
   1.186 +  gNextTest = function() {
   1.187 +    Services.prefs.clearUserPref(PREF_EM_HOTFIX_CERTS + "1.sha1Fingerprint");
   1.188 +
   1.189 +    run_next_test();
   1.190 +  };
   1.191 +
   1.192 +  AddonManagerPrivate.backgroundUpdateCheck();
   1.193 +});

mercurial