def setUp(self): Recommendation.init_db() Recommendation.remove_all()
def test_passing_connection(self): """ Pass in the Redis connection """ # Recommendation.init_db(Redis(host='127.0.0.1', port=6379)) Recommendation.init_db() self.assertIsNotNone(Recommendation.redis)
def test_vcap_services(self): """ Test if VCAP_SERVICES works """ Recommendation.init_db() self.assertIsNotNone(Recommendation.redis)
def init_db(redis=None): """ Initlaize the model """ Recommendation.init_db(redis)