prowler/ui/components/providers/table/column-providers.test.tsx

import { render, screen } from "@testing-library/react";
import userEvent from "@testing-library/user-event";
import type { ReactNode } from "react";
import { afterEach, describe, expect, it, vi } from "vitest";

import {
  PROVIDERS_ROW_TYPE,
  type ProvidersProviderRow,
  type ProvidersTableRow,
} from "@/types/providers-table";

import { getColumnProviders } from "./column-providers";

vi.mock("@/components/shadcn", async () => {
  const tooltip = await vi.importActual<
    typeof import("@/components/shadcn/tooltip")
  >("@/components/shadcn/tooltip");

  return {
    Badge: ({ children, ...props }: { children: ReactNode }) => (
      <span {...props}>{children}</span>
    ),
    ...tooltip,
  };
});

vi.mock("@/components/shadcn/checkbox/checkbox", () => ({
  Checkbox: () => null,
}));

vi.mock("@/components/shadcn/code-snippet/code-snippet", () => ({
  CodeSnippet: ({ value }: { value: string }) => <code>{value}</code>,
}));

vi.mock("@/components/shadcn/entities", () => ({
  DateWithTime: ({ dateTime }: { dateTime: string | null }) => (
    <time>{dateTime}</time>
  ),
  EntityInfo: ({
    entityAlias,
    entityId,
  }: {
    entityAlias?: string;
    entityId?: string;
  }) => <span>{entityAlias ?? entityId}</span>,
}));

vi.mock("@/components/shadcn/table", () => ({
  DataTableColumnHeader: ({ title }: { title: string }) => <span>{title}</span>,
}));

vi.mock("@/components/shadcn/table/data-table-expand-all-toggle", () => ({
  DataTableExpandAllToggle: () => null,
}));

vi.mock("@/components/shadcn/table/data-table-expandable-cell", () => ({
  DataTableExpandableCell: ({ children }: { children: ReactNode }) => (
    <div>{children}</div>
  ),
}));

vi.mock("../link-to-scans", () => ({
  LinkToScans: () => null,
}));

vi.mock("./data-table-row-actions", () => ({
  DataTableRowActions: () => null,
}));

const providerRow: ProvidersProviderRow = {
  id: "provider-1",
  rowType: PROVIDERS_ROW_TYPE.PROVIDER,
  type: "providers",
  attributes: {
    provider: "aws",
    is_dynamic: false,
    is_imported: false,
    uid: "123456789012",
    alias: "Production",
    status: "completed",
    resources: 0,
    connection: {
      connected: true,
      last_checked_at: "2026-01-01T00:00:00Z",
    },
    scanner_args: {
      only_logs: false,
      excluded_checks: [],
      aws_retries_max_attempts: 3,
    },
    inserted_at: "2026-01-01T00:00:00Z",
    updated_at: "2026-01-01T00:00:00Z",
    created_by: {
      object: "user",
      id: "user-1",
    },
  },
  relationships: {
    secret: { data: { id: "secret-1", type: "secrets" } },
    provider_groups: { meta: { count: 0 }, data: [] },
  },
  groupNames: [],
  hasSchedule: false,
};

function renderLastScanCell(row: ProvidersTableRow) {
  const lastScanColumn = getColumnProviders(
    {},
    [],
    [],
    [],
    vi.fn(),
    vi.fn(),
    vi.fn(),
  ).find((column) => column.id === "lastScan");

  const cell = lastScanColumn?.cell;
  if (typeof cell !== "function") {
    throw new Error("Last Scan column cell renderer not found");
  }

  const element = cell({
    row: { original: row },
  } as unknown as Parameters<typeof cell>[0]);

  render(<>{element as ReactNode}</>);
}

function renderStatusCell(row: ProvidersTableRow) {
  const statusColumn = getColumnProviders(
    {},
    [],
    [],
    [],
    vi.fn(),
    vi.fn(),
    vi.fn(),
  ).find((column) => column.id === "status");

  const cell = statusColumn?.cell;
  if (typeof cell !== "function") {
    throw new Error("Status column cell renderer not found");
  }

  const element = cell({
    row: { original: row },
  } as unknown as Parameters<typeof cell>[0]);

  render(<>{element as ReactNode}</>);
}

afterEach(() => {
  vi.unstubAllEnvs();
});

describe("getColumnProviders", () => {
  it("falls back to connection last_checked_at when lastScanAt is undefined", () => {
    renderLastScanCell({ ...providerRow, lastScanAt: undefined });

    expect(screen.getByText("2026-01-01T00:00:00Z")).toBeVisible();
    expect(screen.queryByText("Never")).not.toBeInTheDocument();
  });

  it("treats a null lastScanAt as authoritative", () => {
    renderLastScanCell({ ...providerRow, lastScanAt: null });

    expect(screen.getByText("Never")).toBeVisible();
    expect(screen.queryByText("2026-01-01T00:00:00Z")).not.toBeInTheDocument();
  });

  it("shows Not connected with the imported tooltip in Cloud", async () => {
    // Given
    vi.stubEnv("UI_CLOUD_ENABLED", "true");
    const user = userEvent.setup();
    const importedProvider = {
      ...providerRow,
      attributes: {
        ...providerRow.attributes,
        is_imported: true,
        connection: {
          connected: null,
          last_checked_at: null,
        },
      },
      relationships: {
        ...providerRow.relationships,
        secret: { data: null },
      },
    } satisfies ProvidersProviderRow;

    // When
    renderStatusCell(importedProvider);

    // Then
    expect(screen.getByText("Not connected")).toBeVisible();
    const indicator = screen.getByLabelText("Imported provider");
    expect(indicator).not.toHaveTextContent("Imported");
    expect(indicator).toHaveAttribute("tabindex", "0");
    expect(indicator.querySelector(".lucide-terminal")).toBeInTheDocument();
    expect(indicator.parentElement).toHaveClass("items-stretch");
    expect(screen.queryByRole("tooltip")).not.toBeInTheDocument();
    expect(screen.queryByText("Disconnected")).not.toBeInTheDocument();

    await user.hover(indicator);

    expect(await screen.findByRole("tooltip")).toHaveTextContent(
      /^This provider has findings imported with the Prowler CLI$/,
    );
  });

  it("shows the imported tooltip beside Connected in Cloud", async () => {
    // Given
    vi.stubEnv("UI_CLOUD_ENABLED", "true");
    const user = userEvent.setup();
    const connectedImportedProvider = {
      ...providerRow,
      attributes: {
        ...providerRow.attributes,
        is_imported: true,
        connection: {
          connected: true,
          last_checked_at: "2026-01-01T00:00:00Z",
        },
      },
    } satisfies ProvidersProviderRow;

    // When
    renderStatusCell(connectedImportedProvider);

    // Then
    expect(screen.getByText("Connected")).toBeVisible();
    const indicator = screen.getByLabelText("Imported provider");
    expect(indicator).toBeVisible();
    expect(indicator.querySelector(".lucide-terminal")).toBeInTheDocument();
    expect(indicator.parentElement).toHaveClass("items-stretch");
    expect(indicator).not.toHaveTextContent("Imported");
    expect(screen.queryByRole("tooltip")).not.toBeInTheDocument();

    await user.hover(indicator);

    expect(await screen.findByRole("tooltip")).toHaveTextContent(
      /^This provider has findings imported with the Prowler CLI$/,
    );
  });

  it("keeps Connection failed for imported providers in Cloud", () => {
    // Given
    vi.stubEnv("UI_CLOUD_ENABLED", "true");
    const importedProviderWithCredentials = {
      ...providerRow,
      attributes: {
        ...providerRow.attributes,
        is_imported: true,
        connection: {
          connected: false,
          last_checked_at: null,
        },
      },
    } satisfies ProvidersProviderRow;

    // When
    renderStatusCell(importedProviderWithCredentials);

    // Then
    expect(screen.getByText("Connection failed")).toBeVisible();
    expect(screen.queryByText("Not connected")).not.toBeInTheDocument();
    const indicator = screen.getByLabelText("Imported provider");
    expect(indicator).not.toHaveTextContent("Imported");
  });

  it("does not expose imported provenance in OSS", () => {
    // Given
    vi.stubEnv("UI_CLOUD_ENABLED", "false");
    const importedProvider = {
      ...providerRow,
      attributes: {
        ...providerRow.attributes,
        is_imported: true,
        connection: {
          connected: null,
          last_checked_at: null,
        },
      },
      relationships: {
        ...providerRow.relationships,
        secret: { data: null },
      },
    } satisfies ProvidersProviderRow;

    // When
    renderStatusCell(importedProvider);

    // Then
    expect(screen.getByText("Not connected")).toBeVisible();
    expect(
      screen.queryByLabelText("Imported provider"),
    ).not.toBeInTheDocument();
  });
});