This repository has been archived by the owner on Sep 11, 2024. It is now read-only.
-
-
Notifications
You must be signed in to change notification settings - Fork 827
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add a playwright test for migration progress
- Loading branch information
Showing
6 changed files
with
72,092 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,72 @@ | ||
/* | ||
Copyright 2023-2024 The Matrix.org Foundation C.I.C. | ||
Licensed under the Apache License, Version 2.0 (the "License"); | ||
you may not use this file except in compliance with the License. | ||
You may obtain a copy of the License at | ||
http://www.apache.org/licenses/LICENSE-2.0 | ||
Unless required by applicable law or agreed to in writing, software | ||
distributed under the License is distributed on an "AS IS" BASIS, | ||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
See the License for the specific language governing permissions and | ||
limitations under the License. | ||
*/ | ||
|
||
import path from "path"; | ||
import { readFile } from "node:fs/promises"; | ||
|
||
import { expect, test as base } from "../../element-web-test"; | ||
|
||
const test = base.extend({ | ||
// Replace the `user` fixture with one which populates the indexeddb data before starting the app. | ||
user: async ({ context, pageWithCredentials: page, credentials }, use) => { | ||
await page.route(`/test_indexeddb_cryptostore_dump/*`, async (route, request) => { | ||
const resourcePath = path.join(__dirname, new URL(request.url()).pathname); | ||
const body = await readFile(resourcePath, { encoding: "utf-8" }); | ||
await route.fulfill({ body }); | ||
}); | ||
await page.goto("/test_indexeddb_cryptostore_dump/index.html"); | ||
|
||
await use(credentials); | ||
}, | ||
}); | ||
|
||
test.describe("migration", function () { | ||
test.use({ displayName: "Alice" }); | ||
|
||
test("Should support migration from legacy crypto", async ({ context, user, page }, workerInfo) => { | ||
test.skip(workerInfo.project.name === "Legacy Crypto", "This test only works with Rust crypto."); | ||
test.slow(); | ||
|
||
// We should see a migration progress bar | ||
await page.getByText("Hang tight.").waitFor({ timeout: 60000 }); | ||
|
||
// When the progress bar first loads, it should have a high max (one per megolm session to import), and | ||
// a relatively low value. | ||
const progressBar = page.getByRole("progressbar"); | ||
const initialProgress = parseFloat(await progressBar.getAttribute("value")); | ||
const initialMax = parseFloat(await progressBar.getAttribute("max")); | ||
expect(initialMax).toBeGreaterThan(4000); | ||
expect(initialProgress).toBeGreaterThanOrEqual(0); | ||
expect(initialProgress).toBeLessThanOrEqual(500); | ||
|
||
// Later, the progress should pass 50% | ||
await expect | ||
.poll( | ||
async () => { | ||
const progressBar = page.getByRole("progressbar"); | ||
return ( | ||
(parseFloat(await progressBar.getAttribute("value")) * 100.0) / | ||
parseFloat(await progressBar.getAttribute("max")) | ||
); | ||
}, | ||
{ timeout: 60000 }, | ||
) | ||
.toBeGreaterThan(50); | ||
|
||
// Eventually, we should get a normal matrix chat | ||
await page.waitForSelector(".mx_MatrixChat", { timeout: 120000 }); | ||
}); | ||
}); |
60 changes: 60 additions & 0 deletions
60
playwright/e2e/crypto/test_indexeddb_cryptostore_dump/README.md
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,60 @@ | ||
# Dump of libolm indexeddb cryptostore | ||
|
||
This directory contains, in `dump.json`, a dump of a real indexeddb store from a session using | ||
libolm crypto. | ||
|
||
The corresponding pickle key is `+1k2Ppd7HIisUY824v7JtV3/oEE4yX0TqtmNPyhaD7o`. | ||
|
||
This directory also contains, in `index.html` and `load.js`, a page which will populate indexeddb with the data | ||
(and the pickle key). This can be served via a Playwright [Route](https://playwright.dev/docs/api/class-route) so as to | ||
populate the indexeddb before the main application loads. Note that encrypting the pickle key requires the test User ID | ||
and Device ID, so they must be stored in `localstorage` before loading `index.html`. | ||
|
||
## Creation of the dump file | ||
|
||
The dump was created by pasting the following into the browser console: | ||
|
||
```javascript | ||
async function exportIndexedDb(name) { | ||
const db = await new Promise((resolve, reject) => { | ||
const dbReq = indexedDB.open(name); | ||
dbReq.onerror = reject; | ||
dbReq.onsuccess = () => resolve(dbReq.result); | ||
}); | ||
|
||
const storeNames = db.objectStoreNames; | ||
const exports = {}; | ||
for (const store of storeNames) { | ||
exports[store] = []; | ||
const txn = db.transaction(store, "readonly"); | ||
const objectStore = txn.objectStore(store); | ||
await new Promise((resolve, reject) => { | ||
const cursorReq = objectStore.openCursor(); | ||
cursorReq.onerror = reject; | ||
cursorReq.onsuccess = (event) => { | ||
const cursor = event.target.result; | ||
if (cursor) { | ||
const entry = { value: cursor.value }; | ||
if (!objectStore.keyPath) { | ||
entry.key = cursor.key; | ||
} | ||
exports[store].push(entry); | ||
cursor.continue(); | ||
} else { | ||
resolve(); | ||
} | ||
}; | ||
}); | ||
} | ||
return exports; | ||
} | ||
|
||
window.saveAs( | ||
new Blob([JSON.stringify(await exportIndexedDb("matrix-js-sdk:crypto"), null, 2)], { | ||
type: "application/json;charset=utf-8", | ||
}), | ||
"dump.json", | ||
); | ||
``` | ||
|
||
The pickle key is extracted via `mxMatrixClientPeg.get().crypto.olmDevice.pickleKey`. |
Oops, something went wrong.