模拟 NamedTemporaryFile 写入和关闭文件

Question

我如何在这个例子中用 write 和 close 模拟 NamedTemporaryFile：

def submit_form(self,pdf_bytes,file_path):
      with NamedTemporaryFile(mode='w+b') as temp:            
            temp.write(pdf_bytes)
            
            with open(temp.name, 'rb') as pdf:
                file_path = self._google_cloud_storage.upload_file(file_path,pdf,"application/pdf")
                pdf.close()
            temp.close()

这是我的测试

@patch.object(tempfile,"NamedTemporaryFile",return_value = None)
@patch('builtins.open', new_callable=mock_open())
    def test_submit_form(self):
        self._payment_approval_service.submit_form(Mock(),"testfilepath.pdf") 

测试结果：

TypeError: a bytes-like object is required, not 'Mock'
    in submit_form
        temp.write(pdf_bytes)
    _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ 
    
    args = (<Mock id='140604437181680'>,), kwargs = {}
    
        @_functools.wraps(func)
        def func_wrapper(*args, **kwargs):
    >       return func(*args, **kwargs)
    E       TypeError: a bytes-like object is required, not 'Mock'
   

这意味着 NamedTemporaryFile().write 函数没有被模拟对吗？

Answer 1

你跑

self._payment_approval_service.submit_form(Mock(),"testfilepath.pdf")

其中 Mock() 是 pdf_bytes 参数，这样做

temp.write(pdf_bytes)

自然会导致你得到的错误

 bytes-like object is required, not 'Mock'

所以你应该传递类似b"test data" 的东西

不确定如何实际模拟 self

在任何情况下，您都不需要调用close()，因为您已经在使用with，它会自动执行此操作