Add proper error when can't select data based on given condition

qahive · Mar 27, 2020 · 3bce11b · 3bce11b
1 parent 8ea9cb1
commit 3bce11b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ExcelDataDriver/__init__.py b/ExcelDataDriver/__init__.py
@@ -39,7 +39,7 @@
 from ExcelDataDriver.Config.CaptureScreenShotOption import CaptureScreenShotOption
 
 
-__version__ = '1.1.5'
+__version__ = '1.1.6'
 
 
 class ExcelDataDriver:
@@ -579,7 +579,10 @@ def select_reference_data_based_on_condition(self, alias_name, condition):
             sheet_name          string
             properties_list     dictionary: access to excel property with lower case and use _ instead of space
         """
-        self.reference_data[alias_name]['selected'] = next(data for data in self.reference_data[alias_name]['data'] if eval(condition))
+        try:
+            self.reference_data[alias_name]['selected'] = next(data for data in self.reference_data[alias_name]['data'] if eval(condition))
+        except Exception:
+            raise Exception("Can't select reference data based on condition "+str(condition))
 
     @keyword
     def get_selected_reference_data_property(self, alias_name, property_name):