|
1 /* This Source Code Form is subject to the terms of the Mozilla Public |
|
2 * License, v. 2.0. If a copy of the MPL was not distributed with this |
|
3 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ |
|
4 |
|
5 const Ci = Components.interfaces; |
|
6 const Cu = Components.utils; |
|
7 |
|
8 Cu.import("resource://testing-common/httpd.js"); |
|
9 |
|
10 const PATH = "/file.meh"; |
|
11 var httpserver = new HttpServer(); |
|
12 |
|
13 // Each time, the data consist in a string that should be sniffed as Ogg. |
|
14 const data = "OggS\0meeeh."; |
|
15 var testRan = 0; |
|
16 |
|
17 // If the content-type is not present, or if it's application/octet-stream, it |
|
18 // should be sniffed to application/ogg by the media sniffer. Otherwise, it |
|
19 // should not be changed. |
|
20 const tests = [ |
|
21 // Those three first case are the case of a media loaded in a media element. |
|
22 // The first two should be sniffeed. |
|
23 { contentType: "", |
|
24 expectedContentType: "application/ogg", |
|
25 flags: Ci.nsIChannel.LOAD_TREAT_APPLICATION_OCTET_STREAM_AS_UNKNOWN }, |
|
26 { contentType: "application/octet-stream", |
|
27 expectedContentType: "application/ogg", |
|
28 flags: Ci.nsIChannel.LOAD_TREAT_APPLICATION_OCTET_STREAM_AS_UNKNOWN }, |
|
29 { contentType: "application/something", |
|
30 expectedContentType: "application/something", |
|
31 flags: Ci.nsIChannel.LOAD_TREAT_APPLICATION_OCTET_STREAM_AS_UNKNOWN }, |
|
32 // This last case tests the case of a channel opened while allowing content |
|
33 // sniffers to override the content-type, like in the docshell. |
|
34 { contentType: "application/octet-stream", |
|
35 expectedContentType: "application/ogg", |
|
36 flags: Ci.nsIChannel.LOAD_CALL_CONTENT_SNIFFERS }, |
|
37 { contentType: "", |
|
38 expectedContentType: "application/ogg", |
|
39 flags: Ci.nsIChannel.LOAD_CALL_CONTENT_SNIFFERS }, |
|
40 ]; |
|
41 |
|
42 // A basic listener that reads checks the if we sniffed properly. |
|
43 var listener = { |
|
44 onStartRequest: function(request, context) { |
|
45 do_check_eq(request.QueryInterface(Ci.nsIChannel).contentType, |
|
46 tests[testRan].expectedContentType); |
|
47 }, |
|
48 |
|
49 onDataAvailable: function(request, context, stream, offset, count) { |
|
50 try { |
|
51 var bis = Components.classes["@mozilla.org/binaryinputstream;1"] |
|
52 .createInstance(Components.interfaces.nsIBinaryInputStream); |
|
53 bis.setInputStream(stream); |
|
54 bis.readByteArray(bis.available()); |
|
55 } catch (ex) { |
|
56 do_throw("Error in onDataAvailable: " + ex); |
|
57 } |
|
58 }, |
|
59 |
|
60 onStopRequest: function(request, context, status) { |
|
61 testRan++; |
|
62 runNext(); |
|
63 } |
|
64 }; |
|
65 |
|
66 function setupChannel(url, flags) |
|
67 { |
|
68 var ios = Components.classes["@mozilla.org/network/io-service;1"]. |
|
69 getService(Ci.nsIIOService); |
|
70 var chan = ios.newChannel("http://localhost:" + |
|
71 httpserver.identity.primaryPort + url, "", null); |
|
72 chan.loadFlags |= flags; |
|
73 var httpChan = chan.QueryInterface(Components.interfaces.nsIHttpChannel); |
|
74 return httpChan; |
|
75 } |
|
76 |
|
77 function runNext() { |
|
78 if (testRan == tests.length) { |
|
79 do_test_finished(); |
|
80 return; |
|
81 } |
|
82 var channel = setupChannel(PATH, tests[testRan].flags); |
|
83 httpserver.registerPathHandler(PATH, function(request, response) { |
|
84 response.setHeader("Content-Type", tests[testRan].contentType, false); |
|
85 response.bodyOutputStream.write(data, data.length); |
|
86 }); |
|
87 channel.asyncOpen(listener, channel, null); |
|
88 } |
|
89 |
|
90 function run_test() { |
|
91 httpserver.start(-1); |
|
92 do_test_pending(); |
|
93 try { |
|
94 runNext(); |
|
95 } catch (e) { |
|
96 print("ERROR - " + e + "\n"); |
|
97 } |
|
98 } |