twitter-lite/tests/e2e/mock-relay.mjs

import { createServer } from 'node:http'

const portValue = process.env.TWITTER_LITE_MOCK_RELAY_PORT
const port = Number(portValue)

if (!portValue || !Number.isInteger(port) || port < 1) {
  throw new Error('TWITTER_LITE_MOCK_RELAY_PORT must be a valid port')
}

const failedRetryQueries = new Set()

const isRecord = (value) =>
  value !== null && typeof value === 'object' && !Array.isArray(value)

const tweet = (id, text, username = 'yuta') => ({
  entryId: `tweet-${id}`,
  content: {
    itemContent: {
      tweet_results: {
        result: {
          rest_id: id,
          legacy: {
            full_text: text,
            created_at: 'Mon Jul 13 00:00:00 +0000 2026',
            reply_count: 1,
            retweet_count: 2,
            favorite_count: 3,
            conversation_id_str: id,
          },
          core: {
            user_results: {
              result: {
                rest_id: '42',
                legacy: { screen_name: username, name: 'Yuta' },
              },
            },
          },
        },
      },
    },
  },
})

const cursor = (value) => ({
  entryId: 'cursor-bottom',
  content: { cursorType: 'Bottom', value },
})

const send = (response, payload, status = 200, headers = {}) => {
  response.writeHead(status, {
    'content-type': 'application/json; charset=utf-8',
    ...headers,
  })
  response.end(JSON.stringify(payload))
}

const fail = (response, status, message, headers) =>
  send(response, { errors: [{ message }] }, status, headers)

const variablesFrom = (url) => {
  const encoded = url.searchParams.get('variables')
  if (!encoded) throw new Error('variables are required')
  const variables = JSON.parse(encoded)
  if (!isRecord(variables)) throw new Error('variables must be an object')
  return variables
}

const jsonBodyFrom = async (request) => {
  let body = ''
  for await (const chunk of request) {
    body += chunk
    if (body.length > 1_000_000) throw new Error('request body is too large')
  }
  const parsed = JSON.parse(body)
  if (!isRecord(parsed)) throw new Error('JSON body must be an object')
  return parsed
}

const timeline = (entries) => ({ instructions: [{ entries }] })

const server = createServer(async (request, response) => {
  const url = new URL(request.url ?? '/', `http://127.0.0.1:${port}`)
  const operation = ['UserByScreenName', 'UserTweets', 'SearchTimeline'].find(
    (name) => url.pathname.endsWith(`/${name}`),
  )

  if (!operation) {
    fail(response, 501, 'unsupported read operation')
    return
  }

  const expectedMethod = operation === 'SearchTimeline' ? 'POST' : 'GET'
  if (request.method !== expectedMethod) {
    fail(response, 405, `expected ${expectedMethod}`, {
      allow: expectedMethod,
    })
    return
  }
  if (request.headers['x-profile-name'] !== 'e2e') {
    fail(response, 403, 'invalid profile')
    return
  }

  let variables
  try {
    variables = variablesFrom(url)
  } catch {
    fail(response, 400, 'malformed variables')
    return
  }

  if (operation === 'UserByScreenName') {
    if (
      typeof variables.screen_name !== 'string' ||
      variables.screen_name.length === 0
    ) {
      fail(response, 400, 'screen_name is required')
      return
    }
    send(response, {
      data: {
        user: {
          result: {
            rest_id: '42',
            legacy: { screen_name: variables.screen_name, name: 'Yuta' },
          },
        },
      },
    })
    return
  }

  if (operation === 'UserTweets') {
    const expectedCursor =
      variables.cursor === undefined || variables.cursor === 'user-next'
    if (
      variables.userId !== '42' ||
      variables.count !== 20 ||
      !expectedCursor
    ) {
      fail(response, 400, 'invalid user timeline variables')
      return
    }
    const secondPage = variables.cursor === 'user-next'
    const entries = secondPage
      ? [tweet('u2', 'user page 2')]
      : [tweet('u1', 'user page 1'), cursor('user-next')]
    send(response, {
      data: {
        user: { result: { timeline: { timeline: timeline(entries) } } },
      },
    })
    return
  }

  const contentType = request.headers['content-type'] ?? ''
  if (!/^application\/json(?:\s*;|$)/i.test(contentType)) {
    fail(response, 400, 'SearchTimeline requires JSON')
    return
  }

  let body
  try {
    body = await jsonBodyFrom(request)
  } catch {
    fail(response, 400, 'malformed JSON body')
    return
  }

  const rawQuery = variables.rawQuery
  const expectedCursor =
    variables.cursor === undefined ||
    variables.cursor === `search-next:${rawQuery}`
  if (
    typeof rawQuery !== 'string' ||
    rawQuery.length === 0 ||
    variables.count !== 20 ||
    variables.querySource !== 'typed_query' ||
    !['Top', 'Latest'].includes(variables.product) ||
    !expectedCursor ||
    !isRecord(body.features) ||
    typeof body.queryId !== 'string' ||
    body.queryId.length === 0 ||
    Object.hasOwn(body, 'variables')
  ) {
    fail(response, 400, 'invalid search request')
    return
  }

  const secondPage = variables.cursor !== undefined
  if (
    secondPage &&
    rawQuery.startsWith('retry-') &&
    !failedRetryQueries.has(rawQuery)
  ) {
    failedRetryQueries.add(rawQuery)
    fail(response, 503, 'transient fixture failure')
    return
  }
  if (secondPage && rawQuery.startsWith('slow-')) {
    await new Promise((resolve) => setTimeout(resolve, 1_200))
  }

  const follows = rawQuery.includes('filter:follows') ? 'follows' : 'all'
  const label = `${variables.product} · ${follows}`
  const id = encodeURIComponent(rawQuery)
  const entries = secondPage
    ? [tweet(`s-${id}-2`, `${label} page 2`)]
    : [tweet(`s-${id}-1`, `${label} page 1`), cursor(`search-next:${rawQuery}`)]

  send(response, {
    data: {
      search_by_raw_query: {
        search_timeline: { timeline: timeline(entries) },
      },
    },
  })
})

server.listen(port, '127.0.0.1')