pytest如何模拟s3fs.S3FileSystem打开文件

问题描述

我正在尝试模拟在S3存储桶中打开文件的调用。我拥有的代码是：

# mymodule.py
import s3fs
#...
def __init__(self):
    self.s3_filesystem = s3fs.S3FileSystem(anon=False,key=s3_key,secret=s3_secret)
#...
def mymethod(self):
    with self.s3_filesystem.open(filename,'r') as csv_file:
        file_dataframe = pd.read_csv(csv_file)

我在pytest中的模拟是：

import pytest

def test(mocker)                                       
    mocker.patch('mypackage.mymodule.s3fs.S3FileSystem.open',return_value=open)

但是在运行测试时我得到了错误：

>       with self.s3_filesystem.open(filename,'r') as csv_file:
E       AttributeError: __enter__

知道为什么吗？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）