sharepoint task fail tests

dyvenia · Nov 24, 2023 · cf31f81 · cf31f81
1 parent 1a42c47
commit cf31f81
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/viadot/flows/sharepoint_to_adls.py b/viadot/flows/sharepoint_to_adls.py
@@ -190,14 +190,19 @@ def slugify(name):
         return name.replace(" ", "_").lower()
 
 
+class NoDataReturnedError(Exception):
+    def __init__(self, message):
+        self.message = message
+
+
 @task(slug="check_df")
 def check_if_df_empty(df):
     if len(df.index) == 0:
         logger.info("No data in the response. Df empty")
 
         ## raise ENDRUN(state=Failed())
-        raise FAIL("DF IS EMPTY!")
-        # raise ValueError("DF IS EMPTY!")
+        # raise FAIL("DF IS EMPTY!")
+        # raise NoDataReturnedError("DF IS EMPTY!")
 
 
 class SharepointListToADLS(Flow):