crabapples
/
websockify
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Patch redis module in tests 

Make sure the tests can be run even if redis isn't installed.
This commit is contained in:
Pierre Ossman 2024-02-08 09:43:55 +01:00
parent 0427f63f96
commit bccf1dd258
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_token_plugins.py
View File

@ -2,6 +2,7 @@
""" Unit tests for Token plugins""" """ Unit tests for Token plugins"""
import sys
import unittest import unittest
from unittest.mock import patch, mock_open, MagicMock from unittest.mock import patch, mock_open, MagicMock
from jwcrypto import jwt, jwk from jwcrypto import jwt, jwk
@ -178,6 +179,14 @@ class JWSTokenTestCase(unittest.TestCase):
self.assertEqual(result[1], "remote_port") self.assertEqual(result[1], "remote_port")
class TokenRedisTestCase(unittest.TestCase): class TokenRedisTestCase(unittest.TestCase):
def setUp(self):
try:
import redis
except ImportError:
patcher = patch.dict(sys.modules, {'redis': MagicMock()})
patcher.start()
self.addCleanup(patcher.stop)
@patch('redis.Redis') @patch('redis.Redis')
def test_empty(self, mock_redis): def test_empty(self, mock_redis):
plugin = TokenRedis('127.0.0.1:1234') plugin = TokenRedis('127.0.0.1:1234')