feat: allow implicit detection of context objects

software-mansion · Jul 17, 2024 · 4c2e509 · 4c2e509
1 parent ce914e6
commit 4c2e509
Show file tree

Hide file tree

Showing 10 changed files with 487 additions and 205 deletions.
diff --git a/apps/common-app/src/examples/RuntimeTests/RuntimeTestsExample.tsx b/apps/common-app/src/examples/RuntimeTests/RuntimeTestsExample.tsx
@@ -69,7 +69,7 @@ export default function RuntimeTestsExample() {
           },
         },
         {
-          testSuiteName: 'babelPlugin',
+          testSuiteName: 'babel plugin',
           importTest: () => {
             require('./tests/plugin/fileWorkletization.test');
             require('./tests/plugin/contextObjects.test');

diff --git a/apps/common-app/src/examples/RuntimeTests/tests/plugin/contextObjects.test.tsx b/apps/common-app/src/examples/RuntimeTests/tests/plugin/contextObjects.test.tsx
@@ -22,7 +22,7 @@ describe('Test context objects', () => {
         foo() {
           return 1;
         },
-        __workletObject: true,
+        __workletContextObject: true,
       };
 
       useEffect(() => {
@@ -45,7 +45,7 @@ describe('Test context objects', () => {
       registerValue(SHARED_VALUE_REF, output);
       const contextObject = {
         foo: () => 1,
-        __workletObject: true,
+        __workletContextObject: true,
       };
 
       useEffect(() => {
@@ -73,7 +73,7 @@ describe('Test context objects', () => {
         bar() {
           return this.foo() + 1;
         },
-        __workletObject: true,
+        __workletContextObject: true,
       };
 
       useEffect(() => {
@@ -101,7 +101,7 @@ describe('Test context objects', () => {
         bar() {
           return this.foo.call(contextObject) + 1;
         },
-        __workletObject: true,
+        __workletContextObject: true,
       };
 
       useEffect(() => {
@@ -127,7 +127,7 @@ describe('Test context objects', () => {
         bar() {
           this.foo += 1;
         },
-        __workletObject: true,
+        __workletContextObject: true,
       };
 
       useEffect(() => {
@@ -154,7 +154,7 @@ describe('Test context objects', () => {
         bar() {
           this.foo += 1;
         },
-        __workletObject: true,
+        __workletContextObject: true,
       };
 
       useEffect(() => {

diff --git a/apps/common-app/src/examples/RuntimeTests/tests/plugin/fileWorkletization.test.tsx b/apps/common-app/src/examples/RuntimeTests/tests/plugin/fileWorkletization.test.tsx
@@ -10,28 +10,46 @@ import {
   test,
   expect,
 } from '../../ReanimatedRuntimeTestsRunner/RuntimeTestsApi';
-import { getThree } from './fileWorkletization';
+import { getThree, implicitContextObject } from './fileWorkletization';
 
 const SHARED_VALUE_REF = 'SHARED_VALUE_REF';
 
-describe('Test workletization', () => {
-  const ExampleComponent = () => {
-    const output = useSharedValue<number | null>(null);
-    registerValue(SHARED_VALUE_REF, output);
+describe('Test file workletization', () => {
+  test('Functions and methods are workletized', async () => {
+    const ExampleComponent = () => {
+      const output = useSharedValue<number | null>(null);
+      registerValue(SHARED_VALUE_REF, output);
 
-    useEffect(() => {
-      runOnUI(() => {
-        output.value = getThree();
-      })();
-    });
+      useEffect(() => {
+        runOnUI(() => {
+          output.value = getThree();
+        })();
+      });
 
-    return <View />;
-  };
-
-  test('Test file workletization', async () => {
+      return <View />;
+    };
     await render(<ExampleComponent />);
     await wait(100);
     const sharedValue = await getRegisteredValue(SHARED_VALUE_REF);
     expect(sharedValue.onUI).toBe(3);
   });
+
+  test('WorkletContextObjects are workletized', async () => {
+    const ExampleComponent = () => {
+      const output = useSharedValue<number | null>(null);
+      registerValue(SHARED_VALUE_REF, output);
+
+      useEffect(() => {
+        runOnUI(() => {
+          output.value = implicitContextObject.getFive();
+        })();
+      });
+
+      return <View />;
+    };
+    await render(<ExampleComponent />);
+    await wait(100);
+    const sharedValue = await getRegisteredValue(SHARED_VALUE_REF);
+    expect(sharedValue.onUI).toBe(5);
+  });
 });
diff --git a/apps/common-app/src/examples/RuntimeTests/tests/plugin/fileWorkletization.ts b/apps/common-app/src/examples/RuntimeTests/tests/plugin/fileWorkletization.ts
@@ -13,3 +13,12 @@ const getterContainer = {
 export const getThree = () => {
   return getOne() + getterContainer.getTwo();
 };
+
+export const implicitContextObject = {
+  getFour() {
+    return 4;
+  },
+  getFive() {
+    return this.getFour() + 1;
+  },
+};