Add data-testid attributes to UI elements for unit test and automatio…

…n tests (apache#43736)
ellisms · Nov 13, 2024 · eebd979 · eebd979
1 parent eaae3b2
commit eebd979
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/airflow/ui/src/components/DataTable/CardList.tsx b/airflow/ui/src/components/DataTable/CardList.tsx
@@ -32,7 +32,7 @@ export const CardList = <TData,>({
   isLoading,
   table,
 }: DataTableProps<TData>) => (
-  <Box overflow="auto" width="100%">
+  <Box data-testid="card-list" overflow="auto" width="100%">
     <SimpleGrid {...{ column: { base: 1 }, gap: 2, ...cardDef.gridProps }}>
       {table.getRowModel().rows.map((row) => (
         <Box key={row.id}>

diff --git a/airflow/ui/src/components/DataTable/TableList.tsx b/airflow/ui/src/components/DataTable/TableList.tsx
@@ -40,7 +40,12 @@ export const TableList = <TData,>({
   renderSubComponent,
   table,
 }: DataTableProps<TData>) => (
-  <Table.Root maxH="calc(100vh - 10rem)" overflowY="auto" striped>
+  <Table.Root
+    data-testid="table-list"
+    maxH="calc(100vh - 10rem)"
+    overflowY="auto"
+    striped
+  >
     <Table.Header bg="chakra-body-bg" position="sticky" top={0} zIndex={1}>
       {table.getHeaderGroups().map((headerGroup) => (
         <Table.Row key={headerGroup.id}>

diff --git a/airflow/ui/src/components/SearchBar.tsx b/airflow/ui/src/components/SearchBar.tsx
@@ -62,6 +62,7 @@ export const SearchBar = ({
       startElement={<FiSearch />}
     >
       <Input
+        data-testid="search-dags"
         placeholder="Search Dags"
         pr={150}
         {...inputProps}