73 lines
3.0 KiB
Python
73 lines
3.0 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
""" Unit tests for Authentication plugins"""
|
|
|
|
from websockify.auth_plugins import BasicHTTPAuth, HtpasswdAuth, AuthenticationError
|
|
import unittest
|
|
import tempfile
|
|
|
|
|
|
class BasicHTTPAuthTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.plugin = BasicHTTPAuth('Aladdin:open sesame')
|
|
|
|
def test_no_auth(self):
|
|
headers = {}
|
|
self.assertRaises(AuthenticationError, self.plugin.authenticate, headers, 'localhost', '1234')
|
|
|
|
def test_invalid_password(self):
|
|
headers = {'Authorization': 'Basic QWxhZGRpbjpzZXNhbWUgc3RyZWV0'}
|
|
self.assertRaises(AuthenticationError, self.plugin.authenticate, headers, 'localhost', '1234')
|
|
|
|
def test_valid_password(self):
|
|
headers = {'Authorization': 'Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=='}
|
|
self.plugin.authenticate(headers, 'localhost', '1234')
|
|
|
|
def test_garbage_auth(self):
|
|
headers = {'Authorization': 'Basic xxxxxxxxxxxxxxxxxxxxxxxxxxxx'}
|
|
self.assertRaises(AuthenticationError, self.plugin.authenticate, headers, 'localhost', '1234')
|
|
|
|
try:
|
|
import passlib
|
|
PASSLIB_AVAILABLE = True
|
|
except ImportError:
|
|
PASSLIB_AVAILABLE = False
|
|
|
|
@unittest.skipUnless(PASSLIB_AVAILABLE, "passlib package is not available")
|
|
class HtpasswdAuthTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self._temporary_htpasswd_file = tempfile.NamedTemporaryFile(delete=False)
|
|
|
|
#file generated with `htpasswd -c5i test_auth_plugins.htpasswd Genie <<<"""let's make some Magic!"""; htpasswd -Bi test_auth_plugins.htpasswd Aladdin <<<"""open sesame"""`
|
|
file_content = 'Genie:$6$5EsSBArrdAYDSe.j$v9mqxcSfPQgrM7btHx5wysZ28a1gei62rH75f8nYxwzPT80gbaL4qqxlkIBy.zSTnmG5VW2/RKFXQcGIgqAQq/\nAladdin:$2y$05$HK/O9w/55MSjM2FMefSIbeFKKANQbfR/hlYWk8RlDrR7Qyb5gnuzG'
|
|
|
|
self._temporary_htpasswd_file.write(file_content.encode('utf-8'))
|
|
self._temporary_htpasswd_file.close()
|
|
|
|
self.plugin = HtpasswdAuth(self._temporary_htpasswd_file.name)
|
|
|
|
def test_no_auth(self):
|
|
headers = {}
|
|
self.assertRaises(AuthenticationError, self.plugin.authenticate, headers, 'localhost', '1234')
|
|
|
|
def test_invalid_password(self):
|
|
headers = {'Authorization': 'Basic QWxhZGRpbjpzZXNhbWUgc3RyZWV0'}
|
|
self.assertRaises(AuthenticationError, self.plugin.authenticate, headers, 'localhost', '1234')
|
|
|
|
def test_valid_password(self):
|
|
headers = {'Authorization': 'Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=='}
|
|
self.plugin.authenticate(headers, 'localhost', '1234')
|
|
headers = {'Authorization': 'Basic R2VuaWU6bGV0J3MgbWFrZSBzb21lIE1hZ2ljIQ=='}
|
|
self.plugin.authenticate(headers, 'localhost', '1234')
|
|
|
|
def test_garbage_auth(self):
|
|
headers = {'Authorization': 'Basic xxxxxxxxxxxxxxxxxxxxxxxxxxxx'}
|
|
self.assertRaises(AuthenticationError, self.plugin.authenticate, headers, 'localhost', '1234')
|
|
|
|
def tearDown(self):
|
|
import os
|
|
os.remove(self._temporary_htpasswd_file.name)
|
|
|