def test_open_for_write_file_exists(self):
filename = 'foo/bar/new_file'
with mock.patch('io.open',
mock.mock_open(),
create=True) as mock_open, \
mock.patch('os.path.exists', return_value=True):
utils._open_for_write(filename)
mock_open.assert_called_once_with(filename, 'w')
def test_open_for_write_file_does_not_exist(self):
filename = 'foo/bar/new_file'
with mock.patch('io.open',
mock.mock_open(),
create=True) as mock_open, \
mock.patch('os.path.exists', return_value=False), \
mock.patch('os.makedirs') as mock_makedirs:
utils._open_for_write(filename)
mock_makedirs.assert_called_once_with(os.path.dirname(filename))
mock_open.assert_called_once_with(filename, 'w')
def test_open_for_write(self):
filename = 'foo/bar/new_file'
with utils._open_for_write(filename) as f:
f.write('foo')
with open(filename) as f:
self.assertEqual('foo', f.read())
def test_open_for_write(self):
filename = 'foo/bar/new_file'
with utils._open_for_write(filename) as f:
f.write('foo')
with open(filename) as f:
self.assertEqual('foo', f.read())