def test_save_with_message_object(): attachments = clingy.find("email.txt") mo = mock.mock_open() with mock.patch("{}.open".format(builtins_name), mo): clingy.save(attachments[0]) # Output filenames assert mock.call("foo.txt", "wb") in mo.mock_calls # Output file content handle = mo() assert mock.call(b"foo") in handle.write.mock_calls
def test_save_with_regex(): with open("email.txt") as file: email = file.read() mo = mock.mock_open() with mock.patch("{}.open".format(builtins_name), mo): clingy.save(email, glob="bar*") # Output filenames assert mock.call("bar.txt", "wb") in mo.mock_calls # Output file content handle = mo() assert mock.call(b"bar") in handle.write.mock_calls