diff --git a/airflow-core/src/airflow/ui/src/components/FilterBar/FilterBar.tsx b/airflow-core/src/airflow/ui/src/components/FilterBar/FilterBar.tsx
index 2c830e72f51de..0df8f784ccef0 100644
--- a/airflow-core/src/airflow/ui/src/components/FilterBar/FilterBar.tsx
+++ b/airflow-core/src/airflow/ui/src/components/FilterBar/FilterBar.tsx
@@ -152,6 +152,7 @@ export const FilterBar = ({
_hover={{ bg: "colorPalette.subtle" }}
bg="gray.muted"
borderRadius="full"
+ data-testid="filter-bar-add-button"
variant="outline"
>
diff --git a/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsButton.tsx b/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsButton.tsx
index aaff49bd35a60..b13385691aa4f 100644
--- a/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsButton.tsx
+++ b/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsButton.tsx
@@ -65,6 +65,7 @@ const MarkRunAsButton = ({ dagRun, isHotkeyEnabled = false, withText = true }: P
}
text={translate("dags:runAndTaskActions.markAs.button", { type: translate("dagRun_one") })}
@@ -87,6 +88,7 @@ const MarkRunAsButton = ({ dagRun, isHotkeyEnabled = false, withText = true }: P
>
{
diff --git a/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsDialog.tsx b/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsDialog.tsx
index fa05d68587b2f..da10d6a500fcd 100644
--- a/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsDialog.tsx
+++ b/airflow-core/src/airflow/ui/src/components/MarkAs/Run/MarkRunAsDialog.tsx
@@ -63,6 +63,7 @@ const MarkRunAsDialog = ({ dagRun, onClose, open, state }: Props) => {