@atproto/ozone 0.0.2

package/tests/repo-search.test.ts

@@ -0,0 +1,123 @@
+import { SeedClient, TestNetwork, usersBulkSeed } from '@atproto/dev-env'
+import AtpAgent from '@atproto/api'
+import { paginateAll } from './_util'
+
+describe('admin repo search view', () => {
+  let network: TestNetwork
+  let agent: AtpAgent
+  let sc: SeedClient
+  let headers: { [s: string]: string }
+
+  beforeAll(async () => {
+    network = await TestNetwork.create({
+      dbPostgresSchema: 'ozone_admin_repo_search',
+    })
+    agent = network.pds.getClient()
+    sc = network.getSeedClient()
+    await usersBulkSeed(sc)
+    headers = network.pds.adminAuthHeaders()
+    await network.processAll()
+  })
+
+  afterAll(async () => {
+    await network.close()
+  })
+
+  beforeAll(async () => {
+    await sc.emitModerationEvent({
+      event: { $type: 'com.atproto.admin.defs#modEventTakedown' },
+      subject: {
+        $type: 'com.atproto.admin.defs#repoRef',
+        did: sc.dids['cara-wiegand69.test'],
+      },
+    })
+  })
+
+  it('gives relevant results', async () => {
+    const result = await agent.api.com.atproto.admin.searchRepos(
+      { term: 'car' },
+      { headers },
+    )
+
+    const handles = result.data.repos.map((u) => u.handle)
+
+    const shouldContain = [
+      'cara-wiegand69.test', // Present despite repo takedown
+      'carlos6.test',
+      'carolina-mcdermott77.test',
+    ]
+
+    shouldContain.forEach((handle) => expect(handles).toContain(handle))
+
+    const shouldNotContain = [
+      'sven70.test',
+      'hilario84.test',
+      'santa-hermann78.test',
+      'dylan61.test',
+      'preston-harris.test',
+      'loyce95.test',
+      'melyna-zboncak.test',
+    ]
+
+    shouldNotContain.forEach((handle) => expect(handles).not.toContain(handle))
+  })
+
+  it('finds repo by did', async () => {
+    const term = sc.dids['cara-wiegand69.test']
+    const res = await agent.api.com.atproto.admin.searchRepos(
+      { term },
+      { headers },
+    )
+
+    expect(res.data.repos.length).toEqual(1)
+    expect(res.data.repos[0].did).toEqual(term)
+  })
+
+  it('paginates with term', async () => {
+    const results = (results) => results.flatMap((res) => res.users)
+    const paginator = async (cursor?: string) => {
+      const res = await agent.api.com.atproto.admin.searchRepos(
+        { term: 'p', cursor, limit: 3 },
+        { headers },
+      )
+      return res.data
+    }
+
+    const paginatedAll = await paginateAll(paginator)
+    paginatedAll.forEach((res) =>
+      expect(res.repos.length).toBeLessThanOrEqual(3),
+    )
+
+    const full = await agent.api.com.atproto.admin.searchRepos(
+      { term: 'p' },
+      { headers },
+    )
+
+    expect(full.data.repos.length).toBeGreaterThan(3)
+    expect(results(paginatedAll)).toEqual(results([full.data]))
+  })
+
+  it('paginates without term', async () => {
+    const results = (results) => results.flatMap((res) => res.repos)
+    const paginator = async (cursor?: string) => {
+      const res = await agent.api.com.atproto.admin.searchRepos(
+        { cursor, limit: 3 },
+        { headers },
+      )
+      return res.data
+    }
+
+    const paginatedAll = await paginateAll(paginator, 5)
+    paginatedAll.forEach((res) =>
+      expect(res.repos.length).toBeLessThanOrEqual(3),
+    )
+
+    const full = await agent.api.com.atproto.admin.searchRepos(
+      { limit: 15 },
+      { headers },
+    )
+
+    expect(full.data.repos.length).toEqual(15)
+    expect(results(paginatedAll)).toEqual(results([full.data]))
+  })
+})

package/tests/server.test.ts

@@ -0,0 +1,76 @@
+import { AddressInfo } from 'net'
+import express from 'express'
+import axios, { AxiosError } from 'axios'
+import { TestNetwork } from '@atproto/dev-env'
+import { handler as errorHandler } from '../src/error'
+import { TestOzone } from '@atproto/dev-env/src/ozone'
+
+describe('server', () => {
+  let network: TestNetwork
+  let ozone: TestOzone
+
+  beforeAll(async () => {
+    network = await TestNetwork.create({
+      dbPostgresSchema: 'ozone_server',
+    })
+    ozone = network.ozone
+  })
+
+  afterAll(async () => {
+    await network.close()
+  })
+
+  it('preserves 404s.', async () => {
+    const promise = axios.get(`${ozone.url}/unknown`)
+    await expect(promise).rejects.toThrow('failed with status code 404')
+  })
+
+  it('error handler turns unknown errors into 500s.', async () => {
+    const app = express()
+    app.get('/oops', () => {
+      throw new Error('Oops!')
+    })
+    app.use(errorHandler)
+    const srv = app.listen()
+    const port = (srv.address() as AddressInfo).port
+    const promise = axios.get(`http://localhost:${port}/oops`)
+    await expect(promise).rejects.toThrow('failed with status code 500')
+    srv.close()
+    try {
+      await promise
+    } catch (err: unknown) {
+      const axiosError = err as AxiosError
+      expect(axiosError.response?.status).toEqual(500)
+      expect(axiosError.response?.data).toEqual({
+        error: 'InternalServerError',
+        message: 'Internal Server Error',
+      })
+    }
+  })
+
+  it('healthcheck succeeds when database is available.', async () => {
+    const { data, status } = await axios.get(`${ozone.url}/xrpc/_health`)
+    expect(status).toEqual(200)
+    expect(data).toEqual({ version: '0.0.0' })
+  })
+
+  it('healthcheck fails when database is unavailable.', async () => {
+    await ozone.ctx.db.close()
+    let error: AxiosError
+    try {
+      await axios.get(`${ozone.url}/xrpc/_health`)
+      throw new Error('Healthcheck should have failed')
+    } catch (err) {
+      if (axios.isAxiosError(err)) {
+        error = err
+      } else {
+        throw err
+      }
+    }
+    expect(error.response?.status).toEqual(503)
+    expect(error.response?.data).toEqual({
+      version: '0.0.0',
+      error: 'Service Unavailable',
+    })
+  })
+})

package/tsconfig.build.json

@@ -0,0 +1,4 @@
+{
+  "extends": "./tsconfig.json",
+  "exclude": ["**/*.spec.ts", "**/*.test.ts"]
+}

package/tsconfig.json

@@ -0,0 +1,21 @@
+{
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "rootDir": "./src",
+    "outDir": "./dist",
+    "emitDeclarationOnly": true
+  },
+  "module": "nodenext",
+  "include": ["./src", "__tests__/**/**.ts"],
+  "references": [
+    { "path": "../api/tsconfig.build.json" },
+    { "path": "../common/tsconfig.build.json" },
+    { "path": "../crypto/tsconfig.build.json" },
+    { "path": "../identifier/tsconfig.build.json" },
+    { "path": "../lexicon/tsconfig.build.json" },
+    { "path": "../lex-cli/tsconfig.build.json" },
+    { "path": "../repo/tsconfig.build.json" },
+    { "path": "../uri/tsconfig.build.json" },
+    { "path": "../xrpc-server/tsconfig.build.json" }
+  ]
+}