Exemple #1
0
    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')
Exemple #2
0
    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')
Exemple #3
0
 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())
Exemple #4
0
 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())