0b85fe29f7
After the launch of the project, the testing infrastructure was shared with nginx project in some cases. To avoid port overlap, a decision was made to shift the port range for Unit tests. This problem was resolved a long time ago and is no longer relevant, so it is now safe to use port 8XXX range as the default, as it is more appropriate for testing purposes.
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
|
|
from unit.applications.proto import ApplicationProto
|
|
from unit.option import option
|
|
|
|
|
|
class ApplicationGo(ApplicationProto):
|
|
@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
|
|
|
|
ApplicationGo.prepare_env(script, name, static=static_build)
|
|
|
|
conf = {
|
|
"listeners": {"*:8080": {"pass": f"applications/{script}"}},
|
|
"applications": {
|
|
script: {
|
|
"type": "external",
|
|
"processes": {"spare": 0},
|
|
"working_directory": wdir,
|
|
"executable": executable,
|
|
},
|
|
},
|
|
}
|
|
|
|
self._load_conf(conf, **kwargs)
|