c183bd8749
Class usage came from the unittest framework and it was always redundant after migration to the pytest. This commit removes classes from files containing tests to make them more readable and understandable.
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": {"*:7080": {"pass": f"applications/{script}"}},
|
|
"applications": {
|
|
script: {
|
|
"type": "external",
|
|
"processes": {"spare": 0},
|
|
"working_directory": wdir,
|
|
"executable": executable,
|
|
},
|
|
},
|
|
}
|
|
|
|
self._load_conf(conf, **kwargs)
|