1 /**
2 * Copyright (c) 2021-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 #include "get_test_class.h"
17
18 #include "ets_vm.h"
19 #include "ets_class_linker_extension.h"
20 #include "assembly-emitter.h"
21 #include "assembly-parser.h"
22
23 namespace ark::ets::test {
24
GetTestClass(const char * source,const char * className,const std::string & sourceFilename)25 EtsClass *GetTestClass(const char *source, const char *className, const std::string &sourceFilename)
26 {
27 pandasm::Parser p;
28
29 auto res = p.Parse(source, sourceFilename);
30 auto pf = pandasm::AsmEmitter::Emit(res.Value());
31
32 LanguageContext ctx = Runtime::GetCurrent()->GetLanguageContext(panda_file::SourceLang::ETS);
33
34 auto classLinker = Runtime::GetCurrent()->GetClassLinker();
35 ASSERT(classLinker);
36 classLinker->AddPandaFile(std::move(pf));
37
38 EtsCoroutine *coroutine = EtsCoroutine::GetCurrent();
39 EtsClassLinker *etsClassLinker = coroutine->GetPandaVM()->GetClassLinker();
40
41 EtsClass *klass = etsClassLinker->GetClass(className);
42 if (klass == nullptr) {
43 return nullptr;
44 }
45
46 ctx.InitializeClass(classLinker, coroutine, klass->GetRuntimeClass());
47
48 return klass;
49 }
50
51 } // namespace ark::ets::test
52