@atproto/pds 0.5.12 → 0.5.14

package/tests/proxied/read-after-write.test.ts

@@ -1,382 +0,0 @@
-import assert from 'node:assert'
-import util from 'node:util'
-import { request } from 'undici'
-import {
-  AppBskyEmbedExternal,
-  AppBskyEmbedImages,
-  AppBskyEmbedRecord,
-  AppBskyFeedDefs,
-  AtpAgent,
-} from '@atproto/api'
-import { RecordRef, SeedClient, TestNetwork } from '@atproto/dev-env'
-import type { DidString } from '@atproto/syntax'
-import { app } from '../../src/lexicons/index.js'
-import basicSeed from '../seeds/basic.js'
-
-describe('proxy read after write', () => {
-  let network: TestNetwork
-  let agent: AtpAgent
-  let sc: SeedClient
-
-  let alice: DidString
-  let carol: DidString
-
-  beforeAll(async () => {
-    network = await TestNetwork.create({
-      dbPostgresSchema: 'proxy_read_after_write',
-    })
-    agent = network.pds.getAgent()
-    sc = network.getSeedClient()
-    await basicSeed(sc, { addModLabels: network.bsky })
-    alice = sc.dids.alice
-    carol = sc.dids.carol
-    await network.processAll()
-    await network.bsky.sub.stop()
-  })
-
-  beforeEach(async () => network.processAll())
-  afterAll(async () => network?.close())
-
-  it('handles read after write on profiles', async () => {
-    await sc.updateProfile(alice, { displayName: 'blah' })
-    const res = await agent.api.app.bsky.actor.getProfile(
-      { actor: alice },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    expect(res.data.displayName).toEqual('blah')
-    expect(res.data.description).toBeUndefined()
-  })
-
-  it('handles image formatting', async () => {
-    assert(network.pds.ctx.cfg.bskyAppView)
-    const blob = await sc.uploadFile(
-      alice,
-      '../dev-env/assets/key-landscape-small.jpg',
-      'image/jpeg',
-    )
-    await sc.updateProfile(alice, { displayName: 'blah', avatar: blob.image })
-
-    const res = await agent.api.app.bsky.actor.getProfile(
-      { actor: alice },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    expect(res.data.avatar).toEqual(
-      util.format(
-        network.pds.ctx.cfg.bskyAppView.cdnUrlPattern,
-        'avatar',
-        alice,
-        blob.image.ref.toString(),
-      ),
-    )
-  })
-
-  it('handles read after write on getAuthorFeed', async () => {
-    const res = await agent.api.app.bsky.feed.getAuthorFeed(
-      { actor: alice },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    for (const item of res.data.feed) {
-      if (item.post.author.did === alice) {
-        expect(item.post.author.displayName).toEqual('blah')
-      }
-    }
-  })
-
-  let replyRef1: RecordRef
-  let replyRef2: RecordRef
-
-  it('handles read after write on threads', async () => {
-    const reply1 = await sc.reply(
-      alice,
-      sc.posts[alice][0].ref,
-      sc.posts[alice][0].ref,
-      'another reply',
-    )
-    const reply2 = await sc.reply(
-      alice,
-      sc.posts[alice][0].ref,
-      reply1.ref,
-      'another another reply',
-    )
-    replyRef1 = reply1.ref
-    replyRef2 = reply2.ref
-    const res = await agent.api.app.bsky.feed.getPostThread(
-      { uri: sc.posts[alice][0].ref.uriStr },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    const thread = res.data.thread as app.bsky.feed.defs.ThreadViewPost
-    const layerOne = thread.replies as app.bsky.feed.defs.ThreadViewPost[]
-    expect(layerOne.length).toBe(1)
-    expect(layerOne[0].post.uri).toEqual(reply1.ref.uriStr)
-    const layerTwo = layerOne[0].replies as app.bsky.feed.defs.ThreadViewPost[]
-    expect(layerTwo.length).toBe(1)
-    expect(layerTwo[0].post.uri).toEqual(reply2.ref.uriStr)
-
-    const aliceHandle = sc.accounts[alice].handle
-    const handleUriStr = thread.post.uri.replace(alice, aliceHandle)
-    expect(handleUriStr).not.toEqual(thread.post.uri)
-    const handleRes = await agent.api.app.bsky.feed.getPostThread(
-      { uri: handleUriStr },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    expect(handleRes.data.thread).toEqual(res.data.thread)
-  })
-
-  it('handles read after write on a thread that is not found on appview', async () => {
-    const res = await agent.api.app.bsky.feed.getPostThread(
-      { uri: replyRef1.uriStr },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    const thread = res.data.thread as app.bsky.feed.defs.ThreadViewPost
-    expect(thread.post.uri).toEqual(replyRef1.uriStr)
-    expect(
-      (thread.parent as app.bsky.feed.defs.ThreadViewPost).post.uri,
-    ).toEqual(sc.posts[alice][0].ref.uriStr)
-    expect(thread.replies?.length).toEqual(1)
-    expect(
-      (thread.replies?.at(0) as app.bsky.feed.defs.ThreadViewPost).post.uri,
-    ).toEqual(replyRef2.uriStr)
-
-    const aliceHandle = sc.accounts[alice].handle
-    const handleUriStr = thread.post.uri.replace(alice, aliceHandle)
-    expect(handleUriStr).not.toEqual(thread.post.uri)
-    const handleRes = await agent.api.app.bsky.feed.getPostThread(
-      { uri: handleUriStr },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    expect(handleRes.data.thread).toEqual(res.data.thread)
-  })
-
-  it('handles read after write on threads with record embeds', async () => {
-    assert(network.pds.ctx.cfg.bskyAppView)
-    const img = await sc.uploadFile(
-      alice,
-      '../dev-env/assets/key-landscape-small.jpg',
-      'image/jpeg',
-    )
-    const replyRes1 = await agent.api.app.bsky.feed.post.create(
-      { repo: alice },
-      {
-        text: 'images test',
-        reply: {
-          root: sc.posts[alice][2].ref.raw,
-          parent: sc.posts[alice][2].ref.raw,
-        },
-        embed: {
-          $type: 'app.bsky.embed.images',
-          images: [
-            {
-              image: img.image,
-              aspectRatio: { height: 2, width: 1 },
-              alt: 'alt text',
-            },
-          ],
-        },
-        createdAt: new Date().toISOString(),
-      },
-      sc.getHeaders(alice),
-    )
-    const replyRes2 = await agent.api.app.bsky.feed.post.create(
-      { repo: alice },
-      {
-        text: 'external test',
-        reply: {
-          root: sc.posts[alice][2].ref.raw,
-          parent: {
-            uri: replyRes1.uri,
-            cid: replyRes1.cid,
-          },
-        },
-        embed: {
-          $type: 'app.bsky.embed.external',
-          external: {
-            uri: 'https://example.com',
-            title: 'TestImage',
-            description: 'testLink',
-            thumb: img.image,
-          },
-        },
-        createdAt: new Date().toISOString(),
-      },
-      sc.getHeaders(alice),
-    )
-
-    const res = await agent.api.app.bsky.feed.getPostThread(
-      { uri: sc.posts[alice][2].ref.uriStr },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    assert(AppBskyFeedDefs.isThreadViewPost(res.data.thread))
-    // @ts-ignore "pnpm verify:types" fails though VSCode doesn't complain
-    assert(res.data.thread.replies, 'replies is undefined')
-    // @ts-ignore "pnpm verify:types" fails though VSCode doesn't complain
-    const { replies } = res.data.thread
-    expect(replies.length).toBe(1)
-    assert(AppBskyFeedDefs.isThreadViewPost(replies[0]))
-    expect(replies[0].post.uri).toEqual(replyRes1.uri)
-    const { embed } = replies[0].post
-    assert(AppBskyEmbedImages.isView(embed))
-    expect(embed.images[0].fullsize).toEqual(
-      util.format(
-        network.pds.ctx.cfg.bskyAppView.cdnUrlPattern,
-        'feed_fullsize',
-        alice,
-        img.image.ref.toString(),
-      ),
-    )
-    expect(embed.images[0].aspectRatio).toEqual({ height: 2, width: 1 })
-    expect(embed.images[0].alt).toBe('alt text')
-    assert(replies[0].replies, 'replies[0].replies is undefined')
-    expect(replies[0].replies.length).toBe(1)
-    assert(AppBskyFeedDefs.isThreadViewPost(replies[0].replies[0]))
-    expect(replies[0].replies[0].post.uri).toEqual(replyRes2.uri)
-    const external = replies[0].replies[0].post.embed
-    assert(external)
-    assert(AppBskyEmbedExternal.isView(external))
-    expect(external.external.title).toEqual('TestImage')
-    expect(external.external.thumb).toEqual(
-      util.format(
-        network.pds.ctx.cfg.bskyAppView.cdnUrlPattern,
-        'feed_thumbnail',
-        alice,
-        img.image.ref.toString(),
-      ),
-    )
-  })
-
-  it('handles read after write on threads with record embeds', async () => {
-    const replyRes = await agent.api.app.bsky.feed.post.create(
-      { repo: alice },
-      {
-        text: 'blah',
-        reply: {
-          root: sc.posts[carol][0].ref.raw,
-          parent: sc.posts[carol][0].ref.raw,
-        },
-        embed: {
-          $type: 'app.bsky.embed.record',
-          record: sc.posts[alice][0].ref.raw,
-        },
-        createdAt: new Date().toISOString(),
-      },
-      sc.getHeaders(alice),
-    )
-    const res = await agent.api.app.bsky.feed.getPostThread(
-      { uri: sc.posts[carol][0].ref.uriStr },
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    assert(AppBskyFeedDefs.isThreadViewPost(res.data.thread))
-    assert(res.data.thread.replies, 'replies is undefined')
-    const { replies } = res.data.thread
-    expect(replies.length).toBe(1)
-    assert(AppBskyFeedDefs.isThreadViewPost(replies[0]))
-    expect(replies[0].post.uri).toEqual(replyRes.uri)
-    const embed = replies[0].post.embed
-    assert(AppBskyEmbedRecord.isView(embed))
-    const { record } = embed
-    assert('uri' in record) // @TODO: assert based in "$type"
-    expect(record.uri).toEqual(sc.posts[alice][0].ref.uriStr)
-  })
-
-  it('handles read after write on getTimeline', async () => {
-    const postRes = await agent.api.app.bsky.feed.post.create(
-      { repo: alice },
-      {
-        text: 'poast',
-        createdAt: new Date().toISOString(),
-      },
-      sc.getHeaders(alice),
-    )
-    const res = await agent.api.app.bsky.feed.getTimeline(
-      {},
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    expect(res.data.feed[0].post.uri).toEqual(postRes.uri)
-  })
-
-  it('returns lag headers', async () => {
-    const res = await agent.api.app.bsky.feed.getTimeline(
-      {},
-      { headers: { ...sc.getHeaders(alice) } },
-    )
-    const lag = res.headers['atproto-upstream-lag']
-    assert(lag !== undefined)
-    const parsed = parseInt(lag)
-    expect(parsed > 0).toBe(true)
-  })
-
-  it('negotiates encoding', async () => {
-    const identity = await agent.api.app.bsky.feed.getTimeline(
-      {},
-      { headers: { ...sc.getHeaders(alice), 'accept-encoding': 'identity' } },
-    )
-    expect(identity.headers['content-encoding']).toBeUndefined()
-
-    const gzip = await agent.api.app.bsky.feed.getTimeline(
-      {},
-      {
-        headers: { ...sc.getHeaders(alice), 'accept-encoding': 'gzip, *;q=0' },
-      },
-    )
-    expect(gzip.headers['content-encoding']).toBe('gzip')
-  })
-
-  it('defaults to identity encoding', async () => {
-    // Not using the "agent" because "fetch()" will add "accept-encoding: gzip,
-    // deflate" if not "accept-encoding" header is provided
-    const res = await request(
-      new URL(`/xrpc/app.bsky.feed.getTimeline`, agent.dispatchUrl),
-      {
-        headers: { ...sc.getHeaders(alice) },
-      },
-    )
-    expect(res.statusCode).toBe(200)
-    expect(res.headers['content-encoding']).toBeUndefined()
-  })
-
-  it('falls back to identity encoding', async () => {
-    const invalid = await agent.api.app.bsky.feed.getTimeline(
-      {},
-      { headers: { ...sc.getHeaders(alice), 'accept-encoding': 'invalid' } },
-    )
-
-    expect(invalid.headers['content-encoding']).toBeUndefined()
-  })
-
-  it('errors when failing to negotiate encoding', async () => {
-    await expect(
-      agent.api.app.bsky.feed.getTimeline(
-        {},
-        {
-          headers: {
-            ...sc.getHeaders(alice),
-            'accept-encoding': 'invalid, *;q=0',
-          },
-        },
-      ),
-    ).rejects.toThrow(
-      expect.objectContaining({
-        status: 406,
-        message: 'this service does not support any of the requested encodings',
-      }),
-    )
-  })
-
-  it('errors on invalid content-encoding format', async () => {
-    await expect(
-      agent.api.app.bsky.feed.getTimeline(
-        {},
-        {
-          headers: {
-            ...sc.getHeaders(alice),
-            'accept-encoding': ';q=1',
-          },
-        },
-      ),
-    ).rejects.toThrow(
-      expect.objectContaining({
-        status: 400,
-        message: 'Invalid accept-encoding: ";q=1"',
-      }),
-    )
-  })
-})