michael@0: /* Any copyright is dedicated to the Public Domain.
michael@0:    http://creativecommons.org/publicdomain/zero/1.0/ */
michael@0: 
michael@0: "use strict";
michael@0: 
michael@0: Cu.import("resource://gre/modules/Log.jsm");
michael@0: Cu.import("resource://services-common/utils.js");
michael@0: Cu.import("resource://services-common/hawkrequest.js");
michael@0: 
michael@0: function do_register_cleanup() {
michael@0:   Services.prefs.resetUserPrefs();
michael@0: 
michael@0:   // remove the pref change listener
michael@0:   let hawk = new HAWKAuthenticatedRESTRequest("https://example.com");
michael@0:   hawk._intl.uninit();
michael@0: }
michael@0: 
michael@0: function run_test() {
michael@0:   Log.repository.getLogger("Services.Common.RESTRequest").level =
michael@0:     Log.Level.Trace;
michael@0:   initTestLogging("Trace");
michael@0: 
michael@0:   run_next_test();
michael@0: }
michael@0: 
michael@0: 
michael@0: add_test(function test_intl_accept_language() {
michael@0:   let testCount = 0;
michael@0:   let languages = [
michael@0:     "zu-NP;vo",     // Nepalese dialect of Zulu, defaulting to Volapük
michael@0:     "fa-CG;ik",     // Congolese dialect of Farsei, defaulting to Inupiaq
michael@0:   ];
michael@0: 
michael@0:   function setLanguagePref(lang) {
michael@0:     let acceptLanguage = Cc["@mozilla.org/supports-string;1"]
michael@0:                            .createInstance(Ci.nsISupportsString);
michael@0:     acceptLanguage.data = lang;
michael@0:     Services.prefs.setComplexValue(
michael@0:       "intl.accept_languages", Ci.nsISupportsString, acceptLanguage);
michael@0:   }
michael@0: 
michael@0:   let hawk = new HAWKAuthenticatedRESTRequest("https://example.com");
michael@0: 
michael@0:   Services.prefs.addObserver("intl.accept_languages", checkLanguagePref, false);
michael@0:   setLanguagePref(languages[testCount]);
michael@0: 
michael@0:   function checkLanguagePref() {
michael@0:     var _done = false;
michael@0:     CommonUtils.nextTick(function() {
michael@0:       // Ensure we're only called for the number of entries in languages[].
michael@0:       do_check_true(testCount < languages.length);
michael@0: 
michael@0:       do_check_eq(hawk._intl.accept_languages, languages[testCount]);
michael@0: 
michael@0:       testCount++;
michael@0:       if (testCount < languages.length) {
michael@0:         // Set next language in prefs; Pref service will call checkNextLanguage.
michael@0:         setLanguagePref(languages[testCount]);
michael@0:         return;
michael@0:       }
michael@0: 
michael@0:       // We've checked all the entries in languages[]. Cleanup and move on.
michael@0:       do_print("Checked " + testCount + " languages. Removing checkLanguagePref as pref observer.");
michael@0:       Services.prefs.removeObserver("intl.accept_languages", checkLanguagePref);
michael@0:       run_next_test();
michael@0:       return;
michael@0:     });
michael@0:   }
michael@0: });
michael@0: 
michael@0: add_test(function test_hawk_authenticated_request() {
michael@0:   let onProgressCalled = false;
michael@0:   let postData = {your: "data"};
michael@0: 
michael@0:   // An arbitrary date - Feb 2, 1971.  It ends in a bunch of zeroes to make our
michael@0:   // computation with the hawk timestamp easier, since hawk throws away the
michael@0:   // millisecond values.
michael@0:   let then = 34329600000;
michael@0: 
michael@0:   let clockSkew = 120000;
michael@0:   let timeOffset = -1 * clockSkew;
michael@0:   let localTime = then + clockSkew;
michael@0: 
michael@0:   // Set the accept-languages pref to the Nepalese dialect of Zulu.
michael@0:   let acceptLanguage = Cc['@mozilla.org/supports-string;1'].createInstance(Ci.nsISupportsString);
michael@0:   acceptLanguage.data = 'zu-NP'; // omit trailing ';', which our HTTP libs snip
michael@0:   Services.prefs.setComplexValue('intl.accept_languages', Ci.nsISupportsString, acceptLanguage);
michael@0: 
michael@0:   let credentials = {
michael@0:     id: "eyJleHBpcmVzIjogMTM2NTAxMDg5OC4x",
michael@0:     key: "qTZf4ZFpAMpMoeSsX3zVRjiqmNs=",
michael@0:     algorithm: "sha256"
michael@0:   };
michael@0: 
michael@0:   let server = httpd_setup({
michael@0:     "/elysium": function(request, response) {
michael@0:       do_check_true(request.hasHeader("Authorization"));
michael@0: 
michael@0:       // check that the header timestamp is our arbitrary system date, not
michael@0:       // today's date.  Note that hawk header timestamps are in seconds, not
michael@0:       // milliseconds.
michael@0:       let authorization = request.getHeader("Authorization");
michael@0:       let tsMS = parseInt(/ts="(\d+)"/.exec(authorization)[1], 10) * 1000;
michael@0:       do_check_eq(tsMS, then);
michael@0: 
michael@0:       // This testing can be a little wonky. In an environment where
michael@0:       //   pref("intl.accept_languages") === 'en-US, en'
michael@0:       // the header is sent as:
michael@0:       //   'en-US,en;q=0.5'
michael@0:       // hence our fake value for acceptLanguage.
michael@0:       let lang = request.getHeader("Accept-Language");
michael@0:       do_check_eq(lang, acceptLanguage);
michael@0: 
michael@0:       let message = "yay";
michael@0:       response.setStatusLine(request.httpVersion, 200, "OK");
michael@0:       response.bodyOutputStream.write(message, message.length);
michael@0:     }
michael@0:   });
michael@0: 
michael@0:   function onProgress() {
michael@0:     onProgressCalled = true;
michael@0:   }
michael@0: 
michael@0:   function onComplete(error) {
michael@0:     do_check_eq(200, this.response.status);
michael@0:     do_check_eq(this.response.body, "yay");
michael@0:     do_check_true(onProgressCalled);
michael@0: 
michael@0:     Services.prefs.resetUserPrefs();
michael@0:     let pref = Services.prefs.getComplexValue(
michael@0:       "intl.accept_languages", Ci.nsIPrefLocalizedString);
michael@0:     do_check_neq(acceptLanguage.data, pref.data);
michael@0: 
michael@0:     server.stop(run_next_test);
michael@0:   }
michael@0: 
michael@0:   let url = server.baseURI + "/elysium";
michael@0:   let extra = {
michael@0:     now: localTime,
michael@0:     localtimeOffsetMsec: timeOffset
michael@0:   };
michael@0: 
michael@0:   let request = new HAWKAuthenticatedRESTRequest(url, credentials, extra);
michael@0: 
michael@0:   // Allow hawk._intl to respond to the language pref change
michael@0:   CommonUtils.nextTick(function() {
michael@0:     request.post(postData, onComplete, onProgress);
michael@0:   });
michael@0: });
michael@0: 
michael@0: add_test(function test_hawk_language_pref_changed() {
michael@0:   let languages = [
michael@0:     "zu-NP",        // Nepalese dialect of Zulu
michael@0:     "fa-CG",        // Congolese dialect of Farsi
michael@0:   ];
michael@0: 
michael@0:   let credentials = {
michael@0:     id: "eyJleHBpcmVzIjogMTM2NTAxMDg5OC4x",
michael@0:     key: "qTZf4ZFpAMpMoeSsX3zVRjiqmNs=",
michael@0:     algorithm: "sha256",
michael@0:   };
michael@0: 
michael@0:   function setLanguage(lang) {
michael@0:     let acceptLanguage = Cc["@mozilla.org/supports-string;1"].createInstance(Ci.nsISupportsString);
michael@0:     acceptLanguage.data = lang;
michael@0:     Services.prefs.setComplexValue("intl.accept_languages", Ci.nsISupportsString, acceptLanguage);
michael@0:   }
michael@0: 
michael@0:   let server = httpd_setup({
michael@0:     "/foo": function(request, response) {
michael@0:       do_check_eq(languages[1], request.getHeader("Accept-Language"));
michael@0: 
michael@0:       response.setStatusLine(request.httpVersion, 200, "OK");
michael@0:     },
michael@0:   });
michael@0: 
michael@0:   let url = server.baseURI + "/foo";
michael@0:   let postData = {};
michael@0:   let request;
michael@0: 
michael@0:   setLanguage(languages[0]);
michael@0: 
michael@0:   // A new request should create the stateful object for tracking the current
michael@0:   // language.
michael@0:   request = new HAWKAuthenticatedRESTRequest(url, credentials);
michael@0:   CommonUtils.nextTick(testFirstLanguage);
michael@0: 
michael@0:   function testFirstLanguage() {
michael@0:     do_check_eq(languages[0], request._intl.accept_languages);
michael@0: 
michael@0:     // Change the language pref ...
michael@0:     setLanguage(languages[1]); 
michael@0:     CommonUtils.nextTick(testRequest);
michael@0:   }
michael@0: 
michael@0:   function testRequest() {
michael@0:     // Change of language pref should be picked up, which we can see on the
michael@0:     // server by inspecting the request headers.
michael@0:     request = new HAWKAuthenticatedRESTRequest(url, credentials);
michael@0:     request.post({}, function(error) {
michael@0:       do_check_null(error);
michael@0:       do_check_eq(200, this.response.status);
michael@0: 
michael@0:       Services.prefs.resetUserPrefs();
michael@0: 
michael@0:       server.stop(run_next_test);
michael@0:     });
michael@0:   }
michael@0: });
michael@0: