1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
/* 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/. */
/*
* Test content length for the IMAP protocol. This focuses on necko URLs
* that are run externally.
*/
var { MailServices } = ChromeUtils.import(
"resource:///modules/MailServices.jsm"
);
var { PromiseTestUtils } = ChromeUtils.import(
"resource://testing-common/mailnews/PromiseTestUtils.jsm"
);
var gMsgHdr = null;
// Take a multipart message as we're testing attachment URLs as well
var gFile = do_get_file("../../../data/multipart-complex2");
add_setup(function () {
setupIMAPPump();
// Set up nsIMsgFolderListener to get the header when it's received
MailServices.mfn.addListener(msgAddedListener, MailServices.mfn.msgAdded);
IMAPPump.inbox.clearFlag(Ci.nsMsgFolderFlags.Offline);
});
// Adds some messages directly to a mailbox (eg new mail)
add_task(async function addMessageToServer() {
let URI = Services.io.newFileURI(gFile).QueryInterface(Ci.nsIFileURL);
IMAPPump.mailbox.addMessage(
new ImapMessage(URI.spec, IMAPPump.mailbox.uidnext++, [])
);
let listener = new PromiseTestUtils.PromiseUrlListener();
IMAPPump.inbox.updateFolderWithListener(null, listener);
await listener.promise;
});
function MsgAddedListener() {
this._promise = new Promise(resolve => {
this._resolve = resolve;
});
}
MsgAddedListener.prototype = {
msgAdded(aMsgHdr) {
gMsgHdr = aMsgHdr;
this._resolve();
},
get promise() {
return this._promise;
},
};
var msgAddedListener = new MsgAddedListener();
add_task(async function verifyContentLength() {
await msgAddedListener.promise;
let messageUri = IMAPPump.inbox.getUriForMsg(gMsgHdr);
// Convert this to a URI that necko can run
let messageService = MailServices.messageServiceFromURI(messageUri);
let neckoURL = messageService.getUrlForUri(messageUri);
// Don't use the necko URL directly. Instead, get the spec and create a new
// URL using the IO service
let urlToRun = Services.io.newURI(neckoURL.spec);
// Get a channel from this URI, and check its content length
let channel = Services.io.newChannelFromURI(
urlToRun,
null,
Services.scriptSecurityManager.getSystemPrincipal(),
null,
Ci.nsILoadInfo.SEC_ALLOW_CROSS_ORIGIN_SEC_CONTEXT_IS_NULL,
Ci.nsIContentPolicy.TYPE_OTHER
);
Assert.equal(channel.contentLength, gFile.fileSize);
// Now try an attachment. &part=1.2
let attachmentURL = Services.io.newURI(neckoURL.spec + "&part=1.2");
let attachmentChannel = Services.io.newChannelFromURI(
attachmentURL,
null,
Services.scriptSecurityManager.getSystemPrincipal(),
null,
Ci.nsILoadInfo.SEC_ALLOW_CROSS_ORIGIN_SEC_CONTEXT_IS_NULL,
Ci.nsIContentPolicy.TYPE_OTHER
);
// Currently attachments have their content length set to the length of the
// entire message
Assert.equal(attachmentChannel.contentLength, gFile.fileSize);
});
add_task(function endTest() {
MailServices.mfn.removeListener(msgAddedListener);
teardownIMAPPump();
});
|
