def test_update_server_info_duplicate_sensor_alerts(self):
"""
Tests adding of sensor alerts to the database via update_server_information duplicates sensor alerts.
"""
config_logging(logging.ERROR)
storage = self._init_database()
storage._sensor_alert_life_span = 2147483647 # max signed 32bit integer
system_data = self._create_system_data()
self._create_objects(storage, system_data)
sensor_list = system_data.get_sensors_list()
for i in range(len(sensor_list)):
target_sensor = sensor_list[i % len(sensor_list)]
sensor_alert = ManagerObjSensorAlert()
sensor_alert.sensorId = target_sensor.sensorId
sensor_alert.state = (i % 2)
sensor_alert.description = "Sensor Alert %d" % i
sensor_alert.timeReceived = int(time.time())
sensor_alert.alertLevels = list(target_sensor.alertLevels)
sensor_alert.hasOptionalData = False
sensor_alert.changeState = False
sensor_alert.hasLatestData = False
sensor_alert.dataType = SensorDataType.NONE
system_data.add_sensor_alert(sensor_alert)
storage.update_server_information(0, system_data.get_options_list(),
system_data.get_profiles_list(),
system_data.get_nodes_list(),
system_data.get_sensors_list(),
system_data.get_alerts_list(),
system_data.get_managers_list(),
system_data.get_alert_levels_list(),
system_data.get_sensor_alerts_list())
self.assertEqual(len(system_data.get_sensor_alerts_list()),
len(sensor_list))
# Add same list of sensor alerts again for duplication test.
storage.update_server_information(0, system_data.get_options_list(),
system_data.get_profiles_list(),
system_data.get_nodes_list(),
system_data.get_sensors_list(),
system_data.get_alerts_list(),
system_data.get_managers_list(),
system_data.get_alert_levels_list(),
system_data.get_sensor_alerts_list())
storage._open_connection()
db_sensor_alerts = storage._get_sensor_alerts()
storage._close_connection()
compare_sensor_alerts_content(self,
system_data.get_sensor_alerts_list(),
db_sensor_alerts)
def test_add_sensor_alert(self):
"""
Tests adding of sensor alerts to the database.
"""
config_logging(logging.ERROR)
storage = self._init_database()
system_data = self._create_system_data()
self._create_objects(storage, system_data)
sensor_list = system_data.get_sensors_list()
for i in range(len(sensor_list)):
target_sensor = sensor_list[i % len(sensor_list)]
sensor_alert = ManagerObjSensorAlert()
sensor_alert.sensorId = target_sensor.sensorId
sensor_alert.state = (i % 2)
sensor_alert.description = "Sensor Alert %d" % i
sensor_alert.timeReceived = int(time.time())
sensor_alert.alertLevels = list(target_sensor.alertLevels)
sensor_alert.hasOptionalData = False
sensor_alert.changeState = False
sensor_alert.hasLatestData = False
sensor_alert.dataType = SensorDataType.NONE
system_data.add_sensor_alert(sensor_alert)
storage._open_connection()
for sensor_alert in system_data.get_sensor_alerts_list():
storage._add_sensor_alert(sensor_alert)
storage._conn.commit()
storage._close_connection()
storage._open_connection()
db_sensor_alerts = storage._get_sensor_alerts()
storage._close_connection()
compare_sensor_alerts_content(self,
system_data.get_sensor_alerts_list(),
db_sensor_alerts)