class TestWriteOpener(unittest.TestCase):
def setUp(self):
self.write_opener = WriteOpener()
@mock.patch('tests.files.openers.test_write_opener.builtins.open')
def test_open_file_write_byte_strings(self, mock_open_function):
mock_filename = "file.csv"
mock_file = mock.Mock()
mock_open_function.return_value = mock_file
result = self.write_opener.open_file_write_byte_strings(mock_filename)
self.assertEquals(result, mock_file)
@mock.patch('tests.files.openers.test_write_opener.builtins.open')
def test_open_file_write_unicode_strings(self, mock_open_function):
mock_filename = "file.csv"
mock_file = mock.Mock()
mock_open_function.return_value = mock_file
result = self.write_opener.open_file_write_unicode_strings(mock_filename)
self.assertEquals(result, mock_file)
def setUp(self):
self.write_opener = WriteOpener()
