| 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", {} | |
| self._mock_get_with_info.return_value = None, {} | |
| # 替换 petrel_client.ceph.s3.s3_client.S3Client.get_with_info 函数 | |
| self._patcher = mock.patch( | |
| '', | |
| 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 = '/home/PJLAB/huipeng/petreloss.conf' | |
| c = Client(conf_path=_conf_path) | |
| # self._mock_get_with_info.return_value = "23", {} | |
| # data = c.get('cluster1:s3://lili1.test2/sometest') | |
| data = c.get("s3://hptestbucket/a/b/c.go") | |
| print("data is ", 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() |