unittest: use unittest.mock instead of mock

because the absence of unittest.mock on python2.7, we still fallback to
mock
This commit is contained in:
Yuming Zhu 2024-10-18 13:05:09 +00:00
parent 589e6bbb96
commit ca05418fb5
390 changed files with 966 additions and 451 deletions

View file

@ -6,9 +6,13 @@ import unittest
import pytest
import protonmsg
import mock
try:
from unittest import mock
from unittest.mock import patch, MagicMock
except ImportError:
import mock
from mock import patch, MagicMock
import six
from mock import patch, MagicMock
if six.PY2:
from ConfigParser import SafeConfigParser as ConfigParser
else: