def test_write(self): f = Mock() f.write = Mock() downloaded_file = DownloadedFile( content=b'content', default_path='my_path', io_factory=Mock(), ) downloaded_file.write(f) f.write.assert_called_once_with(b'content')
def test_save_with_arg(self): f = Mock() f.write = Mock() ctx = Mock() ctx.__enter__ = Mock(return_value=f) ctx.__exit__ = Mock(return_value=False) io_factory = Mock(return_value=ctx) downloaded_file = DownloadedFile( content=b'content', default_path='my_path', io_factory=io_factory, ) downloaded_file.save('new_path') io_factory.assert_called_once_with('new_path', 'wb') f.write.assert_called_once_with(b'content')