class TestConnection(unittest.TestCase):
@patch('redisEventTracker.StrictRedis', mock_strict_redis_client)
def setUp(self):
self.et = EventTracker()
self.date = datetime.now().date()
def test_check_date(self):
self.assertEqual(self.et._redis.sismember('dates', self.date), 0, u'Key is exist, or element member of set')
self.assertEqual(self.et._redis.sadd('dates', self.date), 1, u'Key must be added here')
self.assertEqual(self.et._redis.sadd('dates', self.date), 0, u'Key must be already exist')
def test_add_event(self):
self.et.track_event('event')
self.assertEqual(self.et._redis.hget('event', self.date), '1')
def test_inc_event(self):
self.et.track_event('event')
self.et.track_event('event')
self.assertEqual(self.et._redis.hget('event', self.date), '2')
@patch('redisEventTracker.logger')
def test_logging(self, mocked_logger):
self.et._redis.sadd = mock.Mock(side_effect=RedisError('Boom!'))
self.et.track_event('event')
self.assertTrue(mocked_logger.exception.called)
def test_tracker_param(self):
self.assertRaises(TypeError, self.et.track_event)
def setUp(self):
self.et = EventTracker()
self.date = datetime.now().date()
