Added unit tests

atlanhq · Jun 4, 2024 · ace73b0 · ace73b0
1 parent 7f322f1
commit ace73b0
Show file tree

Hide file tree

Showing 8 changed files with 127 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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",
     [