Use unittests.mock instead of third-party mock

As long as it is available. On EL7 we still use python2-mock.
zmiklank · Jan 27, 2022 · 4860968 · 4860968
1 parent b486d0f
commit 4860968
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/rpm/distgen.spec.dg b/rpm/distgen.spec.dg
@@ -30,9 +30,9 @@ BuildRequires: make
 BuildRequires: %{pypkg}-devel
 BuildRequires: %{pypkg}-distro
 BuildRequires: %{pypkg}-jinja2
-BuildRequires: %pypkg-mock
 BuildRequires: %{meh_pypkg}pytest
 %if 0%{?rhel} && 0%{?rhel} < 8
+BuildRequires: %{pypkg}-mock
 BuildRequires: %{pypkg}-pytest-catchlog
 %endif
 BuildRequires: %{meh_pypkg}PyYAML

diff --git a/tests/unittests/test_distro_detection.py b/tests/unittests/test_distro_detection.py
@@ -1,7 +1,9 @@
-import os
-import sys
 import pytest
-from mock import patch
+
+try:
+    from unittest.mock import patch
+except ImportError:
+    from mock import patch
 
 from distgen.distro_version import detect_default_distro