python使用Mock

最新推荐文章于 2024-08-07 15:09:54 发布

不迷信_只迷人

最新推荐文章于 2024-08-07 15:09:54 发布

阅读量558

点赞数

分类专栏： Python 学习文章标签： python

本文链接：https://blog.csdn.net/honeyaya/article/details/105619763

版权

Python 学习专栏收录该内容

31 篇文章 0 订阅

订阅专栏

文章目录

背景

aws的lambda function使用python完成，python连接pymsql，涉及到的UT怎么写？

问题一：使用mock调用lambda function

conn = boto3.resource('s3', region_name='us-east-1')
conn.create_bucket(Bucket="test")

boto3.client('s3', region_name='us-east-1').put_object(Bucket="test", Key="123456789_20190419_0/0/20200419/20200419.publish", Body="testpublish")

event = s3_object_created_event("test", "123456789_20190419_0/0/20200419/20200419.publish")

参考: http://joshuaballoch.github.io/testing-lambda-functions/
about moto: http://docs.getmoto.org/en/latest/docs/getting_started.html

问题二：使用mock操作RDS(mysql)

主要使用patch和side_effect

@patch("pymysql.connect")
@patch("lambda_function.check")
@patch("lambda_function.insert")
@patch("lambda_function.connect")
@patch("lambda_function.send_email")
def test_lambda_handler(self, mock_send_email, mock_connect, mock_insert, mock_check, mock_pyconnect):

error情况

with self.assertRaises(SystemExit) as cm:
    lambda_function.lambda_handler(event, None)
self.assertEqual(cm.exception.code, -4)

相关参考

https://blog.csdn.net/lk142500/article/details/79582176
https://realpython.com/python-testing/
https://docs.python.org/3/library/unittest.mock.html
http://note.qidong.name/2018/02/pytest-mock/
http://devopstarter.info/three-years-later-test-mock/
https://www.programcreek.com/python/example/105254/moto.mock_s3
http://doc.codingdict.com/python_352/library/unittest.mock-examples.html