Add environment support to the discover step

tests/env/main.fmf

@@ -0,0 +1,4 @@
+summary: Check environment variables handling
+tier: 2
+environment:
+    TESTING: WORKS

tests/env/test.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+# vim: dict+=/usr/share/beakerlib/dictionary.vim cpt=.,w,b,u,t,i,k
+
+# Include Beaker environment
+. /usr/share/beakerlib/beakerlib.sh || exit 1
+
+PACKAGE="tmt"
+
+rlJournalStart
+    rlPhaseStartTest
+        rlRun "echo $TESTING | grep WORKS"
+    rlPhaseEnd
+rlJournalEnd

tmt/base.py

@@ -192,7 +192,7 @@ def lint(self):
         elif len(self.summary) > 50:
             echo(verdict(2, 'summary should not exceed 50 characters'))
 
-    def export(self, format_='yaml', keys=None):
+    def export(self, format_='yaml', keys=None, environment=None):
         """
         Export test data into requested format
 
@@ -210,7 +210,13 @@ def export(self, format_='yaml', keys=None):
         data['path'] = self.path
         if self.duration is not None:
             data['duration'] = self.duration
-        if self.environment is not None:
+        # Combine environment variables (plan overrides test)
+        if self.environment is not None or environment is not None:
+            combined_environment = dict()
+            if self.environment:
+                combined_environment.update(self.environment)
+            if environment:
+                combined_environment.update(environment)
             data['environment'] = ' '.join(
                 tmt.utils.shell_variables(self.environment))
         return name, data
@@ -249,6 +255,9 @@ def __init__(self, node, run=None):
             if not isinstance(gates, list):
                 gates = [gates]
 
+        # Environment variables
+        self.environment = node.get('environment')
+
     @staticmethod
     def overview(tree):
         """ Show overview of available plans """
@@ -300,6 +309,9 @@ def show(self):
         self.ls(summary=True)
         for step in self.steps(disabled=True):
             step.show()
+        if self.environment is not None:
+            echo(tmt.utils.format(
+                'environment', self.environment, key_color='blue'))
         if self.opt('verbose'):
             self._sources()
 

tmt/steps/discover/__init__.py

@@ -20,7 +20,9 @@ def save(self):
         """ Save step data to the workdir """
         super(Discover, self).save()
         # Create 'tests.yaml' with the list of tests for the executor
-        tests = dict([test.export(format_='execute') for test in self.tests()])
+        tests = dict([
+            test.export(format_='execute', environment=self.plan.environment)
+            for test in self.tests()])
         self.write('tests.yaml', tmt.utils.dictionary_to_yaml(tests))
 
     def wake(self):

tmt/utils.py

@@ -438,6 +438,11 @@ def format(
         # Otherwise just place each item on a new line
         else:
             output += ('\n' + indent_string).join(value)
+    # Dictionary
+    elif isinstance(value, dict):
+        # Place each key value pair on a separate line
+        output += ('\n' + indent_string).join(
+            f'{item[0]}: {item[1]}' for item in value.items())
     # Text
     elif isinstance(value, str):
         # In 'auto' mode enable wrapping when long lines present