66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
import os
|
|
from subprocess import Popen
|
|
from unit.applications.proto import TestApplicationProto
|
|
|
|
|
|
class TestApplicationGo(TestApplicationProto):
|
|
@classmethod
|
|
def setUpClass(cls, complete_check=True):
|
|
unit = super().setUpClass(complete_check=False)
|
|
|
|
# check go module
|
|
|
|
go_app = TestApplicationGo()
|
|
go_app.testdir = unit.testdir
|
|
proc = go_app.prepare_env('empty', 'app')
|
|
if proc and proc.returncode == 0:
|
|
cls.available['modules']['go'] = []
|
|
|
|
return unit if not complete_check else unit.complete()
|
|
|
|
def prepare_env(self, script, name):
|
|
if not os.path.exists(self.testdir + '/go'):
|
|
os.mkdir(self.testdir + '/go')
|
|
|
|
env = os.environ.copy()
|
|
env['GOPATH'] = self.pardir + '/go'
|
|
|
|
try:
|
|
process = Popen(
|
|
[
|
|
'go',
|
|
'build',
|
|
'-o',
|
|
self.testdir + '/go/' + name,
|
|
self.current_dir + '/go/' + script + '/' + name + '.go',
|
|
],
|
|
env=env,
|
|
)
|
|
|
|
process.communicate()
|
|
|
|
except:
|
|
return None
|
|
|
|
return process
|
|
|
|
def load(self, script, name='app', **kwargs):
|
|
self.prepare_env(script, name)
|
|
|
|
self._load_conf(
|
|
{
|
|
"listeners": {"*:7080": {"pass": "applications/" + script}},
|
|
"applications": {
|
|
script: {
|
|
"type": "external",
|
|
"processes": {"spare": 0},
|
|
"working_directory": self.current_dir
|
|
+ "/go/"
|
|
+ script,
|
|
"executable": self.testdir + "/go/" + name,
|
|
}
|
|
},
|
|
},
|
|
**kwargs
|
|
)
|