1 /*
2  * Copyright (c) 2022-2024 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 
16 #ifndef FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_TEST_EVENT_MANAGER_FRAME_NODE_ON_TREE_H
17 #define FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_TEST_EVENT_MANAGER_FRAME_NODE_ON_TREE_H
18 
19 #define private public
20 #define protected public
21 
22 #include "core/components_ng/base/frame_node.h"
23 namespace OHOS::Ace::NG {
24 class FrameNodeOnTree : public FrameNode {
25 
26     DECLARE_ACE_TYPE(FrameNodeOnTree, FrameNode);
27 public:
28 
29     FrameNodeOnTree(const std::string& tag, int32_t nodeId, const RefPtr<Pattern>& pattern,
30         bool isRoot = false, bool isLayoutNode = false) : FrameNode(tag, nodeId, pattern, isRoot, false)
31     {
32         onMainTree_ = true;
33     }
34 
35     static RefPtr<FrameNode> CreateFrameNode(
36         const std::string& tag, int32_t nodeId, const RefPtr<Pattern>& pattern, bool isRoot = false)
37     {
38         auto frameNode = FrameNode::CreateFrameNode(tag, nodeId, pattern, isRoot);
39         frameNode->onMainTree_ = true;
40         return frameNode;
41     }
42 
GetOrCreateFrameNode(const std::string & tag,int32_t nodeId,const std::function<RefPtr<Pattern> (void)> & patternCreator)43     static RefPtr<FrameNode> GetOrCreateFrameNode(
44         const std::string& tag, int32_t nodeId, const std::function<RefPtr<Pattern>(void)>& patternCreator)
45     {
46         auto frameNode = FrameNode::GetOrCreateFrameNode(tag, nodeId, patternCreator);
47         frameNode->onMainTree_ = true;
48         return frameNode;
49     }
50 };
51 } // namespace OHOS::Ace::NG
52 
53 #endif // FOUNDATION_ACE_FRAMEWORKS_CORE_COMPONENTS_NG_TEST_EVENT_MANAGER_FRAME_NODE_ON_TREE_H