mirror of https://github.com/electron/electron
129 lines
4.9 KiB
Python
Executable File
129 lines
4.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import glob
|
|
import os
|
|
import platform
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
|
|
from lib.util import get_electron_branding, rm_rf, scoped_cwd
|
|
|
|
PROJECT_NAME = get_electron_branding()['project_name']
|
|
PRODUCT_NAME = get_electron_branding()['product_name']
|
|
SOURCE_ROOT = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
|
|
SNAPSHOT_SOURCE = os.path.join(SOURCE_ROOT, 'spec', 'fixtures', 'testsnap.js')
|
|
|
|
def main():
|
|
args = parse_args()
|
|
|
|
source_root = os.path.abspath(args.source_root)
|
|
initial_app_path = os.path.join(source_root, args.build_dir)
|
|
app_path = create_app_copy(initial_app_path)
|
|
|
|
returncode = 0
|
|
try:
|
|
with scoped_cwd(app_path):
|
|
if args.snapshot_files_dir is None:
|
|
snapshot_filename = os.path.join(app_path, 'mksnapshot_args')
|
|
with open(snapshot_filename, encoding='utf-8') as file_in:
|
|
mkargs = file_in.read().splitlines()
|
|
print('running: ' + ' '.join(mkargs + [ SNAPSHOT_SOURCE ]))
|
|
subprocess.check_call(mkargs + [ SNAPSHOT_SOURCE ], cwd=app_path)
|
|
print('ok mksnapshot successfully created snapshot_blob.bin.')
|
|
context_snapshot = 'v8_context_snapshot.bin'
|
|
if platform.system() == 'Darwin':
|
|
if os.environ.get('TARGET_ARCH') == 'arm64':
|
|
context_snapshot = 'v8_context_snapshot.arm64.bin'
|
|
else:
|
|
context_snapshot = 'v8_context_snapshot.x86_64.bin'
|
|
context_snapshot_path = os.path.join(app_path, context_snapshot)
|
|
gen_binary = get_binary_path('v8_context_snapshot_generator', \
|
|
app_path)
|
|
genargs = [ gen_binary, \
|
|
f'--output_file={context_snapshot_path}' ]
|
|
print('running: ' + ' '.join(genargs))
|
|
subprocess.check_call(genargs)
|
|
print('ok v8_context_snapshot_generator successfully created ' \
|
|
+ context_snapshot)
|
|
if args.create_snapshot_only:
|
|
return 0
|
|
else:
|
|
gen_bin_path = os.path.join(args.snapshot_files_dir, '*.bin')
|
|
generated_bin_files = glob.glob(gen_bin_path)
|
|
for bin_file in generated_bin_files:
|
|
shutil.copy2(bin_file, app_path)
|
|
|
|
test_path = os.path.join(SOURCE_ROOT, 'spec', 'fixtures', \
|
|
'snapshot-items-available')
|
|
|
|
if sys.platform == 'darwin':
|
|
bin_files = glob.glob(os.path.join(app_path, '*.bin'))
|
|
app_dir = os.path.join(app_path, f'{PRODUCT_NAME}.app')
|
|
electron = os.path.join(app_dir, 'Contents', 'MacOS', PRODUCT_NAME)
|
|
bin_out_path = os.path.join(app_dir, 'Contents', 'Frameworks',
|
|
f'{PROJECT_NAME} Framework.framework',
|
|
'Resources')
|
|
for bin_file in bin_files:
|
|
shutil.copy2(bin_file, bin_out_path)
|
|
elif sys.platform == 'win32':
|
|
electron = os.path.join(app_path, f'{PROJECT_NAME}.exe')
|
|
else:
|
|
electron = os.path.join(app_path, PROJECT_NAME)
|
|
|
|
print('running: ' + ' '.join([electron, test_path]))
|
|
subprocess.check_call([electron, test_path])
|
|
print('ok successfully used custom snapshot.')
|
|
except subprocess.CalledProcessError as e:
|
|
print('not ok an error was encountered while testing mksnapshot.')
|
|
print(e)
|
|
returncode = e.returncode
|
|
except KeyboardInterrupt:
|
|
print('Other error')
|
|
returncode = 0
|
|
print(f'Returning with error code: {returncode}')
|
|
return returncode
|
|
|
|
|
|
# Create copy of app to install custom snapshot
|
|
def create_app_copy(initial_app_path):
|
|
print('Creating copy of app for testing')
|
|
app_path = os.path.join(os.path.dirname(initial_app_path),
|
|
os.path.basename(initial_app_path)
|
|
+ '-mksnapshot-test')
|
|
|
|
rm_rf(app_path)
|
|
shutil.copytree(initial_app_path, app_path, symlinks=True)
|
|
return app_path
|
|
|
|
def get_binary_path(binary_name, root_path):
|
|
if sys.platform == 'win32':
|
|
binary_path = os.path.join(root_path, f'{binary_name}.exe')
|
|
else:
|
|
binary_path = os.path.join(root_path, binary_name)
|
|
return binary_path
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description='Test mksnapshot')
|
|
parser.add_argument('-b', '--build-dir',
|
|
help='Path to an Electron build folder. \
|
|
Relative to the --source-root.',
|
|
default=None,
|
|
required=True)
|
|
parser.add_argument('--create-snapshot-only',
|
|
help='Just create snapshot files, but do not run test',
|
|
action='store_true')
|
|
parser.add_argument('--snapshot-files-dir',
|
|
help='Directory containing snapshot files to use \
|
|
for testing',
|
|
default=None,
|
|
required=False)
|
|
parser.add_argument('--source-root',
|
|
default=SOURCE_ROOT,
|
|
required=False)
|
|
return parser.parse_args()
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|