如何在while循环中模拟boto3资源方法进行单元测试

import unittest
from unittest.mock import MagicMock

class MyClass:
    def __init__(self):
        self.client = boto3.client('s3')
    
    def my_method(self):
        while True:
            response = self.client.list_buckets()
            # 其他逻辑处理

class MyTest(unittest.TestCase):
    def test_my_method(self):
        # 创建一个Mock对象来模拟boto3.client方法的返回值
        mock_client = MagicMock()
        mock_client.list_buckets.return_value = {'Buckets': [{'Name': 'bucket1'}, {'Name': 'bucket2'}]}
        
        # 创建一个MyClass对象，并将mock_client赋值给其client属性
        my_obj = MyClass()
        my_obj.client = mock_client
        
        # 调用待测试的方法
        my_obj.my_method()
        
        # 断言相关的方法是否被调用
        mock_client.list_buckets.assert_called_once()

if __name__ == '__main__':
    unittest.main()