premieroctet · quentingrchr · Jun 21, 2024 · Jun 21, 2024
diff --git a/src/utils/link/index.test.js → __tests__/src/utils/link.test.js b/src/utils/link/index.test.js → __tests__/src/utils/link.test.js
@@ -1,5 +1,5 @@
+import { getTweetId, isTwitterLink } from '@/utils/link';
 import { describe, expect, test } from '@jest/globals';
-import { isTwitterLink, getTweetId } from './index';
 
 describe('isTwitterLink', () => {
   test('url to user should return true', () => {

diff --git a/__tests__/src/utils/newsletter.test.js b/__tests__/src/utils/newsletter.test.js
@@ -0,0 +1,58 @@
+import { describe, expect, test } from '@jest/globals';
+import { applyInlineStyleToRawHtml } from '@/utils/newsletter';
+
+describe('applyInlineStyleToRawHtml', () => {
+  test('should apply style to h1 tag', () => {
+    const theme = {
+      h1: {
+        color: 'red',
+      },
+    };
+    const rawHtml = '<h1>Test</h1>';
+    const expected = '<h1 style="color: red;">Test</h1>';
+    const result = applyInlineStyleToRawHtml(rawHtml, theme);
+    expect(result).toBe(expected);
+  });
+
+  test('should apply a complex style to only h1 tag', () => {
+    const theme = {
+      h1: {
+        color: 'red',
+        fontSize: '2rem',
+        fontFamily: 'Arial',
+        fontWeight: 'bold',
+        border: '1px solid black',
+        borderRadius: '5px',
+        padding: '10px',
+        margin: '10px',
+        paddingTop: '20px',
+      },
+    };
+    const rawHtml =
+      '<body><div><h1><span>Test</span></h1></div><h2>Test</h2><h3>Test</h3><p>Lorem ipsum dolor sit amet consectetur adipisicing elit. Quisquam, voluptatum.</p><img src="https://picsum.photos/200/300" alt="random image" /><a href="https://www.google.com">Google</a>';
+    const expected =
+      '<body><div><h1 style="color: red; font-size: 2rem; font-family: Arial; font-weight: bold; border: 1px solid black; border-radius: 5px; padding: 10px; margin: 10px; padding-top: 20px;"><span>Test</span></h1></div><h2>Test</h2><h3>Test</h3><p>Lorem ipsum dolor sit amet consectetur adipisicing elit. Quisquam, voluptatum.</p><img src="https://picsum.photos/200/300" alt="random image" /><a href="https://www.google.com">Google</a>';
+    const result = applyInlineStyleToRawHtml(rawHtml, theme);
+    expect(result).toBe(expected);
+  });
+
+  test('should not apply style on <img />, <a> tags as they might have attributes that will be erased by the function', () => {
+    const theme = {
+      h1: {
+        color: 'red',
+      },
+      img: {
+        color: 'red',
+      },
+      a: {
+        color: 'red',
+      },
+    };
+    const rawHtml =
+      '<img src="https://picsum.photos/200/300" alt="random image" /><a href="https://www.google.com">Google</a>';
+    const expected =
+      '<img src="https://picsum.photos/200/300" alt="random image" /><a href="https://www.google.com">Google</a>';
+    const result = applyInlineStyleToRawHtml(rawHtml, theme);
+    expect(result).toBe(expected);
+  });
+});
diff --git a/__tests__/src/utils/string.test.js b/__tests__/src/utils/string.test.js
@@ -0,0 +1,42 @@
+import { isStringEmpty, makeHidden } from '@/utils/string';
+import { describe, expect } from '@jest/globals';
+
+describe('isStringEmpty', () => {
+  it('returns true for empty strings', () => {
+    expect(isStringEmpty('')).toBe(true);
+  });
+
+  it('returns true for strings with only spaces', () => {
+    expect(isStringEmpty('   ')).toBe(true);
+  });
+
+  it('returns false for non-empty strings', () => {
+    expect(isStringEmpty('Hello')).toBe(false);
+  });
+
+  it('returns false for strings with spaces', () => {
+    expect(isStringEmpty('   Hello   ')).toBe(false);
+  });
+});
+
+describe('makeHidden', () => {
+  it('replaces each character with *', () => {
+    expect(makeHidden('hello')).toBe('*****');
+  });
+
+  it('works with empty string', () => {
+    expect(makeHidden('')).toBe('');
+  });
+
+  it('replaces each character including spaces', () => {
+    expect(makeHidden('Hello, world!')).toBe('*************');
+  });
+
+  it('handles Unicode characters', () => {
+    expect(makeHidden('😊🌟')).toBe('**');
+  });
+
+  it('handles long strings', () => {
+    expect(makeHidden('a'.repeat(100))).toBe('*'.repeat(100));
+  });
+});
diff --git a/__tests__/src/utils/url.test.js b/__tests__/src/utils/url.test.js
@@ -0,0 +1,33 @@
+import { getDomainFromUrl, isPdfUrl } from '@/utils/url';
+import { describe, expect } from '@jest/globals';
+
+describe('getDomainFromUrl', () => {
+  it('should extract domain from a valid URL', () => {
+    const url = 'https://www.example.com/path/to/resource';
+    expect(getDomainFromUrl(url)).toBe('example.com');
+  });
+
+  it('should handle URL with protocol', () => {
+    const url = 'http://subdomain.example.org';
+    expect(getDomainFromUrl(url)).toBe('example.org');
+  });
+});
+
+describe('isPdfUrl', () => {
+  it('should return true for valid PDF URLs', () => {
+    // Test cases where the URL ends with '.pdf'
+    expect(isPdfUrl('https://example.com/file.pdf')).toBe(true);
+    expect(isPdfUrl('http://www.test.com/doc.pdf')).toBe(true);
+    expect(isPdfUrl('ftp://ftp.site.com/report.pdf')).toBe(true);
+  });
+
+  it('should return false for invalid PDF URLs', () => {
+    // Test cases where the URL does not end with '.pdf'
+    expect(isPdfUrl('https://example.com/document.pdfx')).toBe(false);
+    expect(isPdfUrl('http://www.test.com/file.PDF')).toBe(false); // Case-sensitive check
+    expect(isPdfUrl('ftp://ftp.site.com/reportpdf')).toBe(false);
+    expect(isPdfUrl('https://example.com/pdf')).toBe(false);
+    expect(isPdfUrl('https://example.com/.pdf')).toBe(false);
+    expect(isPdfUrl('https://example.com/pdf.')).toBe(false);
+  });
+});
diff --git a/src/utils/link/index.ts → src/utils/link.ts b/src/utils/link/index.ts → src/utils/link.ts
diff --git a/src/utils/newsletter/index.ts → src/utils/newsletter.ts b/src/utils/newsletter/index.ts → src/utils/newsletter.ts
@@ -1,4 +1,4 @@
-import baseTheme from '../../emails/theme';
+import baseTheme from '@/emails/theme';
 
 type Theme = {
   [key: string]: {

diff --git a/src/utils/newsletter/index.test.js b/src/utils/newsletter/index.test.js
diff --git a/src/utils/string.ts b/src/utils/string.ts
@@ -12,7 +12,7 @@ export function isStringEmpty(str: string): boolean {
  * Function that returns a hidden string, i.e. a string with all characters replaced by '*'
  */
 export function makeHidden(str: string): string {
-  return str.replace(/./g, '*');
+    return str.replace(/[\s\S]/gu, '*');
 }
 
 /**

diff --git a/src/utils/url.ts b/src/utils/url.ts
@@ -5,4 +5,13 @@ export const getDomainFromUrl = (url: string) => {
   return domain;
 };
 
-export const isPdfUrl = (url: string) => url.indexOf('.pdf') > -1;
+export const isPdfUrl = (url: string): boolean => {
+  const urlObject = new URL(url);
+  const pathname = urlObject.pathname;
+  const hasExtension = pathname.endsWith('.pdf');
+  if (!hasExtension) return false;
+  // Edge case for urls like https://example.com/.pdf should not be considered as pdf
+  const hasFileName = pathname.split('/').pop() !== '.pdf';
+  if (!hasFileName) return false;
+  return hasExtension && hasFileName;
+};