upstream-affine/tests/blocksuite/e2e/image/load.spec.ts

import { readFile } from 'node:fs/promises';
import { fileURLToPath } from 'node:url';

import type { Page } from '@playwright/test';
import { expect } from '@playwright/test';

import {
  enterPlaygroundRoom,
  expectConsoleMessage,
} from '../utils/actions/index.js';
import { test } from '../utils/playwright.js';

const mockImageId = '_e2e_test_image_id_';

async function initMockImage(page: Page) {
  await page.evaluate(sourceId => {
    const { doc } = window;
    doc.captureSync();
    const rootId = doc.addBlock('affine:page');
    const noteId = doc.addBlock('affine:note', {}, rootId);
    doc.addBlock(
      'affine:image',
      {
        sourceId,
        width: 200,
        height: 180,
      },
      noteId
    );
    doc.captureSync();
  }, mockImageId);
}

test('image loading but failed', async ({ page }) => {
  expectConsoleMessage(
    page,
    `Error: Failed to fetch blob ${mockImageId}`,
    'warning'
  );
  expectConsoleMessage(
    page,
    'Failed to load resource: the server responded with a status of 404 (Not Found)'
  );
  expectConsoleMessage(
    page,
    'Error: Image blob is missing!, retrying',
    'warning'
  );

  const room = await enterPlaygroundRoom(page, { blobSource: ['mock'] });
  const timeout = 2000;

  // block image data request, force wait 100ms for loading test,
  // always return 404
  await page.route(
    `**/api/collection/${room}/blob/${mockImageId}`,
    async route => {
      await page.waitForTimeout(timeout);
      // broken image
      return route.fulfill({
        status: 404,
      });
    }
  );

  await initMockImage(page);

  const title = page.locator(
    '.affine-image-fallback-card .affine-image-fallback-card-title-text'
  );

  await expect(title).toHaveText('Image');

  await page.waitForTimeout(3 * timeout);

  const desc = page.locator(
    '.affine-image-fallback-card .affine-image-fallback-card-description'
  );

  await expect(desc).toContainText('Image not found');
});

test('image loading but success', async ({ page }) => {
  expectConsoleMessage(
    page,
    `Error: Failed to fetch blob ${mockImageId}`,
    'warning'
  );
  expectConsoleMessage(
    page,
    'Failed to load resource: the server responded with a status of 404 (Not Found)'
  );
  expectConsoleMessage(
    page,
    'Error: Image blob is missing!, retrying',
    'warning'
  );

  const room = await enterPlaygroundRoom(page, { blobSource: ['mock'] });
  const imageBuffer = await readFile(
    fileURLToPath(new URL('../fixtures/smile.png', import.meta.url))
  );

  const timeout = 2000;
  let count = 0;

  // block image data request, force wait 100ms for loading test,
  // always return 404
  await page.route(
    `**/api/collection/${room}/blob/${mockImageId}`,
    async route => {
      await page.waitForTimeout(timeout);
      count++;
      if (count === 3) {
        return route.fulfill({
          status: 200,
          body: imageBuffer,
        });
      }

      return route.continue();
    }
  );

  await initMockImage(page);

  const title = page.locator(
    '.affine-image-fallback-card .affine-image-fallback-card-title-text'
  );

  await expect(title).toHaveText('Image');

  await page.waitForTimeout(3 * timeout);

  const img = page.locator('.affine-image-container img');
  await expect(img).toBeVisible();
  const src = await img.getAttribute('src');
  expect(src).toBeDefined();
});

test('image loaded successfully', async ({ page }) => {
  const room = await enterPlaygroundRoom(page, { blobSource: ['mock'] });
  const imageBuffer = await readFile(
    fileURLToPath(new URL('../fixtures/smile.png', import.meta.url))
  );
  await page.route(
    `**/api/collection/${room}/blob/${mockImageId}`,
    async route => {
      return route.fulfill({
        status: 200,
        body: imageBuffer,
      });
    }
  );

  await initMockImage(page);

  await page.waitForTimeout(1000);

  const img = page.locator('.affine-image-container img');
  await expect(img).toBeVisible();
  const src = await img.getAttribute('src');
  expect(src).toBeDefined();
});