import os
try:
from cStringIO import StringIO
-except ImportError:
- from io import StringIO
+except ImportError: # pragma: no cover
+ from io import StringIO # pragma: no cover
from textwrap import dedent
import pytest
from ceph_volume import configuration, exceptions
def func():
return True
monkeypatch.setattr(decorators.os, 'getuid', lambda: 0)
- assert decorators.needs_root(func)()() is True
+ assert decorators.needs_root(func)() is True
def test_is_not_root(self, monkeypatch):
def func():
- return True
+ return True # pragma: no cover
monkeypatch.setattr(decorators.os, 'getuid', lambda: 20)
with pytest.raises(exceptions.SuperUserError) as error:
decorators.needs_root(func)()