From 1cf6ab256c1c1a6c2876b99d25373c2bf2731d98 Mon Sep 17 00:00:00 2001
From: Matthias Bisping <matthias.bisping@iqser.com>
Date: Sat, 2 Apr 2022 18:34:13 +0200
Subject: [PATCH] muting logger in tests

---
 config.yaml                           |  2 +-
 image_prediction/exceptions.py        |  4 ++++
 test/unit_tests/mocked_server_test.py | 17 ++++++++++++-----
 3 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/config.yaml b/config.yaml
index 91adeaa..ae589de 100644
--- a/config.yaml
+++ b/config.yaml
@@ -4,7 +4,7 @@ webserver:
   mode: $SERVER_MODE|production  # webserver mode: {development, production}
 
 service:
-  logging_level: DEBUG  # Logging level for service logger
+  logging_level: INFO  # Logging level for service logger
   progressbar: True  # Whether a progress bar over the pages of a document is displayed while processing
   batch_size: $BATCH_SIZE|32  # Number of images in memory simultaneously
   verbose: $VERBOSE|True  # Service prints document processing progress to stdout
diff --git a/image_prediction/exceptions.py b/image_prediction/exceptions.py
index 476f1e8..c91d67b 100644
--- a/image_prediction/exceptions.py
+++ b/image_prediction/exceptions.py
@@ -24,3 +24,7 @@ class UnexpectedLabelFormat(ValueError):
 
 class IncorrectInstantiation(RuntimeError):
     pass
+
+
+class IntentionalTestException(RuntimeError):
+    pass
diff --git a/test/unit_tests/mocked_server_test.py b/test/unit_tests/mocked_server_test.py
index fb50dd4..4629426 100644
--- a/test/unit_tests/mocked_server_test.py
+++ b/test/unit_tests/mocked_server_test.py
@@ -3,11 +3,18 @@ import logging
 
 import pytest
 
+from image_prediction.exceptions import IntentionalTestException
 from image_prediction.flask import make_prediction_server
 from image_prediction.utils import get_logger
 
-logger = get_logger()
-# logger.setLevel(logging.CRITICAL + 1)
+
+@pytest.fixture
+def mute_logger():
+    logger = get_logger()
+    level = logger.level
+    logger.setLevel(logging.CRITICAL + 1)
+    yield
+    logger.setLevel(level)
 
 
 def predict_fn(x: bytes):
@@ -15,7 +22,7 @@ def predict_fn(x: bytes):
     if x == 42:
         return True
     else:
-        raise RuntimeError("intentional test exception")
+        raise IntentionalTestException("This is intended.")
 
 
 @pytest.fixture
@@ -30,12 +37,12 @@ def client(server):
     return server.test_client()
 
 
-def test_server_predict_success(client):
+def test_server_predict_success(client, mute_logger):
     response = client.post("/predict", data="42")
     assert json.loads(response.data)
 
 
-def test_server_predict_failure(client):
+def test_server_predict_failure(client, mute_logger):
     response = client.post("/predict", data="13")
     assert response.status_code == 500