57 lines
2 KiB
Python
57 lines
2 KiB
Python
from __future__ import absolute_import
|
|
|
|
import unittest
|
|
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
import errno
|
|
|
|
from koji import ensuredir
|
|
|
|
|
|
class TestEnsureDir(unittest.TestCase):
|
|
@mock.patch('os.mkdir')
|
|
@mock.patch('os.path.exists')
|
|
@mock.patch('os.path.isdir')
|
|
def test_ensuredir_errors(self, mock_isdir, mock_exists, mock_mkdir):
|
|
mock_exists.return_value = False
|
|
with self.assertRaises(OSError) as cm:
|
|
ensuredir('/')
|
|
self.assertEqual(cm.exception.args[0], 'root directory missing? /')
|
|
mock_mkdir.assert_not_called()
|
|
|
|
mock_exists.return_value = True
|
|
mock_isdir.return_value = False
|
|
with self.assertRaises(OSError) as cm:
|
|
ensuredir('/path/foo/bar')
|
|
self.assertEqual(cm.exception.args[0],
|
|
'Not a directory: /path/foo/bar')
|
|
mock_mkdir.assert_not_called()
|
|
|
|
mock_exists.return_value = False
|
|
mock_isdir.return_value = False
|
|
mock_mkdir.side_effect = OSError(errno.EEXIST, 'error msg')
|
|
with self.assertRaises(OSError) as cm:
|
|
ensuredir('path')
|
|
self.assertEqual(cm.exception.args[0], errno.EEXIST)
|
|
mock_mkdir.assert_called_once_with('path')
|
|
|
|
mock_mkdir.reset_mock()
|
|
mock_mkdir.side_effect = OSError(errno.EEXIST, 'error msg')
|
|
mock_isdir.return_value = True
|
|
ensuredir('path')
|
|
mock_mkdir.assert_called_once_with('path')
|
|
|
|
@mock.patch('os.mkdir')
|
|
@mock.patch('os.path.exists')
|
|
@mock.patch('os.path.isdir')
|
|
def test_ensuredir(self, mock_isdir, mock_exists, mock_mkdir):
|
|
mock_exists.side_effect = [False, False, True]
|
|
mock_isdir.return_value = True
|
|
ensuredir('/path/foo/bar/')
|
|
self.assertEqual(mock_exists.call_count, 3)
|
|
self.assertEqual(mock_isdir.call_count, 1)
|
|
mock_mkdir.assert_has_calls([mock.call('/path/foo'),
|
|
mock.call('/path/foo/bar')])
|