summaryrefslogtreecommitdiffstats
path: root/comm/mailnews/local/test/unit/test_streamHeaders.js
blob: a1e7ef16409bad7c013570d7fe6c1f4824047768 (plain)
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
/* 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/. */

/**
 * This mainly tests that streamHeaders does not result in the crash
 * of bug 752768
 *
 * adapted from test_pop3Pump.js by Kent James <kent@caspia.com>
 */

/* import-globals-from ../../../test/resources/alertTestUtils.js */
load("../../../resources/alertTestUtils.js");

/* import-globals-from ../../../test/resources/POP3pump.js */
load("../../../resources/POP3pump.js");

var { PromiseTestUtils } = ChromeUtils.import(
  "resource://testing-common/mailnews/PromiseTestUtils.jsm"
);

var testSubjects = ["Hello, did you receive my bugmail?"];

var gHdr;

add_task(async function loadMessages() {
  let pop3Resolve;
  let pop3Promise = new Promise(resolve => {
    pop3Resolve = resolve;
  });
  gPOP3Pump.files = ["../../../data/draft1"];
  gPOP3Pump.onDone = pop3Resolve;
  gPOP3Pump.run();
  await pop3Promise;

  // Get message headers for the inbox folder.
  var msgCount = 0;
  for (gHdr of localAccountUtils.inboxFolder.msgDatabase.enumerateMessages()) {
    msgCount++;
    Assert.equal(gHdr.subject, testSubjects[msgCount - 1]);
  }
  Assert.equal(msgCount, 1);
  gPOP3Pump = null;
});

add_task(async function goodStreaming() {
  // Try to stream the headers of the last message.
  let uri = gHdr.folder.getUriForMsg(gHdr);
  let messageService = MailServices.messageServiceFromURI(uri);
  let streamListener = new PromiseTestUtils.PromiseStreamListener();
  messageService.streamHeaders(uri, streamListener, null, true);
  // The message contains this header.
  let streamData = await streamListener.promise;
  Assert.ok(
    streamData.includes(
      "X-Mozilla-Draft-Info: internal/draft; vcard=0; receipt=0; DSN=0; uuencode=0"
    )
  );
});

/**
 * Crash from bug 752768.
 */
add_task(async function badStreaming() {
  // Try to stream the headers of the last message.
  let folder = gHdr.folder;
  let uri = folder.getUriForMsg(gHdr);

  let dbFile = folder.summaryFile;
  // Force an invalid database.
  folder.msgDatabase.forceClosed();
  dbFile.remove(false);
  folder.msgDatabase = null;

  let messageService = MailServices.messageServiceFromURI(uri);
  let haveError = false;
  try {
    let streamListener = new PromiseTestUtils.PromiseStreamListener();
    messageService.streamHeaders(uri, streamListener, null, true);
    await streamListener.promise;
  } catch (e) {
    // Should throw NS_MSG_ERROR_FOLDER_SUMMARY_OUT_OF_DATE (0x80550005).
    haveError = true;
  } finally {
    Assert.ok(
      haveError,
      "Ensure that the stream crashes with NS_MSG_ERROR_FOLDER_SUMMARY_OUT_OF_DATE"
    );
  }
});