Skip to content

Commit

Permalink
Renamed test attribute redis to pubsub
Browse files Browse the repository at this point in the history
  • Loading branch information
Amertz08 committed Jul 27, 2018
1 parent 75dced3 commit 28cae30
Showing 1 changed file with 15 additions and 14 deletions.
29 changes: 15 additions & 14 deletions test_fakeredis.py
Original file line number Diff line number Diff line change
Expand Up @@ -3984,7 +3984,6 @@ def test_searches_for_c_stdlib_and_raises_if_missing(self):


class TestFakeStrictRedisConnectionErrors(unittest.TestCase):

def create_redis(self):
return fakeredis.FakeStrictRedis(db=0, connected=False)

Expand Down Expand Up @@ -4404,33 +4403,35 @@ def test_hscan_iter(self):


class TestPubSubConnected(unittest.TestCase):
def create_redis(self):
return fakeredis.FakePubSub(connected=False)

def setUp(self):
self.redis = fakeredis.FakePubSub(connected=False)
self.pubsub = self.create_redis()

def tearDown(self):
del self.redis
del self.pubsub

def test_basic_subscript(self):
with self.assertRaises(redis.ConnectionError):
self.redis.subscribe('logs')
self.pubsub.subscribe('logs')

def test_subscript_conn_lost(self):
self.redis.connected = True
self.redis.subscribe('logs')
self.redis.connected = False
self.pubsub.connected = True
self.pubsub.subscribe('logs')
self.pubsub.connected = False
with self.assertRaises(redis.ConnectionError):
self.redis.get_message()
self.pubsub.get_message()

def test_put_listen(self):
self.redis.connected = True
count = self.redis.put('logs', 'mymessage', 'subscribe')
self.pubsub.connected = True
count = self.pubsub.put('logs', 'mymessage', 'subscribe')
self.assertEqual(count, 1, 'Message could should be 1')
self.redis.connected = False
self.pubsub.connected = False
with self.assertRaises(redis.ConnectionError):
self.redis.get_message()
self.redis.connected = True
msg = self.redis.get_message()
self.pubsub.get_message()
self.pubsub.connected = True
msg = self.pubsub.get_message()
check = {
'type': 'subscribe',
'pattern': None,
Expand Down

0 comments on commit 28cae30

Please sign in to comment.