feat: improve quoting mechanism

Author: smeijer

src/lib/getExpression.js

@@ -1,5 +1,17 @@
-export function escape(val) {
-  return val.replace(/'/g, `\\'`);
+export function wrapInQuotes(val) {
+  if (!val.includes(`'`)) {
+    return `'${val}'`;
+  }
+
+  if (!val.includes('"')) {
+    return `"${val}"`;
+  }
+
+  if (!val.includes('`')) {
+    return `\`${val}\``;
+  }
+
+  return `'${val.replace(/'/g, `\\'`)}'`;
 }
 
 export function getFieldName(method) {
@@ -10,11 +22,13 @@ export function getExpression({ method, data }) {
   const field = getFieldName(method);
 
   if (method === 'getByRole' && data.role && data.name) {
-    return `screen.getByRole('${data.role}', { name: '${escape(data.name)}' })`;
+    return `screen.getByRole('${data.role}', { name: ${wrapInQuotes(
+      data.name,
+    )} })`;
   }
 
   if (data[field]) {
-    return `screen.${method}('${escape(data[field])}')`;
+    return `screen.${method}(${wrapInQuotes(data[field])})`;
   }
 
   return '';