From 908f2a9877bc838dee95424510294b9aad527003 Mon Sep 17 00:00:00 2001
From: Jussi Vatjus-Anttila <jussiva@gmail.com>
Date: Mon, 2 Sep 2024 12:57:05 +0300
Subject: [PATCH] take logger in use

---
 pytest_opentmi/OpenTmiReport.py | 16 ++++++++++++----
 pytest_opentmi/plugin.py        |  4 ++++
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/pytest_opentmi/OpenTmiReport.py b/pytest_opentmi/OpenTmiReport.py
index 5700508..9a8656c 100644
--- a/pytest_opentmi/OpenTmiReport.py
+++ b/pytest_opentmi/OpenTmiReport.py
@@ -6,6 +6,7 @@
 import datetime
 import uuid
 import inspect
+import logging
 from multiprocessing.dummy import Pool as ThreadPool
 # 3rd party modules
 from opentmi_client import OpenTmiClient, Result
@@ -13,6 +14,8 @@
 # app modules
 from . import __pytest_info__
 
+logger = logging.getLogger(__name__)
+
 
 # pylint: disable=too-many-instance-attributes
 class OpenTmiReport:
@@ -283,7 +286,8 @@ def _link_session(self, session, result):  # pylint: disable=unused-argument
     def _upload_report(self, result: Result):
         try:
             data = self._client.post_result(result)
-            print(data)
+            logger.info(f"Uploaded {result.tcid} successfully, id: {data['id']}")
+            logger.debug(f"Uploaded {result.tcid} successfully, data: {data}")
             self._uploaded_success += 1
         except Exception:  # pylint: disable=broad-except
             self._uploaded_failed += 1
@@ -298,8 +302,11 @@ def _upload_reports(self, session):
         [self._link_session(session, result) for result in self.results]
 
         # pylint: disable=expression-not-assigned
-        [print(result.data) for result in self.results]
-        [print(test) for test in self.tests]
+        logger.info(f'Test results to be upload ({len(self.results)})')
+        [logger.debug(result.data) for result in self.results]
+
+        logger.info(f'Test cases to be upload ({len(self.tests)})')
+        [logger.debug(test) for test in self.tests]
 
         token = self.config.getoption("opentmi_token")
         pool = ThreadPool(10)
@@ -310,8 +317,9 @@ def _upload_reports(self, session):
             pool.map(self._upload_report, self.results)
             pool.close()
             pool.join()
+            logger.info('All results uploaded successfully')
         except Exception as error:  # pylint: disable=broad-except
-            print(error)
+            logger.error(error)
 
     @staticmethod
     def _get_test_key(item):
diff --git a/pytest_opentmi/plugin.py b/pytest_opentmi/plugin.py
index e86e101..ad88cdf 100644
--- a/pytest_opentmi/plugin.py
+++ b/pytest_opentmi/plugin.py
@@ -2,10 +2,13 @@
 pytest plugin
 """
 import os
+import logging
 # app modules
 from . import __version__
 from .OpenTmiReport import OpenTmiReport
 
+logger = logging.getLogger(__name__)
+
 
 # pylint: disable=unused-argument
 def pytest_report_header(config):
@@ -59,6 +62,7 @@ def pytest_configure(config):
             # prevent opening opentmi reporter on slave nodes (xdist)
             config._opentmi = OpenTmiReport(config)  # pylint: disable=protected-access
             config.pluginmanager.register(config._opentmi)  # pylint: disable=protected-access
+            logger.debug(f'Opentmi reporter enabled: {host}')
 
 
 def pytest_unconfigure(config):