bambuddy/frontend/src/__tests__/components/AddPrinterDiscovery.test.tsx

/**
 * Tests for AddPrinterModal discovery subnet auto-detection.
 */

import { describe, it, expect, beforeEach } from 'vitest';
import { screen, waitFor } from '@testing-library/react';
import userEvent from '@testing-library/user-event';
import { render } from '../utils';
import { PrintersPage } from '../../pages/PrintersPage';
import { http, HttpResponse } from 'msw';
import { server } from '../mocks/server';

const mockPrinters = [
  {
    id: 1,
    name: 'X1 Carbon',
    ip_address: '192.168.1.100',
    serial_number: '00M09A350100001',
    access_code: '12345678',
    model: 'X1C',
    enabled: true,
    nozzle_diameter: 0.4,
    nozzle_type: 'hardened_steel',
    location: null,
    auto_archive: true,
    created_at: '2024-01-01T00:00:00Z',
    updated_at: '2024-01-01T00:00:00Z',
  },
];

const mockPrinterStatus = {
  connected: true,
  state: 'IDLE',
  progress: 0,
  layer_num: 0,
  total_layers: 0,
  temperatures: { nozzle: 25, bed: 25, chamber: 25 },
  remaining_time: 0,
  filename: null,
  wifi_signal: -50,
  vt_tray: [],
};

describe('AddPrinterModal Discovery', () => {
  beforeEach(() => {
    server.use(
      http.get('/api/v1/printers/', () => {
        return HttpResponse.json(mockPrinters);
      }),
      http.get('/api/v1/printers/:id/status', () => {
        return HttpResponse.json(mockPrinterStatus);
      }),
      http.get('/api/v1/queue/', () => {
        return HttpResponse.json([]);
      })
    );
  });

  it('auto-populates subnet from discovery info in Docker mode', async () => {
    server.use(
      http.get('/api/v1/discovery/info', () => {
        return HttpResponse.json({
          is_docker: true,
          ssdp_running: false,
          scan_running: false,
          subnets: ['10.0.0.0/24'],
        });
      })
    );

    render(<PrintersPage />);

    // Wait for printer page to load
    await waitFor(() => {
      expect(screen.getByText('X1 Carbon')).toBeInTheDocument();
    });

    // Click the Add Printer button
    const addButton = screen.getByText(/add printer/i);
    await userEvent.click(addButton);

    // Wait for the modal and discovery info to load
    await waitFor(() => {
      // Should show subnet dropdown with detected subnet
      const subnetSelect = screen.getByDisplayValue('10.0.0.0/24');
      expect(subnetSelect).toBeInTheDocument();
    });
  });

  it('shows dropdown when multiple subnets detected in Docker mode', async () => {
    server.use(
      http.get('/api/v1/discovery/info', () => {
        return HttpResponse.json({
          is_docker: true,
          ssdp_running: false,
          scan_running: false,
          subnets: ['192.168.1.0/24', '10.0.0.0/24'],
        });
      })
    );

    render(<PrintersPage />);

    await waitFor(() => {
      expect(screen.getByText('X1 Carbon')).toBeInTheDocument();
    });

    const addButton = screen.getByText(/add printer/i);
    await userEvent.click(addButton);

    await waitFor(() => {
      // Should show a select element (dropdown) with both subnets and
      // the trailing "Custom subnet..." sentinel that lets a user enter
      // a CIDR for a printer on a different L3 segment (#1564).
      const selectElement = screen.getByDisplayValue('192.168.1.0/24');
      expect(selectElement.tagName).toBe('SELECT');

      const options = selectElement.querySelectorAll('option');
      expect(options).toHaveLength(3);
      expect(options[0].textContent).toBe('192.168.1.0/24');
      expect(options[1].textContent).toBe('10.0.0.0/24');
      expect(options[2].textContent).toMatch(/custom subnet/i);
    });
  });

  it('shows text input when no subnets detected in Docker mode', async () => {
    server.use(
      http.get('/api/v1/discovery/info', () => {
        return HttpResponse.json({
          is_docker: true,
          ssdp_running: false,
          scan_running: false,
          subnets: [],
        });
      })
    );

    render(<PrintersPage />);

    await waitFor(() => {
      expect(screen.getByText('X1 Carbon')).toBeInTheDocument();
    });

    const addButton = screen.getByText(/add printer/i);
    await userEvent.click(addButton);

    await waitFor(() => {
      // Should show a text input with placeholder
      const textInput = screen.getByPlaceholderText('192.168.1.0/24');
      expect(textInput).toBeInTheDocument();
      expect(textInput.tagName).toBe('INPUT');
    });
  });

  it('shows the subnet picker in non-Docker mode too, with a Custom option (#1564)', async () => {
    // Pre-#1564 the subnet picker was gated on isDocker and a native
    // install only saw the "Discover Printers" button. SSDP doesn't
    // cross routers, so users with a printer on a different L3 segment
    // had no path to scan it. The picker is now always visible.
    server.use(
      http.get('/api/v1/discovery/info', () => {
        return HttpResponse.json({
          is_docker: false,
          ssdp_running: false,
          scan_running: false,
          subnets: ['192.168.1.0/24'],
        });
      })
    );

    render(<PrintersPage />);

    await waitFor(() => {
      expect(screen.getByText('X1 Carbon')).toBeInTheDocument();
    });

    const addButton = screen.getByText(/add printer/i);
    await userEvent.click(addButton);

    // Detected subnet is the default value; "Custom subnet..." sits
    // alongside it so the user can pick a foreign CIDR.
    await waitFor(() => {
      const selectElement = screen.getByDisplayValue('192.168.1.0/24');
      expect(selectElement.tagName).toBe('SELECT');
      const options = selectElement.querySelectorAll('option');
      expect(options).toHaveLength(2);
      expect(options[0].textContent).toBe('192.168.1.0/24');
      expect(options[1].textContent).toMatch(/custom subnet/i);
    });

    // Default selection leaves the SSDP path in place — the button
    // still reads "Discover Printers on Network", not "Scan Subnet".
    expect(screen.getByText(/discover printers/i)).toBeInTheDocument();
  });
});