1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import java_types
def Generate(jni_obj, *, gen_jni_class, script_name):
proxy_class = java_types.JavaClass(
f'{jni_obj.java_class.full_name_with_slashes}Jni')
visibility = 'public ' if jni_obj.proxy_visibility == 'public' else ''
interface_name = jni_obj.proxy_interface.name_with_dots
gen_jni = gen_jni_class.name
type_resolver = java_types.TypeResolver(proxy_class)
type_resolver.imports = list(jni_obj.type_resolver.imports)
sb = []
sb.append(f"""\
//
// This file was generated by {script_name}
//
package {jni_obj.java_class.class_without_prefix.package_with_dots};
import org.jni_zero.CheckDiscard;
import org.jni_zero.JniStaticTestMocker;
import org.jni_zero.NativeLibraryLoadedStatus;
import {gen_jni_class.full_name_with_dots};
""")
# Copy over all imports (some will be unused, but oh well).
for c in type_resolver.imports:
sb.append(f'import {c.full_name_with_dots};\n')
sb.append(f"""
@CheckDiscard("crbug.com/993421")
{visibility}class {proxy_class.name} implements {interface_name} {{
private static {interface_name} testInstance;
public static final JniStaticTestMocker<{interface_name}> TEST_HOOKS =
new JniStaticTestMocker<{interface_name}>() {{
@Override
public void setInstanceForTesting({interface_name} instance) {{
if (!{gen_jni}.TESTING_ENABLED) {{
throw new RuntimeException(
"Tried to set a JNI mock when mocks aren't enabled!");
}}
testInstance = instance;
}}
}};
""")
for native in jni_obj.proxy_natives:
call_params = native.params.to_call_str()
sig_params = native.params.to_java_declaration(type_resolver)
return_type_str = native.return_type.to_java(type_resolver)
return_prefix = ''
if not native.return_type.is_void():
return_prefix = f'return ({return_type_str}) '
sb.append(f"""
@Override
public {return_type_str} {native.name}({sig_params}) {{
{return_prefix}{gen_jni}.{native.proxy_name}({call_params});
}}
""")
sb.append(f"""
public static {interface_name} get() {{
if ({gen_jni}.TESTING_ENABLED) {{
if (testInstance != null) {{
return testInstance;
}}
if ({gen_jni}.REQUIRE_MOCK) {{
throw new UnsupportedOperationException(
"No mock found for the native implementation of {interface_name}. "
+ "The current configuration requires implementations be mocked.");
}}
}}
NativeLibraryLoadedStatus.checkLoaded();
return new {proxy_class.name}();
}}
}}
""")
return ''.join(sb)
|
