diff --git a/tests/unit/model/a_d_l_s_container_test.py b/tests/unit/model/a_d_l_s_container_test.py index 680c2248..58883773 100644 --- a/tests/unit/model/a_d_l_s_container_test.py +++ b/tests/unit/model/a_d_l_s_container_test.py @@ -41,6 +41,20 @@ def test_create(): assert sut.qualified_name == f"{ADLS_ACCOUNT_QUALIFIED_NAME}/{ADLS_CONTAINER_NAME}" +def test_overload_creator(): + sut = ADLSContainer.creator( + name=ADLS_CONTAINER_NAME, + adls_account_qualified_name=ADLS_ACCOUNT_QUALIFIED_NAME, + connection_qualified_name=ADLS_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == ADLS_CONTAINER_NAME + assert sut.adls_account_qualified_name == ADLS_ACCOUNT_QUALIFIED_NAME + assert sut.connection_qualified_name == ADLS_CONNECTION_QUALIFIED_NAME + assert sut.connector_name == ADLS_CONNECTOR_TYPE + assert sut.qualified_name == f"{ADLS_ACCOUNT_QUALIFIED_NAME}/{ADLS_CONTAINER_NAME}" + + @pytest.mark.parametrize( "qualified_name, name, message", [ diff --git a/tests/unit/model/a_d_l_s_object_test.py b/tests/unit/model/a_d_l_s_object_test.py index f08b78ea..9b6b0919 100644 --- a/tests/unit/model/a_d_l_s_object_test.py +++ b/tests/unit/model/a_d_l_s_object_test.py @@ -48,6 +48,26 @@ def test_create(): ) +def test_overload_creator(): + sut = ADLSObject.creator( + name=ADLS_OBJECT_NAME, + adls_container_qualified_name=ADLS_CONTAINER_QUALIFIED_NAME, + adls_account_qualified_name=get_parent_qualified_name( + ADLS_CONTAINER_QUALIFIED_NAME + ), + connection_qualified_name=ADLS_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == ADLS_OBJECT_NAME + assert sut.adls_container_qualified_name == ADLS_CONTAINER_QUALIFIED_NAME + assert sut.qualified_name == f"{ADLS_CONTAINER_QUALIFIED_NAME}/{ADLS_OBJECT_NAME}" + assert sut.connection_qualified_name == ADLS_CONNECTION_QUALIFIED_NAME + assert sut.connector_name == ADLS_CONNECTOR_TYPE + assert sut.adls_account_qualified_name == get_parent_qualified_name( + ADLS_CONTAINER_QUALIFIED_NAME + ) + + # Test cases for creating ADLSObject for modification @pytest.mark.parametrize( "qualified_name, name, message", diff --git a/tests/unit/model/a_p_i_path_test.py b/tests/unit/model/a_p_i_path_test.py index 73f8ac93..20d0b602 100644 --- a/tests/unit/model/a_p_i_path_test.py +++ b/tests/unit/model/a_p_i_path_test.py @@ -44,6 +44,21 @@ def test_create(): assert sut.api_path_raw_u_r_i == API_PATH_RAW_URI +def test_overload_creator(): + sut = APIPath.creator( + path_raw_uri=API_PATH_RAW_URI, + spec_qualified_name=API_SPEC_QUALIFIED_NAME, + connection_qualified_name=API_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == API_PATH_NAME + assert sut.connection_qualified_name == API_CONNECTION_QUALIFIED_NAME + assert sut.qualified_name == API_PATH_QUALIFIED_NAME + assert sut.connector_name == API_CONNECTOR_TYPE + assert sut.api_spec_qualified_name == API_SPEC_QUALIFIED_NAME + assert sut.api_path_raw_u_r_i == API_PATH_RAW_URI + + @pytest.mark.parametrize( "qualified_name, name, message", [ diff --git a/tests/unit/model/airflow_task_test.py b/tests/unit/model/airflow_task_test.py index 7cc8caf3..cee95f79 100644 --- a/tests/unit/model/airflow_task_test.py +++ b/tests/unit/model/airflow_task_test.py @@ -39,6 +39,20 @@ def test_creator(): assert task.qualified_name == f"{AIRFLOW_DAG_QUALIFIED_NAME}/{AIRFLOW_TASK_NAME}" +def test_overload_creator(): + task = AirflowTask.creator( + name=AIRFLOW_TASK_NAME, + airflow_dag_qualified_name=AIRFLOW_DAG_QUALIFIED_NAME, + connection_qualified_name=AIRFLOW_CONNECTION_QUALIFIED_NAME, + ) + + assert task.name == AIRFLOW_TASK_NAME + assert task.connector_name == AtlanConnectorType.AIRFLOW + assert task.airflow_dag_qualified_name == AIRFLOW_DAG_QUALIFIED_NAME + assert task.connection_qualified_name == AIRFLOW_CONNECTION_QUALIFIED_NAME + assert task.qualified_name == f"{AIRFLOW_DAG_QUALIFIED_NAME}/{AIRFLOW_TASK_NAME}" + + @pytest.mark.parametrize( "qualified_name, name, message", [ diff --git a/tests/unit/model/gcs_object_test.py b/tests/unit/model/gcs_object_test.py index 6dcff63d..28547649 100644 --- a/tests/unit/model/gcs_object_test.py +++ b/tests/unit/model/gcs_object_test.py @@ -3,6 +3,7 @@ from pyatlan.model.assets import GCSObject from tests.unit.model.constants import ( GCS_BUCKET_QUALIFIED_NAME, + GCS_CONNECTION_QUALIFIED_NAME, GCS_OBJECT_NAME, GCS_OBJECT_QUALIFIED_NAME, ) @@ -30,6 +31,20 @@ def test_create(): assert sut.name == GCS_OBJECT_NAME assert sut.gcs_bucket_qualified_name == GCS_BUCKET_QUALIFIED_NAME assert sut.qualified_name == GCS_OBJECT_QUALIFIED_NAME + assert sut.connection_qualified_name == GCS_CONNECTION_QUALIFIED_NAME + + +def test_overload_creator(): + sut = GCSObject.creator( + name=GCS_OBJECT_NAME, + gcs_bucket_qualified_name=GCS_BUCKET_QUALIFIED_NAME, + connection_qualified_name=GCS_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == GCS_OBJECT_NAME + assert sut.gcs_bucket_qualified_name == GCS_BUCKET_QUALIFIED_NAME + assert sut.qualified_name == GCS_OBJECT_QUALIFIED_NAME + assert sut.connection_qualified_name == GCS_CONNECTION_QUALIFIED_NAME @pytest.mark.parametrize( diff --git a/tests/unit/model/preset_chart_test.py b/tests/unit/model/preset_chart_test.py index d5fa9b2a..f4a9c5ea 100644 --- a/tests/unit/model/preset_chart_test.py +++ b/tests/unit/model/preset_chart_test.py @@ -41,6 +41,22 @@ def test_create(): assert sut.connector_name == PRESET_CONNECTOR_TYPE +def test_overload_creator(): + sut = PresetChart.creator( + name=PRESET_CHART_NAME, + preset_dashboard_qualified_name=PRESET_DASHBOARD_QUALIFIED_NAME, + connection_qualified_name=PRESET_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == PRESET_CHART_NAME + assert sut.preset_dashboard_qualified_name == PRESET_DASHBOARD_QUALIFIED_NAME + assert sut.connection_qualified_name == PRESET_CONNECTION_QUALIFIED_NAME + assert ( + sut.qualified_name == f"{PRESET_DASHBOARD_QUALIFIED_NAME}/{PRESET_CHART_NAME}" + ) + assert sut.connector_name == PRESET_CONNECTOR_TYPE + + @pytest.mark.parametrize( "qualified_name, name, message", [ diff --git a/tests/unit/model/preset_dashboard_test.py b/tests/unit/model/preset_dashboard_test.py index d8f60c46..e13769c7 100644 --- a/tests/unit/model/preset_dashboard_test.py +++ b/tests/unit/model/preset_dashboard_test.py @@ -42,6 +42,23 @@ def test_create(): assert sut.connector_name == PRESET_CONNECTOR_TYPE +def test_overload_creator(): + sut = PresetDashboard.creator( + name=PRESET_DASHBOARD_NAME, + preset_workspace_qualified_name=PRESET_WORKSPACE_QUALIFIED_NAME, + connection_qualified_name=PRESET_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == PRESET_DASHBOARD_NAME + assert sut.preset_workspace_qualified_name == PRESET_WORKSPACE_QUALIFIED_NAME + assert sut.connection_qualified_name == PRESET_CONNECTION_QUALIFIED_NAME + assert ( + sut.qualified_name + == f"{PRESET_WORKSPACE_QUALIFIED_NAME}/{PRESET_DASHBOARD_NAME}" + ) + assert sut.connector_name == PRESET_CONNECTOR_TYPE + + @pytest.mark.parametrize( "qualified_name, name, message", [ diff --git a/tests/unit/model/preset_dataset_test.py b/tests/unit/model/preset_dataset_test.py index ae86fa62..98301ef1 100644 --- a/tests/unit/model/preset_dataset_test.py +++ b/tests/unit/model/preset_dataset_test.py @@ -41,6 +41,22 @@ def test_create(): assert sut.connector_name == PRESET_CONNECTOR_TYPE +def test_creator(): + sut = PresetDataset.creator( + name=PRESET_DATASET_NAME, + preset_dashboard_qualified_name=PRESET_DASHBOARD_QUALIFIED_NAME, + connection_qualified_name=PRESET_CONNECTION_QUALIFIED_NAME, + ) + + assert sut.name == PRESET_DATASET_NAME + assert sut.preset_dashboard_qualified_name == PRESET_DASHBOARD_QUALIFIED_NAME + assert sut.connection_qualified_name == PRESET_CONNECTION_QUALIFIED_NAME + assert ( + sut.qualified_name == f"{PRESET_DASHBOARD_QUALIFIED_NAME}/{PRESET_DATASET_NAME}" + ) + assert sut.connector_name == PRESET_CONNECTOR_TYPE + + @pytest.mark.parametrize( "qualified_name, name, message", [