提交 2a4e9a85 编写于 作者: C Cleber Rosa

selftests/unit/test_distro.py: port from flexmock to mock (unittest.mock)

The first line says it all, this is a pretty straightforward port.
Signed-off-by: NCleber Rosa <crosa@redhat.com>
上级 5a147faf
import os
import re
import unittest
from flexmock import flexmock
try:
from unittest import mock
except ImportError:
import mock
from avocado.utils import distro
......@@ -64,10 +66,11 @@ class ProbeTest(unittest.TestCase):
CHECK_FILE = distro_file
CHECK_FILE_DISTRO_NAME = distro_name
flexmock(os.path)
os.path.should_receive('exists').and_return(True)
my_probe = MyProbe()
with mock.patch('avocado.utils.distro.os.path.exists',
return_value=True) as mocked:
probed_distro_name = my_probe.name_for_file()
mocked.assert_called_once_with(distro_file)
self.assertEqual(distro_name, probed_distro_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册