def test_connection_success(self): getConfig(config_file_path, DEBUG, reload=True) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(True, redisConn.connect()) self.assertEqual(True, redisConn.testConnection()) self.assertEqual(True, redisConn.close())
class UtilsRedis(Utils): def __init__(self, config_file_path): self.redisConnector = RedisDBConnectorBASIC(config_file_path) self.redisConnector.connect() def deleteElements(self, name): if self.redisConnector.testConnection(): self.redisConnector.conn.delete(name) return True else: return False def countElements(self, name): if self.redisConnector.testConnection(): return self.redisConnector.conn.zcard(name) else: return False def getRedisChannelListener(self, channelName): if self.redisConnector.testConnection(): pubsub = self.redisConnector.conn.pubsub() pubsub.subscribe(channelName) return pubsub else: return False
def test_connection_with_wrong_password(self): getConfig(config_file_path, DEBUG, reload=True) getConfig(config_file_path, DEBUG).set('Redis', 'password', 'asdasd') redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(False, redisConn.connect()) self.assertEqual(False, redisConn.testConnection()) self.assertEqual(False, redisConn.close())
def test_modelname_does_not_exist(self): getConfig(config_file_path, DEBUG, reload=True) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(True, redisConn.connect()) self.assertEqual( False, redisConn.insertData('idontexist', { 'a': 4, 'b': 2, 'c': 3 })) self.assertEqual(True, redisConn.close())
def test_negative_batchsize(self): getConfig(config_file_path, DEBUG, reload=True) getConfig(config_file_path, DEBUG).set('General', 'batchsize', -1) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(True, redisConn.connect()) self.assertEqual( False, redisConn.insertData(tableName, { 'a': 4, 'b': 2, 'c': 3 })) self.assertEqual(True, redisConn.close())
def test_streaming_insert(self): self.assertEqual(True, utils.deleteElements(tableName)) getConfig(config_file_path, DEBUG, reload=True) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(True, redisConn.connect()) self.assertEqual( True, redisConn.insertData(tableName, { 'a': 4, 'b': 2, 'c': 3 })) self.assertEqual( True, redisConn.insertData(tableName, { 'a': 2, 'b': 2, 'c': 3 })) self.assertEqual( True, redisConn.insertData(tableName, { 'a': 9, 'b': 2, 'c': 3 })) self.assertEqual(True, redisConn.close()) self.assertEqual(3, utils.countElements(tableName))
def test_batch_insert_closing_connection_before_finish(self): self.assertEqual(True, utils.deleteElements(tableName)) getConfig(config_file_path, DEBUG, reload=True) getConfig(config_file_path, DEBUG).set('General', 'batchsize', 2) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(True, redisConn.connect()) self.assertEqual( True, redisConn.insertData(tableName, { 'a': 4, 'b': 2, 'c': 3 })) self.assertEqual(0, utils.countElements(tableName)) self.assertEqual(True, redisConn.close()) self.assertEqual(1, utils.countElements(tableName))
def test_insert_duplicate_data(self): getConfig(config_file_path, DEBUG, reload=True) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(True, redisConn.connect()) self.assertEqual( True, redisConn.insertData(tableName, { 'a': 5, 'b': 2, 'c': 3 })) self.assertEqual( True, redisConn.insertData(tableName, { 'a': 5, 'b': 2, 'c': 3 })) self.assertEqual(True, redisConn.close())
def __init__(self, config_file_path): self.redisConnector = RedisDBConnectorBASIC(config_file_path) self.redisConnector.connect()
def test_close_not_opened_connection(self): getConfig(config_file_path, DEBUG, reload=True) redisConn = RedisDBConnectorBASIC(config_file_path) self.assertEqual(False, redisConn.close())