def test_calculate_free_space(self): dev_info = mock.Mock() dev_info.f_bsize = 1048576 dev_info.f_bavail = 2 with patch.object(os, 'statvfs', return_value=dev_info) as st_mock: self.assertEqual(calculate_free_space('/tmp/dir/file'), 2) st_mock.assert_called_once_with('/tmp/dir')
def _check_free_space(self): """Check `self.dst_path` free space :raises: errors.NotEnoughFreeSpace """ free_space = calculate_free_space(self.dst_path) if free_space < self.required_free_space: raise errors.NotEnoughFreeSpace( u'Not enough free space, path - "{0}", ' 'free space - "{1}", ' 'required free space - "{2}"'.format(self.dst_path, free_space, self.required_free_space))