From c3288baa89c8745d1bb1dc5774942dc4d51583cc Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Feb 03 2025 13:14:15 +0000
Subject: choose correct import machinery in unit test


Fixes https://pagure.io/koji/issue/4306

---

diff --git a/tests/test_lib/test_plugin.py b/tests/test_lib/test_plugin.py
index 24b3683..d80e76e 100644
--- a/tests/test_lib/test_plugin.py
+++ b/tests/test_lib/test_plugin.py
@@ -9,17 +9,14 @@ from six.moves import range
 import sys
 import unittest
 
-try:
-    import imp
-    importlib = None
-except ImportError:
-    import importlib
-    imp = None
-
 import koji
 import koji.util
 import koji.plugin
 
+# check which import machinery the lib is using
+imp = getattr(koji.plugin, 'imp', None)
+importlib = koji.plugin.importlib
+
 
 class TestCallbackDecorators(unittest.TestCase):