Merged with the default branch.

1 files changed, 25 insertions, 7 deletions

diff --git a/test/unit/applications/lang/go.py b/test/unit/applications/lang/go.py
index e4ab8ffa..15ac1cd9 100644
--- a/test/unit/applications/lang/go.py
+++ b/test/unit/applications/lang/go.py
@@ -4,12 +4,22 @@ from unit.applications.proto import TestApplicationProto
 
 
 class TestApplicationGo(TestApplicationProto):
-    def load(self, script, name='app'):
+    @classmethod
+    def setUpClass(cls, complete_check=True):
+        unit = super().setUpClass(complete_check=False)
 
-        if not os.path.isdir(self.testdir + '/go'):
-            os.mkdir(self.testdir + '/go')
+        # check go module
+
+        go_app = TestApplicationGo()
+        go_app.testdir = unit.testdir
+        if go_app.prepare_env('empty', 'app').returncode == 0:
+            cls.available['modules']['go'] = []
 
-        go_app_path = self.current_dir + '/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'
@@ -19,12 +29,18 @@ class TestApplicationGo(TestApplicationProto):
                 'build',
                 '-o',
                 self.testdir + '/go/' + name,
-                go_app_path + script + '/' + name + '.go',
+                self.current_dir + '/go/' + script + '/' + name + '.go',
             ],
             env=env,
         )
+
         process.communicate()
 
+        return process
+
+    def load(self, script, name='app'):
+        self.prepare_env(script, name)
+
         self._load_conf(
             {
                 "listeners": {"*:7080": {"pass": "applications/" + script}},
@@ -32,8 +48,10 @@ class TestApplicationGo(TestApplicationProto):
                     script: {
                         "type": "external",
                         "processes": {"spare": 0},
-                        "working_directory": go_app_path + script,
-                        "executable": self.testdir + '/go/' + name,
+                        "working_directory": self.current_dir
+                        + "/go/"
+                        + script,
+                        "executable": self.testdir + "/go/" + name,
                     }
                 },
             }