Hotfix 864

chaos_genius/utils/metadata_api_config.py

@@ -52,6 +52,14 @@
     }
 }
 
+TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY_THIRD_PARTY = {
+    "tables": True,
+    "views": False,
+    "materialized_views": False,
+    "supported_aggregations": ["mean", "sum", "count"],
+    "supports_multidim_dd": True,
+}
+
 SUPPORTED_DATASOURCES_FOR_MULTIDIM_DD = [
     datasource
     for datasource, conf in TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY.items()

chaos_genius/views/data_source_view.py

@@ -41,6 +41,7 @@
 from chaos_genius.utils.metadata_api_config import (
     SCHEMAS_AVAILABLE,
     TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY,
+    TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY_THIRD_PARTY,
 )
 
 blueprint = Blueprint("api_data_source", __name__)
@@ -469,6 +470,7 @@ def check_views_availability():
     schema_exist = False
     message = ""
     status = "failure"
+    supported_aggregations = []
 
     try:
         data = request.get_json()
@@ -486,12 +488,15 @@ def check_views_availability():
                 )
 
             datasource_name = getattr(ds_data, "connection_type")
+            datasource_capability = (
+                TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY_THIRD_PARTY
+                if ds_data.is_third_party
+                else TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY[datasource_name]
+            )
             schema_exist = SCHEMAS_AVAILABLE.get(datasource_name, False)
-            views = TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY[datasource_name]["views"]
-            supported_aggregations = TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY[datasource_name]["supported_aggregations"]
-            materialize_views = TABLE_VIEW_MATERIALIZED_VIEW_AVAILABILITY[
-                datasource_name
-            ]["materialized_views"]
+            views = datasource_capability["views"]
+            supported_aggregations = datasource_capability["supported_aggregations"]
+            materialize_views = datasource_capability["materialized_views"]
             status = "success"
     except Exception as err:
         message = "Error in fetching table info: {}".format(err)