[Fetch Migration] Unit test updates

Signed-off-by: Kartik Ganesh <[email protected]>
kartg · Nov 17, 2023 · edac99b · edac99b
1 parent 2d72a50
commit edac99b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/FetchMigration/python/tests/test_fetch_orchestrator.py b/FetchMigration/python/tests/test_fetch_orchestrator.py
@@ -8,6 +8,7 @@
 #
 
 import copy
+import logging
 import os
 import unittest
 from unittest import mock
@@ -23,6 +24,12 @@
 
 
 class TestFetchOrchestrator(unittest.TestCase):
+    # Run before each test
+    def setUp(self) -> None:
+        logging.disable(logging.CRITICAL)
+
+    def tearDown(self) -> None:
+        logging.disable(logging.NOTSET)
 
     @patch('migration_monitor.run')
     @patch('subprocess.Popen')