hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-17 01:03:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

Kotlin: Add an integration test for TRAP compression

Browse Source
This commit is contained in:
Ian Lynagh
2022-08-19 13:20:20 +01:00
parent 62779944e8
commit a6cee9edf9
2 changed files with 50 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
java/ql/integration-tests/posix-only/kotlin/trap_compression/test.kt Normal file
View File

@@ -0,0 +1,2 @@
class Test {
}

48
java/ql/integration-tests/posix-only/kotlin/trap_compression/test.py Normal file
View File

@@ -0,0 +1,48 @@
from create_database_utils import *
def check_extension(directory, expected_extension):
if expected_extension == '.trap':
# We start TRAP files with a comment
expected_start = b'//'
elif expected_extension == '.trap.gz':
# The GZip magic numbers
expected_start = b'\x1f\x8b'
else:
raise Exception('Unknown expected extension ' + expected_extension)
count = check_extension_worker(directory, expected_extension, expected_start)
if count != 1:
raise Exception('Expected 1 relevant file, but found ' + str(count) + ' in ' + directory)
def check_extension_worker(directory, expected_extension, expected_start):
count = 0
for f in os.listdir(directory):
x = os.path.join(directory, f)
if os.path.isdir(x):
count += check_extension_worker(x, expected_extension, expected_start)
else:
if f.startswith('test.kt') and not f.endswith('.set'):
if f.endswith(expected_extension):
with open(x, 'rb') as f_in:
content = f_in.read()
if content.startswith(expected_start):
count += 1
else:
raise Exception('Unexpected start to content of ' + x)
else:
raise Exception('Expected test.kt TRAP file to have extension ' + expected_extension + ', but found ' + x)
return count
run_codeql_database_create(['kotlinc test.kt'], test_db="default-db", db=None, lang="java")
check_extension('default-db/trap', '.trap.gz')
os.environ["CODEQL_EXTRACTOR_JAVA_OPTION_TRAP_COMPRESSION"] = "nOnE"
run_codeql_database_create(['kotlinc test.kt'], test_db="none-db", db=None, lang="java")
check_extension('none-db/trap', '.trap')
os.environ["CODEQL_EXTRACTOR_JAVA_OPTION_TRAP_COMPRESSION"] = "gzip"
run_codeql_database_create(['kotlinc test.kt'], test_db="gzip-db", db=None, lang="java")
check_extension('gzip-db/trap', '.trap.gz')
os.environ["CODEQL_EXTRACTOR_JAVA_OPTION_TRAP_COMPRESSION"] = "brotli"
run_codeql_database_create(['kotlinc test.kt'], test_db="brotli-db", db=None, lang="java")
check_extension('brotli-db/trap', '.trap.gz')
os.environ["CODEQL_EXTRACTOR_JAVA_OPTION_TRAP_COMPRESSION"] = "invalidValue"
run_codeql_database_create(['kotlinc test.kt'], test_db="invalid-db", db=None, lang="java")
check_extension('invalid-db/trap', '.trap.gz')