From ea909046da4e4d38f9020b5f5131a4de66cd642d Mon Sep 17 00:00:00 2001
From: Yu Ming Zhu <yzhu@redhat.com>
Date: Mar 22 2024 22:43:58 +0000
Subject: fix unittests


---

diff --git a/tests/test_add_tag_handler.py b/tests/test_add_tag_handler.py
index bf7b3a5..356433a 100644
--- a/tests/test_add_tag_handler.py
+++ b/tests/test_add_tag_handler.py
@@ -22,7 +22,6 @@
 # Written by Chenxiong Qi <cqi@redhat.com>
 #            Qixiang Wan <qwan@redhat.com>
 
-from configparser import ConfigParser
 import json
 import mock
 import os
@@ -30,10 +29,15 @@ import shutil
 import tempfile
 from argparse import Namespace
 
+import six
 from tests import UrsaMajorTestCase, MockResponse, make_mmd, TEST_DATA_DIR
 from ursa_major import ModuleConfig, MBS_BUILD_STATES
 from ursa_major.handlers.add_tag import ModuleInfo, AddTagHandler
 from ursa_major.mbs import MBS
+if six.PY3:  # SafeConfigParser == ConfigParser, former deprecated in >= 3.2
+    from six.moves.configparser import ConfigParser
+else:
+    from six.moves.configparser import SafeConfigParser as ConfigParser
 
 
 class AddTagHandlerTestCase(UrsaMajorTestCase):
diff --git a/tests/test_update_tag.py b/tests/test_update_tag.py
index d4be06e..1a8f897 100644
--- a/tests/test_update_tag.py
+++ b/tests/test_update_tag.py
@@ -22,7 +22,6 @@
 # Written by Chenxiong Qi <cqi@redhat.com>
 #            Qixiang Wan <qwan@redhat.com>
 
-from configparser import ConfigParser
 import json
 import mock
 import os
@@ -40,6 +39,11 @@ from tests import TEST_DATA_DIR
 from ursa_major import MBS_BUILD_STATES
 from ursa_major.handlers.update_tag import UpdateTagHandler
 
+if six.PY3:  # SafeConfigParser == ConfigParser, former deprecated in >= 3.2
+    from six.moves.configparser import ConfigParser
+else:
+    from six.moves.configparser import SafeConfigParser as ConfigParser
+
 
 class TestUpdateTagHandler(unittest.TestCase):
     def setUp(self):