From 77080849cba9f15f6569d062051aa6869852bf48 Mon Sep 17 00:00:00 2001
From: David Cho-Lerat <david.cho-lerat@sonarsource.com>
Date: Tue, 31 Oct 2023 16:33:06 +0100
Subject: [PATCH] [NO JIRA] Fix some code smells on new code

---
 .../src/main/js/helpers/testSelector.ts       | 47 +++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/server/sonar-web/src/main/js/helpers/testSelector.ts b/server/sonar-web/src/main/js/helpers/testSelector.ts
index b4f8adcdf90..1e4b9af3dde 100644
--- a/server/sonar-web/src/main/js/helpers/testSelector.ts
+++ b/server/sonar-web/src/main/js/helpers/testSelector.ts
@@ -130,6 +130,10 @@ abstract class ChainingQuery implements ReactTestingQuery {
   byDisplayValue(...args: Parameters<BoundFunction<GetByBoundAttribute>>): ReactTestingQuery {
     return this.by(new DispatchByDisplayValue(args));
   }
+
+  byTitle(...args: Parameters<BoundFunction<GetByBoundAttribute>>): ReactTestingQuery {
+    return this.by(new DispatchByTitle(args));
+  }
 }
 
 class ChainDispatch extends ChainingQuery {
@@ -341,6 +345,45 @@ class DispatchByTestId extends ChainingQuery {
   }
 }
 
+class DispatchByTitle extends ChainingQuery {
+  readonly args: Parameters<BoundFunction<GetByBoundAttribute>>;
+
+  constructor(args: Parameters<BoundFunction<GetByBoundAttribute>>) {
+    super();
+    this.args = args;
+  }
+
+  find<T extends HTMLElement = HTMLElement>(
+    container?: HTMLElement,
+    waitForOptions?: waitForOptions,
+  ) {
+    return maybeScreen(container).findByTitle<T>(...this.args, waitForOptions);
+  }
+
+  findAll<T extends HTMLElement = HTMLElement>(
+    container?: HTMLElement,
+    waitForOptions?: waitForOptions,
+  ) {
+    return maybeScreen(container).findAllByTitle<T>(...this.args, waitForOptions);
+  }
+
+  get<T extends HTMLElement = HTMLElement>(container?: HTMLElement) {
+    return maybeScreen(container).getByTitle<T>(...this.args);
+  }
+
+  getAll<T extends HTMLElement = HTMLElement>(container?: HTMLElement) {
+    return maybeScreen(container).getAllByTitle<T>(...this.args);
+  }
+
+  query<T extends HTMLElement = HTMLElement>(container?: HTMLElement) {
+    return maybeScreen(container).queryByTitle<T>(...this.args);
+  }
+
+  queryAll<T extends HTMLElement = HTMLElement>(container?: HTMLElement) {
+    return maybeScreen(container).queryAllByTitle<T>(...this.args);
+  }
+}
+
 class DispatchByDisplayValue extends ChainingQuery {
   readonly args: Parameters<BoundFunction<GetByBoundAttribute>>;
 
@@ -439,6 +482,10 @@ export function byTestId(...args: Parameters<BoundFunction<GetByBoundAttribute>>
   return new DispatchByTestId(args);
 }
 
+export function byTitle(...args: Parameters<BoundFunction<GetByBoundAttribute>>) {
+  return new DispatchByTitle(args);
+}
+
 export function byDisplayValue(...args: Parameters<BoundFunction<GetByBoundAttribute>>) {
   return new DispatchByDisplayValue(args);
 }
-- 
2.39.5