7934dcabbc
Previously, it was necessary to support older versions of Python for compatibility. F-strings were released in Python 3.6. Python 3.5 was marked as unsupported by the end of 2020, so now it's possible to start using f-strings safely for better readability and performance.
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
|
|
from unit.applications.proto import TestApplicationProto
|
|
from unit.option import option
|
|
|
|
|
|
class TestApplicationGo(TestApplicationProto):
|
|
@staticmethod
|
|
def prepare_env(script, name='app', static=False):
|
|
try:
|
|
subprocess.check_output(['which', 'go'])
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
temp_dir = f'{option.temp_dir}/go/'
|
|
|
|
if not os.path.exists(temp_dir):
|
|
os.mkdir(temp_dir)
|
|
|
|
cache_dir = f'{option.cache_dir}/go-build'
|
|
|
|
if not os.path.exists(cache_dir):
|
|
os.mkdir(cache_dir)
|
|
|
|
env = os.environ.copy()
|
|
env['GOPATH'] = f'{option.current_dir}/build/go'
|
|
env['GOCACHE'] = cache_dir
|
|
|
|
shutil.copy2(f'{option.test_dir}/go/{script}/{name}.go', temp_dir)
|
|
|
|
if static:
|
|
args = [
|
|
'go',
|
|
'build',
|
|
'-tags',
|
|
'netgo',
|
|
'-ldflags',
|
|
'-extldflags "-static"',
|
|
'-o',
|
|
f'{temp_dir}{name}',
|
|
f'{temp_dir}{name}.go',
|
|
]
|
|
else:
|
|
args = [
|
|
'go',
|
|
'build',
|
|
'-o',
|
|
f'{temp_dir}{name}',
|
|
f'{temp_dir}{name}.go',
|
|
]
|
|
|
|
replace_path = f'{option.current_dir}/build/go/src/unit.nginx.org/go'
|
|
|
|
with open(f'{temp_dir}go.mod', 'w') as f:
|
|
f.write(
|
|
f"""module test/app
|
|
require unit.nginx.org/go v0.0.0
|
|
replace unit.nginx.org/go => {replace_path}
|
|
"""
|
|
)
|
|
|
|
if option.detailed:
|
|
print(f'\n$ GOPATH={env["GOPATH"]} {" ".join(args)}')
|
|
|
|
try:
|
|
output = subprocess.check_output(
|
|
args, env=env, cwd=temp_dir, stderr=subprocess.STDOUT
|
|
)
|
|
|
|
except KeyboardInterrupt:
|
|
raise
|
|
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
return output
|
|
|
|
def load(self, script, name='app', **kwargs):
|
|
static_build = False
|
|
|
|
wdir = f'{option.test_dir}/go/{script}'
|
|
executable = f'{option.temp_dir}/go/{name}'
|
|
|
|
if 'isolation' in kwargs and 'rootfs' in kwargs['isolation']:
|
|
wdir = "/go/"
|
|
executable = f"/go/{name}"
|
|
static_build = True
|
|
|
|
TestApplicationGo.prepare_env(script, name, static=static_build)
|
|
|
|
conf = {
|
|
"listeners": {"*:7080": {"pass": f"applications/{script}"}},
|
|
"applications": {
|
|
script: {
|
|
"type": "external",
|
|
"processes": {"spare": 0},
|
|
"working_directory": wdir,
|
|
"executable": executable,
|
|
},
|
|
},
|
|
}
|
|
|
|
self._load_conf(conf, **kwargs)
|