nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-09-19 04:59:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

tor: Convert tests to pytest style

Browse Source 
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org>
Reviewed-by: Joseph Nuthalapati <njoseph@thoughtworks.com>
This commit is contained in:
Sunil Mohan Adapa 2019-05-01 16:09:12 -07:00 committed by Joseph Nuthalapati
parent a1064ec9ff
commit 2595e309ab
No known key found for this signature in database
GPG Key ID: 5398F00A2FA43C35
1 changed files with 26 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
plinth/modules/tor/tests/test_tor.py
View File

@ -14,32 +14,30 @@
# You should have received a copy of the GNU Affero General Public License # You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
# #
""" """
Tests for Tor module. Tests for Tor module.
""" """
import os import pytest
import unittest
from django.core.exceptions import ValidationError from django.core.exceptions import ValidationError
from plinth.modules.tor import forms, utils from plinth.modules.tor import forms, utils
euid = os.geteuid()
class TestTor:
"""Test cases for testing the Tor module."""
class TestTor(unittest.TestCase): @staticmethod
"""Test cases for testing the tor module.""" @pytest.mark.usefixtures('needs_root')
@unittest.skipUnless(euid == 0, 'Needs to be root') def test_is_apt_transport_tor_enabled():
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.
""" """
utils.is_apt_transport_tor_enabled() utils.is_apt_transport_tor_enabled()
@unittest.skipUnless(euid == 0, 'Needs to be root') @staticmethod
def test_get_status(self): @pytest.mark.usefixtures('needs_root', 'load_cfg')
def test_get_status():
"""Test that get_status does not raise any unhandled exceptions. """Test that get_status does not raise any unhandled exceptions.
This should work regardless of whether tor is installed, or This should work regardless of whether tor is installed, or
@ -48,9 +46,11 @@ class TestTor(unittest.TestCase):
utils.get_status() utils.get_status()
class TestTorForm(unittest.TestCase): class TestTorForm:
"""Test whether Tor configration form works.""" """Test whether Tor configration form works."""
def test_bridge_validator(self):
@staticmethod
def test_bridge_validator():
"""Test upstream bridges' form field validator.""" """Test upstream bridges' form field validator."""
validator = forms.bridges_validator validator = forms.bridges_validator
@ -85,12 +85,18 @@ class TestTorForm(unittest.TestCase):
'\n') '\n')
# Invalid number for parts # Invalid number for parts
self.assertRaises(ValidationError, validator, ' ') with pytest.raises(ValidationError):
validator(' ')
# Invalid IP address/port # Invalid IP address/port
self.assertRaises(ValidationError, validator, '73.237.165.384:9001') with pytest.raises(ValidationError):
self.assertRaises(ValidationError, validator, '73.237.165.184:90001') validator('73.237.165.384:9001')
self.assertRaises(ValidationError, validator,
'[a2001:db8:85a3:8d3:1319:8a2e:370:7348]:443') with pytest.raises(ValidationError):
self.assertRaises(ValidationError, validator, validator('73.237.165.184:90001')
'[2001:db8:85a3:8d3:1319:8a2e:370:7348]:90443')
with pytest.raises(ValidationError):
validator('[a2001:db8:85a3:8d3:1319:8a2e:370:7348]:443')
with pytest.raises(ValidationError):
validator('[2001:db8:85a3:8d3:1319:8a2e:370:7348]:90443')