summaryrefslogtreecommitdiffstats
path: root/toolkit/mozapps/update/tests/unit_aus_update/updateSyncManager.js
blob: a0f3fe080f6b472a7d289883bd7a8e59237182bb (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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
/* Any copyright is dedicated to the Public Domain.
 * http://creativecommons.org/publicdomain/zero/1.0/
 */

// This test verifies that the update sync manager is working correctly by
// a) making sure we're the only one that's opened it to begin with, and then
// b) starting a second copy of the same binary and making sure we can tell we
//    are no longer the only one that's opened it.

const { Subprocess } = ChromeUtils.importESModule(
  "resource://gre/modules/Subprocess.sys.mjs"
);

// Save off the real GRE directory and binary path before we register our
// mock directory service which overrides them both.
const thisBinary = Services.dirsvc.get("XREExeF", Ci.nsIFile);
const greDir = Services.dirsvc.get("GreD", Ci.nsIFile);

add_task(async function() {
  setupTestCommon();

  // First check that we believe we exclusively hold the lock.
  let syncManager = Cc["@mozilla.org/updates/update-sync-manager;1"].getService(
    Ci.nsIUpdateSyncManager
  );
  Assert.ok(
    !syncManager.isOtherInstanceRunning(),
    "no other instance is running yet"
  );

  // Now start a second copy of this xpcshell binary so that something else
  // takes the same lock. First we'll define its command line.
  // Most of the child's code is in a separate script file, so all the command
  // line has to do is set up a few required path strings we need to pass
  // through to the child, and then include the script file.
  const args = [
    "-g",
    greDir.path,
    "-e",
    `
      const customGreDirPath = "${getApplyDirFile(
        DIR_RESOURCES
      ).path.replaceAll("\\", "\\\\")}";
      const customGreBinDirPath = "${getApplyDirFile(DIR_MACOS).path.replaceAll(
        "\\",
        "\\\\"
      )}";
      const customExePath = "${getApplyDirFile(
        DIR_MACOS + FILE_APP_BIN
      ).path.replaceAll("\\", "\\\\")}";
      const customUpdDirPath = "${getMockUpdRootD().path.replaceAll(
        "\\",
        "\\\\"
      )}";
      const customOldUpdDirPath = "${getMockUpdRootD(true).path.replaceAll(
        "\\",
        "\\\\"
      )}";
    `,
    "-f",
    getTestDirFile("syncManagerTestChild.js").path,
  ];

  // Run the second copy two times, to show the lock is usable after having
  // been closed.
  for (let runs = 0; runs < 2; runs++) {
    // Now we can actually invoke the process.
    debugDump(
      `launching child process at ${thisBinary.path} with args ${args}`
    );
    Subprocess.call({
      command: thisBinary.path,
      arguments: args,
      stderr: "stdout",
    });

    // It will take the new xpcshell a little time to start up, but we should see
    // the effect on the lock within at most a few seconds.
    await TestUtils.waitForCondition(
      () => syncManager.isOtherInstanceRunning(),
      "waiting for child process to take the lock"
    ).catch(e => {
      // Rather than throwing out of waitForCondition(), catch and log the failure
      // manually so that we get output that's a bit more readable.
      Assert.ok(
        syncManager.isOtherInstanceRunning(),
        "child process has the lock"
      );
    });

    // The lock should have been closed when the process exited, but we'll allow
    // a little time for the OS to clean up the handle.
    await TestUtils.waitForCondition(
      () => !syncManager.isOtherInstanceRunning(),
      "waiting for child process to release the lock"
    ).catch(e => {
      Assert.ok(
        !syncManager.isOtherInstanceRunning(),
        "child process has released the lock"
      );
    });
  }

  doTestFinish();
});