1/* 2 * Copyright (c) 2025 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16import * as path from 'path'; 17import { PluginTestContext, PluginTester } from '../../../utils/plugin-tester'; 18import { BuildConfig, mockBuildConfig } from '../../../utils/artkts-config'; 19import { getRootPath, MOCK_ENTRY_DIR_PATH } from '../../../utils/path-config'; 20import { parseDumpSrc } from '../../../utils/parse-string'; 21import { memoNoRecheck } from '../../../utils/plugins'; 22 23const METHOD_DIR_PATH: string = 'memo/methods'; 24 25const buildConfig: BuildConfig = mockBuildConfig(); 26buildConfig.compileFiles = [ 27 path.resolve(getRootPath(), MOCK_ENTRY_DIR_PATH, METHOD_DIR_PATH, 'declare-and-call.ets'), 28]; 29 30const pluginTester = new PluginTester('test memo method', buildConfig); 31 32const expectedScript: string = ` 33import { memo as memo, __memo_context_type as __memo_context_type, __memo_id_type as __memo_id_type } from \"@ohos.arkui.stateManagement\"; 34function main() {} 35@memo() ((__memo_context: __memo_context_type, __memo_id: __memo_id_type): void => { 36 const __memo_scope = __memo_context.scope<void>(((__memo_id) + (<some_random_number>)), 0); 37 if (__memo_scope.unchanged) { 38 __memo_scope.cached; 39 return; 40 } 41 new AA().x(__memo_context, ((__memo_id) + (<some_random_number>))); 42 const a: A = new AA(); 43 a.x(__memo_context, ((__memo_id) + (<some_random_number>))); 44 { 45 __memo_scope.recache(); 46 return; 47 } 48}); 49declare abstract class A { 50 @memo() public x(__memo_context: __memo_context_type, __memo_id: __memo_id_type): void 51 public test_signature(@memo() arg1: ((__memo_context: __memo_context_type, __memo_id: __memo_id_type)=> void), @memo() arg2: ((__memo_context: __memo_context_type, __memo_id: __memo_id_type)=> void) | undefined, @memo() arg3: ((__memo_context: __memo_context_type, __memo_id: __memo_id_type)=> void) | undefined | ((__memo_context: __memo_context_type, __memo_id: __memo_id_type)=> int) | undefined, @memo() x: ((__memo_context: __memo_context_type, __memo_id: __memo_id_type, y: ((z: @memo() ((__memo_context: __memo_context_type, __memo_id: __memo_id_type)=> void))=> void))=> void)): @memo() ((__memo_context: __memo_context_type, __memo_id: __memo_id_type)=> void) 52 public constructor() {} 53} 54class AA extends A { 55 public x(__memo_context: __memo_context_type, __memo_id: __memo_id_type): void { 56 const __memo_scope = __memo_context.scope<void>(((__memo_id) + (<some_random_number>)), 0); 57 if (__memo_scope.unchanged) { 58 __memo_scope.cached; 59 return; 60 } 61 { 62 __memo_scope.recache(); 63 return; 64 } 65 } 66 public constructor() {} 67} 68`; 69 70function testMemoTransformer(this: PluginTestContext): void { 71 expect(parseDumpSrc(this.scriptSnapshot ?? '')).toBe(parseDumpSrc(expectedScript)); 72} 73 74pluginTester.run( 75 'transform declare methods and calls', 76 [memoNoRecheck], 77 { 78 'checked:memo-no-recheck': [testMemoTransformer], 79 }, 80 { 81 stopAfter: 'checked', 82 } 83); 84