File size: 2,261 Bytes
2abfccb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import os
import unittest
import common_util
from unittest import mock
from petrel_client.client import Client
test_dir = os.path.dirname(os.path.realpath(__file__))
class TestRead(unittest.TestCase):
def setUp(self):
# 实例化一个 Mock 对象，用于替换 petrel_client.ceph.s3.s3_client.S3Client.get_with_info 函数
self._mock_get_with_info = mock.Mock()
self._mock_get_with_info.return_value = "23", {}
# 替换 petrel_client.ceph.s3.s3_client.S3Client.get_with_info 函数
self._patcher = mock.patch(
'petrel_client.ceph.s3.s3_client.S3Client.get_with_info',
self._mock_get_with_info)
self._patcher.start()
pass
def tearDown(self):
self._patcher.stop()
pass
# 1. 在setUp注入Mock方法，替换petrel_client.ceph.s3.s3_client.S3Client.get_with_info函数，并在tearDown停止Mock
def test_read1(self):
_conf_path = test_dir + '/conf/petreloss.conf'
c = Client(conf_path=_conf_path)
# self._mock_get_with_info.return_value = "23", {}
data = c.get('cluster1:s3://lili1.test2/sometest')
# print(data)
self.assertEqual("23", data)
# 2.使用@mock.patch指定被替换的方法，并默认按参数列表的顺序替换该方法
@mock.patch("petrel_client.ceph.s3.s3_client.S3Client.get_with_info")
#@mock.patch("...")
def test_read2(self, mock_get_with_info):
mock_get_with_info.return_value = "15", {}
_conf_path = test_dir + '/conf/petreloss.conf'
c = Client(conf_path=_conf_path)
data = c.get('cluster1:s3://lili1.test2/sometest')
self.assertEqual("15", data)
# 3.使用with mock.patch 替换掉指定模块
def test_read3(self):
mock_get_with_info = mock.Mock()
mock_get_with_info.return_value = "15", {}
_conf_path = test_dir + '/conf/petreloss.conf'
c = Client(conf_path=_conf_path)
with mock.patch(
'petrel_client.ceph.s3.s3_client.S3Client.get_with_info',
mock_get_with_info):
data = c.get('cluster1:s3://lili1.test2/sometest')
self.assertEqual("15", data)
if __name__ == '__main__':
common_util.run_test() |