summaryrefslogtreecommitdiffstats
path: root/toolkit/components/places/tests/unit/test_missing_builtin_folders.js
blob: 826cc37ff7b42968e3343938c341f7a33c3d2a1a (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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/* Any copyright is dedicated to the Public Domain.
   http://creativecommons.org/publicdomain/zero/1.0/ */

"use strict";

/**
 * This file tests that a missing built-in folders (child of root) are correctly
 * fixed when the database is loaded.
 */

const ALL_ROOT_GUIDS = [
  PlacesUtils.bookmarks.menuGuid,
  PlacesUtils.bookmarks.unfiledGuid,
  PlacesUtils.bookmarks.toolbarGuid,
  PlacesUtils.bookmarks.tagsGuid,
  PlacesUtils.bookmarks.mobileGuid,
];

const INITIAL_ROOT_GUIDS = [
  PlacesUtils.bookmarks.menuGuid,
  PlacesUtils.bookmarks.tagsGuid,
  PlacesUtils.bookmarks.unfiledGuid,
];

add_task(async function setup() {
  // This file has the toolbar and mobile folders missing.
  await setupPlacesDatabase("missingBuiltIn.sqlite");

  // Check database contents to be migrated.
  let path = PathUtils.join(PathUtils.profileDir, DB_FILENAME);
  let db = await Sqlite.openConnection({ path });

  let rows = await db.execute(
    `
    SELECT guid FROM moz_bookmarks
    WHERE parent = (SELECT id from moz_bookmarks WHERE guid = :guid)
  `,
    {
      guid: PlacesUtils.bookmarks.rootGuid,
    }
  );

  let guids = rows.map(row => row.getResultByName("guid"));
  Assert.deepEqual(
    guids,
    INITIAL_ROOT_GUIDS,
    "Initial database should have only the expected GUIDs"
  );

  await db.close();
});

add_task(async function test_database_recreates_roots() {
  Assert.ok(
    PlacesUtils.history.databaseStatus ==
      PlacesUtils.history.DATABASE_STATUS_OK ||
      PlacesUtils.history.databaseStatus ==
        PlacesUtils.history.DATABASE_STATUS_UPGRADED,
    "Should successfully access the database for the first time"
  );

  let db = await PlacesUtils.promiseDBConnection();
  let rootId = await PlacesUtils.promiseItemId(PlacesUtils.bookmarks.rootGuid);
  for (let guid of ALL_ROOT_GUIDS) {
    let rows = await db.execute(
      `
      SELECT id, parent FROM moz_bookmarks
      WHERE guid = :guid
    `,
      { guid }
    );

    Assert.equal(rows.length, 1, "Should have exactly one row for the root");

    Assert.equal(
      rows[0].getResultByName("parent"),
      rootId,
      "Should have been created with the correct parent"
    );

    let root = await PlacesUtils.bookmarks.fetch(guid);

    Assert.equal(root.guid, guid, "GUIDs should match");
    Assert.equal(
      root.parentGuid,
      PlacesUtils.bookmarks.rootGuid,
      "Should have the correct parent GUID"
    );
    Assert.equal(
      root.type,
      PlacesUtils.bookmarks.TYPE_FOLDER,
      "Should have the correct type"
    );

    let id = rows[0].getResultByName("id");
    Assert.equal(
      await PlacesUtils.promiseItemId(guid),
      id,
      "Should return the correct id from promiseItemId"
    );
    Assert.equal(
      await PlacesUtils.promiseItemGuid(id),
      guid,
      "Should return the correct guid from promiseItemGuid"
    );
  }

  let rows = await db.execute(
    `
    SELECT 1 FROM moz_bookmarks
    WHERE parent = (SELECT id from moz_bookmarks WHERE guid = :guid)
  `,
    {
      guid: PlacesUtils.bookmarks.rootGuid,
    }
  );

  Assert.equal(
    rows.length,
    ALL_ROOT_GUIDS.length,
    "Root folder should have the expected number of children"
  );
});