Skip tor tests unless root.

This commit is contained in:
James Valleroy 2015-12-01 16:53:44 -05:00 committed by Sunil Mohan Adapa
parent 66625c35a3
commit 90a1c334fe

View File

@ -19,19 +19,24 @@
Tests for tor module. Tests for tor module.
""" """
import os
import unittest import unittest
euid = os.geteuid()
if euid == 0:
from ..tor import is_apt_transport_tor_enabled, get_hs, get_status from ..tor import is_apt_transport_tor_enabled, get_hs, get_status
class TestTor(unittest.TestCase): class TestTor(unittest.TestCase):
"""Test cases for testing the tor module.""" """Test cases for testing the tor module."""
@unittest.skipUnless(euid == 0, 'Needs to be root')
def test_is_apt_transport_tor_enabled(self): def test_is_apt_transport_tor_enabled(self):
"""Test that is_apt_transport_tor_enabled does not raise any unhandled """Test that is_apt_transport_tor_enabled does not raise any unhandled
exceptions. exceptions.
""" """
is_apt_transport_tor_enabled() is_apt_transport_tor_enabled()
@unittest.skipUnless(euid == 0, 'Needs to be root')
def test_get_hs(self): def test_get_hs(self):
"""Test that get_hs does not raise any unhandled exceptions. """Test that get_hs does not raise any unhandled exceptions.
@ -40,6 +45,7 @@ class TestTor(unittest.TestCase):
""" """
get_hs() get_hs()
@unittest.skipUnless(euid == 0, 'Needs to be root')
def test_get_status(self): def test_get_status(self):
"""Test that get_status does not raise any unhandled exceptions. """Test that get_status does not raise any unhandled exceptions.