Skip to content

Commit c12fb48

Browse files
committed
Produce qlformat compliant shared query test files
1 parent 272596a commit c12fb48

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

scripts/generate_rules/generate_package_files.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,19 @@ def write_shared_implementation(package_name, rule_id, query, language_name, ql_
181181
.replace("/", ".")
182182
+ "\n"
183183
)
184-
f.write("\n");
185-
f.write("class TestFileQuery extends " + str(query["shared_implementation_short_name"]) + "SharedQuery, TestQuery { }\n")
184+
f.write("\n")
185+
class_name = str(query["shared_implementation_short_name"]) + "SharedQuery"
186+
f.write("class TestFileQuery extends " + class_name + ",")
187+
# ql formatting of this line depends on the line length
188+
if len(class_name) > 61:
189+
# Line break required after comma
190+
f.write("\n TestQuery\n{ }\n")
191+
elif len(class_name) > 57:
192+
# Line break required after `{`
193+
f.write(" TestQuery {\n}\n")
194+
else:
195+
# Under 100 characters, can be formatted on the same line
196+
f.write(" TestQuery { }\n")
186197

187198
# Create an empty test file, if one doesn't already exist
188199
shared_impl_test_dir.joinpath(

0 commit comments

Comments
 (0)