From ad26078a8f860340a7be4418cf6776dd38cc1aea Mon Sep 17 00:00:00 2001
From: Eric Vergnaud <eric.vergnaud@databricks.com>
Date: Fri, 21 Jun 2024 13:59:05 +0200
Subject: [PATCH] whitelist dbldatagen

---
 .../labs/ucx/source_code/known.json           | 64 ++++++++++++++++++-
 1 file changed, 63 insertions(+), 1 deletion(-)

diff --git a/src/databricks/labs/ucx/source_code/known.json b/src/databricks/labs/ucx/source_code/known.json
index bf8a6e23eb..7e9ab0f8d3 100644
--- a/src/databricks/labs/ucx/source_code/known.json
+++ b/src/databricks/labs/ucx/source_code/known.json
@@ -297,6 +297,68 @@
   "databricks-sdk": {
     "databricks.sdk": []
   },
+  "dbldatagen": {
+    "dbldatagen": [],
+    "dbldatagen._version": [],
+    "dbldatagen.column_generation_spec": [],
+    "dbldatagen.column_spec_options": [],
+    "dbldatagen.constraints": [],
+    "dbldatagen.constraints.chained_relation": [],
+    "dbldatagen.constraints.constraint": [],
+    "dbldatagen.constraints.literal_range_constraint": [],
+    "dbldatagen.constraints.literal_relation_constraint": [],
+    "dbldatagen.constraints.negative_values": [],
+    "dbldatagen.constraints.positive_values": [],
+    "dbldatagen.constraints.ranged_values_constraint": [],
+    "dbldatagen.constraints.sql_expr": [],
+    "dbldatagen.constraints.unique_combinations": [],
+    "dbldatagen.data_analyzer": [],
+    "dbldatagen.data_generator": [
+      {
+        "code": "legacy-context-in-shared-clusters",
+        "message": "sparkContext is not supported on UC Shared Clusters. Rewrite it using spark"
+      }
+    ],
+    "dbldatagen.datagen_constants": [],
+    "dbldatagen.datarange": [],
+    "dbldatagen.datasets": [],
+    "dbldatagen.datasets.basic_geometries": [],
+    "dbldatagen.datasets.basic_process_historian": [],
+    "dbldatagen.datasets.basic_telematics": [],
+    "dbldatagen.datasets.basic_user": [],
+    "dbldatagen.datasets.benchmark_groupby": [],
+    "dbldatagen.datasets.dataset_provider": [],
+    "dbldatagen.datasets.multi_table_telephony_provider": [],
+    "dbldatagen.datasets_object": [],
+    "dbldatagen.daterange": [],
+    "dbldatagen.distributions": [],
+    "dbldatagen.distributions.beta": [],
+    "dbldatagen.distributions.data_distribution": [],
+    "dbldatagen.distributions.exponential_distribution": [],
+    "dbldatagen.distributions.gamma": [],
+    "dbldatagen.distributions.normal_distribution": [],
+    "dbldatagen.function_builder": [],
+    "dbldatagen.html_utils": [],
+    "dbldatagen.nrange": [],
+    "dbldatagen.schema_parser": [
+      {
+        "code": "table-migrate",
+        "message": "Can't migrate table_name argument in 'sparkSession.sql(f'create temporary table {table_name}{table_body}using csv')' because its value cannot be computed"
+      },
+      {
+        "code": "table-migrate",
+        "message": "Can't migrate table_name argument in 'sparkSession.sql(f'drop table {table_name}')' because its value cannot be computed"
+      },
+      {
+        "code": "table-migrate",
+        "message": "Can't migrate table_name argument in 'sparkSession.sql(f'select * from {table_name}')' because its value cannot be computed"
+      }
+    ],
+    "dbldatagen.spark_singleton": [],
+    "dbldatagen.text_generator_plugins": [],
+    "dbldatagen.text_generators": [],
+    "dbldatagen.utils": []
+  },
   "debugpy": {
     "debugpy": []
   },
@@ -2288,4 +2350,4 @@
     "zipp.compat.py310": [],
     "zipp.glob": []
   }
-}
+}
\ No newline at end of file