[PATCH v2 1/5] dts: add tg node test run setup and teardown

[Juraj Linkeš]

The setup and teardown for the test run 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>
Reviewed-by: Luca Vizzarro <luca.vizzarro at arm.com>
Reviewed-by: Jeremy Spewock <jspewock at iol.unh.edu>
---
 dts/framework/runner.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/dts/framework/runner.py b/dts/framework/runner.py
index 4e0254f453..565e581310 100644
--- a/dts/framework/runner.py
+++ b/dts/framework/runner.py
@@ -428,6 +428,7 @@ def _run_test_run(
         test_run_result.add_sut_info(sut_node.node_info)
         try:
             sut_node.set_up_test_run(test_run_config)
+            tg_node.set_up_test_run(test_run_config)
             test_run_result.update_setup(Result.PASS)
         except Exception as e:
             self._logger.exception("Test run setup failed.")
@@ -448,6 +449,7 @@ def _run_test_run(
             try:
                 self._logger.set_stage(DtsStage.test_run_teardown)
                 sut_node.tear_down_test_run()
+                tg_node.tear_down_test_run()
                 test_run_result.update_teardown(Result.PASS)
             except Exception as e:
                 self._logger.exception("Test run teardown failed.")
-- 
2.34.1