[PATCH v1 1/4] dts: add tg node execution setup and teardown

[Juraj Linkeš]

The setup and teardown for the execution stage was meant to be run on
the tg node too, but was mistakenly omitted.

Fixes: cecfe0aabf58 ("dts: add traffic generator abstractions")
Signed-off-by: Juraj Linkeš <juraj.linkes at pantheon.tech>
---
 dts/framework/runner.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/dts/framework/runner.py b/dts/framework/runner.py
index db8e3ba96b..2f25a3e941 100644
--- a/dts/framework/runner.py
+++ b/dts/framework/runner.py
@@ -422,6 +422,7 @@ def _run_execution(
         execution_result.add_sut_info(sut_node.node_info)
         try:
             sut_node.set_up_execution(execution)
+            tg_node.set_up_execution(execution)
             execution_result.update_setup(Result.PASS)
         except Exception as e:
             self._logger.exception("Execution setup failed.")
@@ -438,6 +439,7 @@ def _run_execution(
             try:
                 self._logger.set_stage(DtsStage.execution_teardown)
                 sut_node.tear_down_execution()
+                tg_node.tear_down_execution()
                 execution_result.update_teardown(Result.PASS)
             except Exception as e:
                 self._logger.exception("Execution teardown failed.")
-- 
2.34.1