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())