11from unittest .mock import Mock
22
3- from databricks .sdk .service .compute import AutoScale , ClusterDetails
3+ from databricks .sdk .service .compute import AutoScale , ClusterDetails , ClusterSource
44from databricks .sdk .service .jobs import BaseJob , JobSettings , NotebookTask , Task
55
66from databricks .labs .ucx .assessment .crawlers import ClustersCrawler , JobsCrawler
@@ -82,6 +82,7 @@ def test_job_assessment():
8282 spark_env_vars = None ,
8383 spark_version = "13.3.x-cpu-ml-scala2.12" ,
8484 cluster_id = "0807-225846-motto493" ,
85+ cluster_source = ClusterSource .UI ,
8586 ),
8687 ClusterDetails (
8788 autoscale = AutoScale (min_workers = 1 , max_workers = 6 ),
@@ -90,6 +91,16 @@ def test_job_assessment():
9091 spark_env_vars = None ,
9192 spark_version = "9.3.x-cpu-ml-scala2.12" ,
9293 cluster_id = "0810-225833-atlanta69" ,
94+ cluster_source = ClusterSource .UI ,
95+ ),
96+ ClusterDetails (
97+ autoscale = AutoScale (min_workers = 1 , max_workers = 6 ),
98+ spark_conf = {"spark.databricks.delta.preview.enabled" : "true" },
99+ spark_context_id = 5134472582179566666 ,
100+ spark_env_vars = None ,
101+ spark_version = "13.3.x-cpu-ml-scala2.12" ,
102+ cluster_id = "0810-229933-chicago12" ,
103+ cluster_source = ClusterSource .JOB ,
93104 ),
94105 ]
95106 result_set = JobsCrawler (Mock (), MockBackend (), "ucx" )._assess_jobs (
0 commit comments