def testDirExists(self, makedirs_mock):
result = object()
open_mock = mock.Mock()
open_mock.return_value = result
self.assertEqual(util.SafeOpen(self.path, 'r', open_=open_mock),
result)
open_mock.assert_called_with(self.path, 'r')
makedirs_mock.assert_called_once_with(self.dir, 0700)
def testOk(self, makedirs_mock, mknod_mock):
result = object()
open_mock = mock.Mock(return_value=result)
self.assertEqual(util.SafeOpen(self.path, 'r', open_=open_mock),
result)
open_mock.assert_called_with(self.path, 'r')
makedirs_mock.assert_called_once_with(self.dir, 0700)
mknod_mock.assert_called_once_with(self.path, 0600 | stat.S_IFREG)
def testDirExists(self):
self.mox.StubOutWithMock(os, 'makedirs')
os.makedirs(self.dir, 0700).AndRaise(OSError)
result = object()
open_ = self.mox.CreateMockAnything()
open_(self.path, 'r').AndReturn(result)
mox.Replay(open_)
self.mox.ReplayAll()
self.assertEqual(util.SafeOpen(self.path, 'r', open_=open_), result)
self.mox.VerifyAll()
mox.Verify(open_)
def testOk(self):
self.mox.StubOutWithMock(os, 'makedirs')
os.makedirs(self.dir, 0700)
self.mox.StubOutWithMock(os, 'mknod')
os.mknod(self.path, 0600 | stat.S_IFREG)
result = object()
open_ = self.mox.CreateMockAnything()
open_(self.path, 'r').AndReturn(result)
mox.Replay(open_)
self.mox.ReplayAll()
self.assertEqual(util.SafeOpen(self.path, 'r', open_=open_), result)
self.mox.VerifyAll()
mox.Verify(open_)