Test

Arcus92 · Jun 15, 2024 · 890fb9f · 890fb9f
1 parent a821642
commit 890fb9f
Show file tree

Hide file tree

Showing 9 changed files with 14 additions and 7 deletions.
diff --git a/src/pgsRendererInternal.ts b/src/pgsRendererInternal.ts
@@ -148,7 +148,7 @@ export class PgsRendererInternal {
             // Builds the subtitle.
             const pixelData = this.getPixelDataFromComposition(compositionObject, palette, ctxObjects);
             if (pixelData) {
-                this.context?.drawImage(pixelData, window.horizontalPosition, window.verticalPosition);
+                this.context.drawImage(pixelData, window.horizontalPosition, window.verticalPosition);
             }
         }
     }

diff --git a/tests/files/test-0.png b/tests/files/test-0.png
diff --git a/tests/files/test-0.rgba b/tests/files/test-0.rgba
diff --git a/tests/files/test-1.png b/tests/files/test-1.png
diff --git a/tests/files/test-1.rgba b/tests/files/test-1.rgba
diff --git a/tests/files/test-2.png b/tests/files/test-2.png
diff --git a/tests/files/test-2.rgba b/tests/files/test-2.rgba
diff --git a/tests/files/test.sup b/tests/files/test.sup
diff --git a/tests/pgsRenderer.test.ts b/tests/pgsRenderer.test.ts
@@ -9,22 +9,29 @@ test('load and render pgs', () => {
     const dataSup = fs.readFileSync(`${__dirname}/files/test.sup`);
 
     const canvas = document.createElement("canvas");
+    const context = canvas.getContext("2d")!;
     const renderer = new PgsRendererInternal();
     renderer.setCanvas(canvas);
     renderer.loadFromBuffer(dataSup);
 
     // Helper function to render and compare the image in the test directory.
     // Since we only set pixel data and don't use font rendering this should be deterministic on every machine.
     const expectImageAtTimestamp = (filename: string, timestamp: number) => {
-        const dataPng = fs.readFileSync(`${__dirname}/files/${filename}`, 'base64');
+        const dataExpected = fs.readFileSync(`${__dirname}/files/${filename}`);
 
         renderer.renderAtTimestamp(timestamp);
-        const dataResult = canvas.toDataURL('image/png').replace('data:image/png;base64,', '');
+        const dataResult = context.getImageData(0, 0, canvas.width, canvas.height).data;
 
-        expect(dataResult).toEqual(dataPng);
+        expect(dataResult.length).toBe(dataExpected.length);
+
+        for (let i = 0; i < dataResult.length; i++) {
+            if (dataResult[i] != dataExpected[i]) {
+                fail(`Unexpected pixel data ${filename}!`);
+            }
+        }
     }
 
-    expectImageAtTimestamp('test-0.png', 1.5);
-    expectImageAtTimestamp('test-1.png', 2.5);
-    expectImageAtTimestamp('test-2.png', 3.5);
+    expectImageAtTimestamp('test-0.rgba', 1.5);
+    expectImageAtTimestamp('test-1.rgba', 2.5);
+    expectImageAtTimestamp('test-2.rgba', 3.5);
 });