@atproto/pds 0.4.217 → 0.4.219

package/tests/_puppeteer.ts

@@ -1,27 +1,71 @@
 import assert from 'node:assert'
-import { type Browser, type Page } from 'puppeteer'
+import { type Browser, Handler, type Page, Target, TargetType } from 'puppeteer'
 
 export class PageHelper implements AsyncDisposable {
   constructor(protected readonly page: Page) {}
 
-  async goto(url: string) {
-    await this.page.goto(url)
+  async goto(url: string | URL) {
+    await this.page.goto(url.toString())
+  }
+
+  isClosed() {
+    return this.page.isClosed()
+  }
+
+  async title() {
+    await this.waitForNetworkIdle()
+    return this.page.title()
   }
 
   async waitForNetworkIdle() {
     await this.page.waitForNetworkIdle()
   }
 
-  async navigationAction(run: () => Promise<unknown>): Promise<void> {
-    const promise = this.page.waitForNavigation()
+  async waitForPopup(run: () => Promise<unknown>): Promise<PageHelper> {
+    const browser = this.page.browser()
+    const popupPromise = new Promise<Page | null>((resolve, reject) => {
+      const cleanup = () => {
+        clearTimeout(timeout)
+        browser.off('targetcreated', targetcreated)
+      }
+
+      const timeout = setTimeout(() => {
+        cleanup()
+        reject(new Error('Timeout waiting for popup'))
+      }, 5_000)
+
+      const targetcreated: Handler<Target> = async (target) => {
+        switch (target.type()) {
+          case TargetType.BACKGROUND_PAGE:
+          case TargetType.PAGE: {
+            cleanup()
+            resolve(target.page())
+          }
+        }
+      }
+
+      browser.once('targetcreated', targetcreated)
+    })
+
+    await run()
+    const popup = await popupPromise
+    assert(popup, 'Popup page not found')
+
+    return new PageHelper(popup)
+  }
+
+  async navigationAction(run: () => unknown | Promise<unknown>): Promise<void> {
+    const promise = this.page.waitForNavigation({ timeout: 10_000 })
     await run()
     await promise
-    await this.waitForNetworkIdle()
   }
 
-  async checkTitle(expected: string) {
-    await this.waitForNetworkIdle()
-    await expect(this.page.title()).resolves.toBe(expected)
+  async navigationClick(text: string, tag = 'button') {
+    return this.navigationAction(() => this.clickOnText(text, tag))
+  }
+
+  async assertTitle(expected: string) {
+    await expect(this.title()).resolves.toBe(expected)
   }
 
   async clickOn(selector: string) {
@@ -30,8 +74,12 @@ export class PageHelper implements AsyncDisposable {
     return elementHandle
   }
 
-  async clickOnButton(text: string) {
-    return this.clickOn(`button::-p-text(${JSON.stringify(text)})`)
+  async clickOnText(text: string, tag = 'button') {
+    return this.clickOn(`${tag}::-p-text(${JSON.stringify(text)})`)
+  }
+
+  async clickOnAriaLabel(label: string, tag = 'button') {
+    return this.clickOn(`${tag}[aria-label=${JSON.stringify(label)}]`)
   }
 
   async typeIn(selector: string, text: string) {
@@ -46,16 +94,22 @@ export class PageHelper implements AsyncDisposable {
   }
 
   async ensureTextVisibility(text: string, tag = 'p') {
-    await this.page.waitForSelector(`${tag}::-p-text(${JSON.stringify(text)})`)
+    await this.page.waitForSelector(
+      `${tag}::-p-text(${JSON.stringify(text)})`,
+      {
+        visible: true,
+        timeout: 5_000,
+      },
+    )
   }
 
   protected async getVisibleElement(selector: string) {
-    const elementHandle = await this.page.waitForSelector(selector)
-
-    expect(elementHandle).not.toBeNull()
-    assert(elementHandle)
+    const elementHandle = await this.page.waitForSelector(selector, {
+      visible: true,
+      timeout: 5_000,
+    })
 
-    await expect(elementHandle.isVisible()).resolves.toBe(true)
+    assert(elementHandle, `Element not found: ${selector}`)
 
     return elementHandle
   }

package/tests/file-uploads.test.ts

@@ -1,10 +1,15 @@
+import assert from 'node:assert'
 import fs from 'node:fs/promises'
 import { gzipSync } from 'node:zlib'
 import * as uint8arrays from 'uint8arrays'
 import { randomBytes } from '@atproto/crypto'
 import { SeedClient, TestNetworkNoAppView } from '@atproto/dev-env'
 import { Client, DidString } from '@atproto/lex'
-import { BlobRef } from '@atproto/lex-data'
+import {
+  TypedBlobRef,
+  getBlobCidString,
+  isTypedBlobRef,
+} from '@atproto/lex-data'
 import { AppContext } from '../src'
 import { ActorDb } from '../src/actor-store/db'
 import { DiskBlobStore } from '../src/disk-blobstore'
@@ -40,7 +45,7 @@ describe('file uploads', () => {
     await network.close()
   })
 
-  let smallBlob: BlobRef
+  let smallBlob: TypedBlobRef
   let smallFile: Uint8Array
 
   it('handles client abort', async () => {
@@ -78,6 +83,7 @@ describe('file uploads', () => {
       headers: sc.getHeaders(alice),
       encoding: 'image/jpeg',
     })
+    assert(isTypedBlobRef(res.body.blob))
     smallBlob = res.body.blob
 
     const found = await aliceDb.db
@@ -123,7 +129,7 @@ describe('file uploads', () => {
     expect(uint8arrays.equals(smallFile, body)).toBeTruthy()
   })
 
-  let largeBlob: BlobRef
+  let largeBlob: TypedBlobRef
   let largeFile: Uint8Array
 
   it('does not allow referencing a file that is outside blob constraints', async () => {
@@ -132,6 +138,7 @@ describe('file uploads', () => {
       headers: sc.getHeaders(alice),
       encoding: 'image/jpeg',
     })
+    assert(isTypedBlobRef(res.body.blob))
     largeBlob = res.body.blob
 
     const profilePromise = sc.updateProfile(alice, {
@@ -195,7 +202,7 @@ describe('file uploads', () => {
     const blob = await aliceDb.db
       .selectFrom('blob')
       .selectAll()
-      .where('cid', '=', uploadAfterPermanent.blob.ref.toString())
+      .where('cid', '=', getBlobCidString(uploadAfterPermanent.blob))
       .executeTakeFirstOrThrow()
     expect(blob.tempKey).toEqual(null)
   })
@@ -208,6 +215,7 @@ describe('file uploads', () => {
         'content-encoding': 'gzip',
       },
     })
+    assert(isTypedBlobRef(uploaded.blob))
     expect(uploaded.blob.ref.equals(smallBlob.ref)).toBeTruthy()
   })
 
@@ -221,7 +229,7 @@ describe('file uploads', () => {
     const found = await aliceDb.db
       .selectFrom('blob')
       .selectAll()
-      .where('cid', '=', res.body.blob.ref.toString())
+      .where('cid', '=', getBlobCidString(res.body.blob))
       .executeTakeFirst()
 
     expect(found?.mimeType).toBe('image/jpeg')
@@ -237,7 +245,7 @@ describe('file uploads', () => {
     const found = await aliceDb.db
       .selectFrom('blob')
       .selectAll()
-      .where('cid', '=', res.body.blob.ref.toString())
+      .where('cid', '=', getBlobCidString(res.body.blob))
       .executeTakeFirst()
 
     expect(found?.mimeType).toBe('image/png')
@@ -253,7 +261,7 @@ describe('file uploads', () => {
     const found = await aliceDb.db
       .selectFrom('blob')
       .selectAll()
-      .where('cid', '=', res.body.blob.ref.toString())
+      .where('cid', '=', getBlobCidString(res.body.blob))
       .executeTakeFirst()
 
     expect(found?.mimeType).toBe('test/fake')
@@ -269,7 +277,7 @@ describe('file uploads', () => {
     const found = await aliceDb.db
       .selectFrom('blob')
       .selectAll()
-      .where('cid', '=', res.body.blob.ref.toString())
+      .where('cid', '=', getBlobCidString(res.body.blob))
       .executeTakeFirst()
 
     expect(found?.mimeType).toBe('text/plain')
@@ -285,7 +293,7 @@ describe('file uploads', () => {
     const found = await aliceDb.db
       .selectFrom('blob')
       .selectAll()
-      .where('cid', '=', res.body.blob.ref.toString())
+      .where('cid', '=', getBlobCidString(res.body.blob))
       .executeTakeFirst()
 
     expect(found?.mimeType).toBe('application/json')

package/tests/oauth.test.ts

@@ -1,14 +1,9 @@
 import { once } from 'node:events'
-import {
-  IncomingMessage,
-  Server,
-  ServerResponse,
-  createServer,
-} from 'node:http'
+import { Server, createServer } from 'node:http'
 import { AddressInfo } from 'node:net'
 import { type Browser, launch } from 'puppeteer'
 import { TestNetworkNoAppView } from '@atproto/dev-env'
-import files from '@atproto/oauth-client-browser-example' with { type: 'json' }
+import { oauthClientAssetsMiddleware } from './_oauth_client_assets_middleware.js'
 import { PageHelper } from './_puppeteer.js'
 
 describe('oauth', () => {
@@ -44,7 +39,7 @@ describe('oauth', () => {
       password: 'alice-pass',
     })
 
-    server = createServer(clientHandler)
+    server = createServer(oauthClientAssetsMiddleware)
     server.listen(0)
     await once(server, 'listening')
 
@@ -67,26 +62,24 @@ describe('oauth', () => {
 
   // This uses prompt=create under the hood:
   it('Allows to sign-up through OAuth', async () => {
-    const page = await PageHelper.from(browser, { languages })
+    await using page = await PageHelper.from(browser, { languages })
 
     await page.goto(appUrl)
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
-    await page.navigationAction(async () => {
-      await page.clickOnButton(`Sign up with ${new URL(network.pds.url).host}`)
-    })
+    await page.navigationClick(`Sign up with ${new URL(network.pds.url).host}`)
 
-    await page.checkTitle('Créer un compte')
+    await page.assertTitle("S'inscrire")
 
     await page.typeInInput('handle', 'bob')
 
-    await page.clickOnButton('Suivant')
+    await page.clickOnText('Suivant')
 
     await page.typeInInput('email', 'bob@test.com')
     await page.typeInInput('password', 'bob-pass')
 
-    await page.clickOnButton("S'inscrire")
+    await page.clickOnText("S'inscrire")
 
     await page.ensureTextVisibility(
       `L'application demande un contrôle total sur votre identité, ce qui signifie qu'elle pourrait casser de façon permanente, ou même usurper, votre compte. N'authorisez l'accès qu'aux applications auxquelles vous faites vraiment confiance.`,
@@ -96,54 +89,38 @@ describe('oauth', () => {
     // the client to resolve the account's did
     await network.processAll()
 
-    await page.navigationAction(async () => {
-      await page.clickOnButton("Authoriser l'accès")
-    })
+    await page.navigationClick('Autoriser')
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
     await page.ensureTextVisibility('Token info', 'h2')
 
-    await page.clickOn('button[aria-label="User menu"]')
+    await page.clickOnAriaLabel('User menu')
 
-    await page.clickOnButton('Sign out')
+    await page.clickOnText('Sign out')
 
     await page.waitForNetworkIdle()
-
-    // TODO: Find out why we can't use "using" here
-    await page[Symbol.asyncDispose]()
   })
 
-  it('Allows login or signup through OAuth via a choice', async () => {
-    const page = await PageHelper.from(browser, { languages })
+  it('Allows canceling the OAuth flow', async () => {
+    await using page = await PageHelper.from(browser, { languages })
 
     await page.goto(appUrl)
 
-    await page.checkTitle('OAuth Client Example')
-
-    await page.navigationAction(async () => {
-      await page.clickOnButton(`Login with ${new URL(network.pds.url).host}`)
-    })
+    await page.assertTitle('OAuth Client Example')
 
-    await page.checkTitle('Authentification')
+    await page.navigationClick(`Login with ${new URL(network.pds.url).host}`)
 
-    await page.ensureTextVisibility('Annuler', 'button')
-    await page.ensureTextVisibility('Se connecter', 'button')
-    await page.ensureTextVisibility('Créer un nouveau compte', 'button')
+    await page.assertTitle("S'identifier")
 
     // Cancel the OAuth flow:
-    await page.navigationAction(async () => {
-      await page.clickOnButton('Annuler')
-    })
+    await page.navigationClick('Annuler')
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
     await page.ensureTextVisibility('Login with the Atmosphere', 'h2')
 
     await page.waitForNetworkIdle()
-
-    // TODO: Find out why we can't use "using" here
-    await page[Symbol.asyncDispose]()
   })
 
   it('allows resetting the password', async () => {
@@ -153,31 +130,29 @@ describe('oauth', () => {
         // noop
       })
 
-    const page = await PageHelper.from(browser, { languages })
+    await using page = await PageHelper.from(browser, { languages })
 
     await page.goto(appUrl)
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
-    await page.navigationAction(async () => {
-      const input = await page.typeIn('input[name="identifier"]', 'alice.test')
+    const input = await page.typeInInput('identifier', 'alice.test')
 
-      await input.press('Enter')
-    })
+    await page.navigationAction(async () => input.press('Enter'))
 
-    await page.checkTitle('Connexion')
+    await page.assertTitle('Connexion')
 
-    await page.clickOnButton('Oublié ?')
+    await page.clickOnText('Oublié ?')
 
-    await page.checkTitle('Mot de passe oublié')
+    await page.assertTitle('Mot de passe oublié')
 
     await page.typeInInput('email', 'alice@test.com')
 
     expect(sendTemplateMock).toHaveBeenCalledTimes(0)
 
-    await page.clickOnButton('Suivant')
+    await page.clickOnText('Suivant')
 
-    await page.checkTitle('Réinitialiser le mot de passe')
+    await page.assertTitle('Réinitialiser le mot de passe')
 
     expect(sendTemplateMock).toHaveBeenCalledTimes(1)
 
@@ -191,34 +166,29 @@ describe('oauth', () => {
 
     await page.typeInInput('password', 'alice-new-pass')
 
-    await page.clickOnButton('Suivant')
+    await page.clickOnText('Suivant')
 
-    await page.checkTitle('Mot de passe mis à jour')
+    await page.assertTitle('Mot de passe mis à jour')
 
     await page.ensureTextVisibility('Mot de passe mis à jour !', 'h2')
 
-    // TODO: Find out why we can't use "using" here
-    await page[Symbol.asyncDispose]()
-
     sendTemplateMock.mockRestore()
   })
 
   it('Allows to sign-in through OAuth', async () => {
-    const page = await PageHelper.from(browser, { languages })
+    await using page = await PageHelper.from(browser, { languages })
 
     await page.goto(appUrl)
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
-    await page.navigationAction(async () => {
-      const input = await page.typeIn('input[name="identifier"]', 'alice.test')
+    const input = await page.typeInInput('identifier', 'alice.test')
 
-      await input.press('Enter')
-    })
+    await page.navigationAction(async () => input.press('Enter'))
 
-    await page.checkTitle('Connexion')
+    await page.assertTitle('Connexion')
 
-    await page.typeIn('input[type="password"]', 'alice-new-pass')
+    await page.typeInInput('password', 'alice-new-pass')
 
     // Make sure the warning is visible
     await page.ensureTextVisibility('Avertissement', 'h3')
@@ -227,79 +197,46 @@ describe('oauth', () => {
       'label::-p-text(Se souvenir de ce compte sur cet appareil)',
     )
 
-    await page.clickOnButton('Se connecter')
+    await page.clickOnText('Se connecter')
 
-    await page.checkTitle("Authoriser l'accès")
+    await page.assertTitle('Autoriser')
 
-    await page.navigationAction(async () => {
-      await page.clickOnButton("Authoriser l'accès")
-    })
+    await page.navigationClick('Autoriser')
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
     await page.ensureTextVisibility('Token info', 'h2')
 
-    await page.clickOn('button[aria-label="User menu"]')
+    await page.clickOnAriaLabel('User menu')
 
-    await page.clickOnButton('Sign out')
+    await page.clickOnText('Sign out')
 
     await page.waitForNetworkIdle()
-
-    // TODO: Find out why we can't use "using" here
-    await page[Symbol.asyncDispose]()
   })
 
   it('remembers the session', async () => {
-    const page = await PageHelper.from(browser, { languages })
+    await using page = await PageHelper.from(browser, { languages })
 
     await page.goto(appUrl)
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
-    await page.navigationAction(async () => {
-      const input = await page.typeIn('input[name="identifier"]', 'alice.test')
+    const input = await page.typeInInput('identifier', 'alice.test')
 
-      await input.press('Enter')
-    })
+    await page.navigationAction(async () => input.press('Enter'))
 
-    await page.checkTitle("Authoriser l'accès")
+    await page.assertTitle('Autoriser')
 
-    await page.navigationAction(async () => {
-      await page.clickOnButton("Authoriser l'accès")
-    })
+    await page.navigationClick('Autoriser')
 
-    await page.checkTitle('OAuth Client Example')
+    await page.assertTitle('OAuth Client Example')
 
     await page.ensureTextVisibility('Token info', 'h2')
 
-    await page.clickOn('button[aria-label="User menu"]')
+    await page.clickOnAriaLabel('User menu')
 
-    await page.clickOnButton('Sign out')
+    await page.clickOnText('Sign out')
 
     await page.waitForNetworkIdle()
-
-    // TODO: Find out why we can't use "using" here
-    await page[Symbol.asyncDispose]()
   })
 })
-
-function clientHandler(
-  req: IncomingMessage,
-  res: ServerResponse,
-  next?: (err?: unknown) => void,
-): void {
-  const path = req.url?.split('?')[0].slice(1) || 'index.html'
-  const file = Object.hasOwn(files, path) ? files[path] : null
-
-  if (file) {
-    res
-      .writeHead(200, 'OK', { 'content-type': file.mime })
-      .end(Buffer.from(file.data, 'base64'))
-  } else if (next) {
-    next()
-  } else {
-    res
-      .writeHead(404, 'Not Found', { 'content-type': 'text/plain' })
-      .end('Page not found')
-  }
-}