openmetadata-ingestion 1.6.0.0rc1__py3-none-any.whl → 1.6.0.0rc2__py3-none-any.whl

metadata/ingestion/source/dashboard/{mstr → microstrategy}/metadata.py

@@ -8,7 +8,7 @@
 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
-"""Mstr source module"""
+"""MicroStrategy source module"""
 import traceback
 from typing import Iterable, List, Optional
 
@@ -16,8 +16,8 @@ from metadata.generated.schema.api.data.createChart import CreateChartRequest
 from metadata.generated.schema.api.data.createDashboard import CreateDashboardRequest
 from metadata.generated.schema.api.lineage.addLineage import AddLineageRequest
 from metadata.generated.schema.entity.data.chart import Chart
-from metadata.generated.schema.entity.services.connections.dashboard.mstrConnection import (
-    MstrConnection,
+from metadata.generated.schema.entity.services.connections.dashboard.microStrategyConnection import (
+    MicroStrategyConnection,
 )
 from metadata.generated.schema.entity.services.ingestionPipelines.status import (
     StackTraceError,
@@ -34,7 +34,7 @@ from metadata.ingestion.api.models import Either
 from metadata.ingestion.api.steps import InvalidSourceException
 from metadata.ingestion.ometa.ometa_api import OpenMetadata
 from metadata.ingestion.source.dashboard.dashboard_service import DashboardServiceSource
-from metadata.ingestion.source.dashboard.mstr.models import (
+from metadata.ingestion.source.dashboard.microstrategy.models import (
     MstrDashboard,
     MstrDashboardDetails,
     MstrPage,
@@ -47,9 +47,9 @@ from metadata.utils.logger import ingestion_logger
 logger = ingestion_logger()
 
 
-class MstrSource(DashboardServiceSource):
+class MicrostrategySource(DashboardServiceSource):
     """
-    MSTR Source Class
+    Microstrategy Source Class
     """
 
     @classmethod
@@ -60,10 +60,10 @@ class MstrSource(DashboardServiceSource):
         pipeline_name: Optional[str] = None,
     ):
         config = WorkflowSource.model_validate(config_dict)
-        connection: MstrConnection = config.serviceConnection.root.config
-        if not isinstance(connection, MstrConnection):
+        connection: MicroStrategyConnection = config.serviceConnection.root.config
+        if not isinstance(connection, MicroStrategyConnection):
             raise InvalidSourceException(
-                f"Expected MstrConnection, but got {connection}"
+                f"Expected MicroStrategyConnection, but got {connection}"
             )
         return cls(config, metadata)
 
@@ -75,14 +75,18 @@ class MstrSource(DashboardServiceSource):
 
         if self.client.is_project_name():
             project = self.client.get_project_by_name()
-            dashboards.extend(self.client.get_dashboards_list(project.id, project.name))
-
-        if not self.client.is_project_name():
-            for project in self.client.get_projects_list():
+            if project:
                 dashboards.extend(
                     self.client.get_dashboards_list(project.id, project.name)
                 )
 
+        if not self.client.is_project_name():
+            for project in self.client.get_projects_list():
+                if project:
+                    dashboards.extend(
+                        self.client.get_dashboards_list(project.id, project.name)
+                    )
+
         return dashboards
 
     def get_dashboard_name(self, dashboard: MstrDashboard) -> str:
@@ -91,6 +95,18 @@ class MstrSource(DashboardServiceSource):
         """
         return dashboard.name
 
+    def get_project_name(self, dashboard_details: MstrDashboard) -> Optional[str]:
+        """
+        Get dashboard project name
+        """
+        try:
+            return dashboard_details.projectName
+        except Exception as exc:
+            logger.debug(
+                f"Cannot get project name from dashboard [{dashboard_details.name}] - [{exc}]"
+            )
+        return None
+
     def get_dashboard_details(self, dashboard: MstrDashboard) -> MstrDashboardDetails:
         """
         Get Dashboard Details
@@ -106,40 +122,41 @@ class MstrSource(DashboardServiceSource):
         """
         Method to Get Dashboard Entity
         """
-        try:
-            dashboard_url = (
-                f"{clean_uri(self.service_connection.hostPort)}/MicroStrategyLibrary/app/"
-                f"{dashboard_details.projectId}/{dashboard_details.id}"
-            )
-            dashboard_request = CreateDashboardRequest(
-                name=EntityName(dashboard_details.id),
-                displayName=dashboard_details.name,
-                sourceUrl=SourceUrl(dashboard_url),
-                project=dashboard_details.projectName,
-                charts=[
-                    FullyQualifiedEntityName(
-                        fqn.build(
-                            self.metadata,
-                            entity_type=Chart,
-                            service_name=self.context.get().dashboard_service,
-                            chart_name=chart,
+        if dashboard_details:
+            try:
+                dashboard_url = (
+                    f"{clean_uri(str(self.service_connection.hostPort))}/MicroStrategyLibrary/app/"
+                    f"{dashboard_details.projectId}/{dashboard_details.id}"
+                )
+                dashboard_request = CreateDashboardRequest(
+                    name=EntityName(dashboard_details.id),
+                    displayName=dashboard_details.name,
+                    sourceUrl=SourceUrl(dashboard_url),
+                    project=dashboard_details.projectName,
+                    charts=[
+                        FullyQualifiedEntityName(
+                            fqn.build(
+                                self.metadata,
+                                entity_type=Chart,
+                                service_name=self.context.get().dashboard_service,
+                                chart_name=chart,
+                            )
                         )
+                        for chart in self.context.get().charts or []
+                    ],
+                    service=self.context.get().dashboard_service,
+                    owners=self.get_owner_ref(dashboard_details=dashboard_details),
+                )
+                yield Either(right=dashboard_request)
+                self.register_record(dashboard_request=dashboard_request)
+            except Exception as exc:
+                yield Either(
+                    left=StackTraceError(
+                        name=dashboard_details.id,
+                        error=f"Error yielding dashboard for {dashboard_details}: {exc}",
+                        stackTrace=traceback.format_exc(),
                     )
-                    for chart in self.context.get().charts or []
-                ],
-                service=self.context.get().dashboard_service,
-                owners=self.get_owner_ref(dashboard_details=dashboard_details),
-            )
-            yield Either(right=dashboard_request)
-            self.register_record(dashboard_request=dashboard_request)
-        except Exception as exc:
-            yield Either(
-                left=StackTraceError(
-                    name=dashboard_details.id,
-                    error=f"Error yielding dashboard for {dashboard_details}: {exc}",
-                    stackTrace=traceback.format_exc(),
                 )
-            )
 
     def yield_dashboard_lineage_details(
         self, dashboard_details: MstrDashboardDetails, db_service_name: str
@@ -156,14 +173,15 @@ class MstrSource(DashboardServiceSource):
         Returns:
             Iterable[CreateChartRequest]
         """
-        try:
-            for chapter in dashboard_details.chapters:
-                for page in chapter.pages:
-                    yield from self._yield_chart_from_visualization(page)
+        if dashboard_details:
+            try:
+                for chapter in dashboard_details.chapters:
+                    for page in chapter.pages:
+                        yield from self._yield_chart_from_visualization(page)
 
-        except Exception as exc:
-            logger.debug(traceback.format_exc())
-            logger.warning(f"Error creating dashboard: {exc}")
+            except Exception as exc:
+                logger.debug(traceback.format_exc())
+                logger.warning(f"Error creating dashboard: {exc}")
 
     def _yield_chart_from_visualization(
         self, page: MstrPage
@@ -192,3 +210,8 @@ class MstrSource(DashboardServiceSource):
                         stackTrace=traceback.format_exc(),
                     )
                 )
+
+    def close(self):
+        # close the api session
+        self.client.close_api_session()
+        self.metadata.close()

metadata/ingestion/source/dashboard/{mstr → microstrategy}/models.py

@@ -9,10 +9,10 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 """
-MSTR Models
+MicroStrategy Models
 """
 from datetime import datetime
-from typing import List, Optional
+from typing import Any, List, Optional
 
 from pydantic import BaseModel
 
@@ -69,7 +69,8 @@ class MstrSearchResult(BaseModel):
 
 
 class MstrSearchResultList(BaseModel):
-    results: Optional[List[MstrSearchResult]]
+    totalItems: Optional[int] = 0
+    result: Optional[List[MstrSearchResult]] = None
 
 
 class MstrDashboard(BaseModel):
@@ -142,3 +143,8 @@ class MstrDashboardDetails(BaseModel):
     projectName: str
     currentChapter: str
     chapters: List[MstrChapter]
+
+
+class AuthHeaderCookie(BaseModel):
+    auth_header: dict
+    auth_cookies: Any

metadata/ingestion/source/dashboard/microstrategy/service_spec.py

@@ -0,0 +1,6 @@
+from metadata.ingestion.source.dashboard.microstrategy.metadata import (
+    MicrostrategySource,
+)
+from metadata.utils.service_spec.default import DefaultDatabaseSpec
+
+ServiceSpec = DefaultDatabaseSpec(metadata_source_class=MicrostrategySource)

metadata/ingestion/source/dashboard/powerbi/client.py

@@ -64,11 +64,10 @@ class PowerBiApiClient:
             client_credential=self.config.clientSecret.get_secret_value(),
             authority=self.config.authorityURI + self.config.tenantId,
         )
-        self.auth_token = self.get_auth_token()
         client_config = ClientConfig(
             base_url="https://api.powerbi.com",
             api_version="v1.0",
-            auth_token=lambda: self.auth_token,
+            auth_token=self.get_auth_token,
             auth_header="Authorization",
             allow_redirects=True,
             retry_codes=[429],

metadata/ingestion/source/dashboard/superset/db_source.py

@@ -201,7 +201,7 @@ class SupersetDBSource(SupersetSourceMixin):
             except Exception as exc:
                 yield Either(
                     left=StackTraceError(
-                        name=chart_json.id,
+                        name=str(chart_json.id),
                         error=f"Error yielding Chart [{chart_json.id} - {chart_json.slice_name}]: {exc}",
                         stackTrace=traceback.format_exc(),
                     )

metadata/ingestion/source/database/athena/metadata.py

@@ -157,6 +157,14 @@ class AthenaSource(ExternalTableLineageMixin, CommonDbSourceService):
             return True, partition_details
         return False, None
 
+    def get_location_path(self, table_name: str, schema_name: str) -> Optional[str]:
+        """
+        Method to fetch the location path of the table
+        """
+        return self.external_location_map.get(
+            (self.context.get().database, schema_name, table_name)
+        )
+
     def yield_tag(
         self, schema_name: str
     ) -> Iterable[Either[OMetaTagAndClassification]]:

metadata/ingestion/source/database/azuresql/service_spec.py

@@ -1,10 +1,12 @@
 from metadata.ingestion.source.database.azuresql.lineage import AzuresqlLineageSource
 from metadata.ingestion.source.database.azuresql.metadata import AzuresqlSource
 from metadata.ingestion.source.database.azuresql.usage import AzuresqlUsageSource
+from metadata.sampler.sqlalchemy.azuresql.sampler import AzureSQLSampler
 from metadata.utils.service_spec.default import DefaultDatabaseSpec
 
 ServiceSpec = DefaultDatabaseSpec(
     metadata_source_class=AzuresqlSource,
     lineage_source_class=AzuresqlLineageSource,
     usage_source_class=AzuresqlUsageSource,
+    sampler_class=AzureSQLSampler,
 )

metadata/ingestion/source/database/bigquery/profiler/profiler.py

@@ -22,7 +22,7 @@ class BigQueryProfiler(BigQueryProfilerInterface):
         **kwargs,
     ) -> List[SystemProfile]:
         return self.system_metrics_computer.get_system_metrics(
-            table=runner.table,
+            table=runner.dataset,
             usage_location=self.service_connection_config.usageLocation,
         )
 

metadata/ingestion/source/database/bigquery/service_spec.py

@@ -4,6 +4,7 @@ from metadata.ingestion.source.database.bigquery.profiler.profiler import (
     BigQueryProfiler,
 )
 from metadata.ingestion.source.database.bigquery.usage import BigqueryUsageSource
+from metadata.sampler.sqlalchemy.bigquery.sampler import BigQuerySampler
 from metadata.utils.service_spec.default import DefaultDatabaseSpec
 
 ServiceSpec = DefaultDatabaseSpec(
@@ -11,4 +12,5 @@ ServiceSpec = DefaultDatabaseSpec(
     lineage_source_class=BigqueryLineageSource,
     usage_source_class=BigqueryUsageSource,
     profiler_class=BigQueryProfiler,
+    sampler_class=BigQuerySampler,
 )

metadata/ingestion/source/database/common_nosql_source.py

@@ -52,6 +52,7 @@ from metadata.utils.constants import DEFAULT_DATABASE
 from metadata.utils.datalake.datalake_utils import DataFrameColumnParser
 from metadata.utils.filters import filter_by_schema, filter_by_table
 from metadata.utils.logger import ingestion_logger
+from metadata.utils.ssl_manager import check_ssl_and_init
 
 logger = ingestion_logger()
 
@@ -73,7 +74,13 @@ class CommonNoSQLSource(DatabaseServiceSource, ABC):
         )
         self.metadata = metadata
         self.service_connection = self.config.serviceConnection.root.config
+        self.ssl_manager = check_ssl_and_init(self.service_connection)
+        if self.ssl_manager:
+            self.service_connection = self.ssl_manager.setup_ssl(
+                self.service_connection
+            )
         self.connection_obj = get_connection(self.service_connection)
+
         self.test_connection()
 
     def prepare(self):

metadata/ingestion/source/database/databricks/metadata.py

@@ -131,7 +131,7 @@ def _get_table_columns(self, connection, table_name, schema, db_name):
         query = DATABRICKS_GET_TABLE_COMMENTS.format(
             database_name=db_name, schema_name=schema, table_name=table_name
         )
-        cursor = get_table_comment_result(
+        rows = get_table_comment_result(
             self,
             connection=connection,
             query=query,
@@ -140,8 +140,6 @@ def _get_table_columns(self, connection, table_name, schema, db_name):
             schema=schema,
         )
 
-        rows = cursor.fetchall()
-
     except exc.OperationalError as e:
         # Does the table exist?
         regex_fmt = r"TExecuteStatementResp.*SemanticException.*Table not found {}"

metadata/ingestion/source/database/databricks/service_spec.py

@@ -1,3 +1,6 @@
+from metadata.data_quality.interface.sqlalchemy.databricks.test_suite_interface import (
+    DatabricksTestSuiteInterface,
+)
 from metadata.ingestion.source.database.databricks.lineage import (
     DatabricksLineageSource,
 )
@@ -13,4 +16,5 @@ ServiceSpec = DefaultDatabaseSpec(
     lineage_source_class=DatabricksLineageSource,
     usage_source_class=DatabricksUsageSource,
     profiler_class=DatabricksProfilerInterface,
+    test_suite_class=DatabricksTestSuiteInterface,
 )

metadata/ingestion/source/database/datalake/clients/azure_blob.py

@@ -26,9 +26,6 @@ from metadata.utils.constants import DEFAULT_DATABASE
 
 
 class DatalakeAzureBlobClient(DatalakeBaseClient):
-    def __init__(self, client: BlobServiceClient):
-        self._client = client
-
     @classmethod
     def from_config(cls, config: AzureConfig) -> "DatalakeAzureBlobClient":
         try:

metadata/ingestion/source/database/datalake/clients/base.py

@@ -13,10 +13,19 @@
 Datalake Base Client
 """
 from abc import ABC, abstractmethod
-from typing import Callable, Iterable, Optional
+from typing import Any, Callable, Iterable, Optional
 
 
 class DatalakeBaseClient(ABC):
+    """Base DL client implementation"""
+
+    def __init__(self, client: Any, **kwargs):
+        self._client = client
+
+    @property
+    def client(self) -> Any:
+        return self._client
+
     @classmethod
     @abstractmethod
     def from_config(cls, config) -> "DatalakeBaseClient":

metadata/ingestion/source/database/datalake/clients/gcs.py

@@ -33,9 +33,11 @@ from metadata.utils.credentials import GOOGLE_CREDENTIALS, set_google_credential
 
 class DatalakeGcsClient(DatalakeBaseClient):
     def __init__(
-        self, client: storage.Client, temp_credentials_file_path_list: List[str]
+        self,
+        client: storage.Client,
+        temp_credentials_file_path_list: List[str],
     ):
-        self._client = client
+        super().__init__(client=client)
         self._temp_credentials_file_path_list = temp_credentials_file_path_list
 
     @property

metadata/ingestion/source/database/datalake/clients/s3.py

@@ -25,9 +25,6 @@ from metadata.utils.s3_utils import list_s3_objects
 
 
 class DatalakeS3Client(DatalakeBaseClient):
-    def __init__(self, client):
-        self._client = client
-
     @classmethod
     def from_config(cls, config: S3Config) -> "DatalakeS3Client":
         if not config.securityConfig:

metadata/ingestion/source/database/datalake/service_spec.py

@@ -1,9 +1,16 @@
+from metadata.data_quality.interface.pandas.pandas_test_suite_interface import (
+    PandasTestSuiteInterface,
+)
 from metadata.ingestion.source.database.datalake.metadata import DatalakeSource
 from metadata.profiler.interface.pandas.profiler_interface import (
     PandasProfilerInterface,
 )
+from metadata.sampler.pandas.sampler import DatalakeSampler
 from metadata.utils.service_spec.default import DefaultDatabaseSpec
 
 ServiceSpec = DefaultDatabaseSpec(
-    metadata_source_class=DatalakeSource, profiler_class=PandasProfilerInterface
+    metadata_source_class=DatalakeSource,
+    profiler_class=PandasProfilerInterface,
+    test_suite_class=PandasTestSuiteInterface,
+    sampler_class=DatalakeSampler,
 )

metadata/ingestion/source/database/dbt/constants.py

@@ -82,6 +82,7 @@ NONE_KEYWORDS_LIST = ["none", "null"]
 DBT_CATALOG_FILE_NAME = "catalog.json"
 DBT_MANIFEST_FILE_NAME = "manifest.json"
 DBT_RUN_RESULTS_FILE_NAME = "run_results"
+DBT_SOURCES_FILE_NAME = "sources.json"
 
 
 class SkipResourceTypeEnum(Enum):
@@ -91,6 +92,7 @@ class SkipResourceTypeEnum(Enum):
 
     ANALYSIS = "analysis"
     TEST = "test"
+    SOURCE = "source"
 
 
 class CompiledQueriesEnum(Enum):
@@ -127,6 +129,7 @@ class DbtTestFailureEnum(Enum):
 
     FAILURE = "failure"
     FAIL = "fail"
+    ERROR = "error"
 
 
 class DbtCommonEnum(Enum):
@@ -137,6 +140,7 @@ class DbtCommonEnum(Enum):
     OWNER = "owner"
     NODES = "nodes"
     SOURCES = "sources"
+    SOURCES_FILE = "sources_file"
     SOURCE = "source"
     RESOURCETYPE = "resource_type"
     MANIFEST_NODE = "manifest_node"

metadata/ingestion/source/database/dbt/dbt_config.py

@@ -43,6 +43,7 @@ from metadata.ingestion.source.database.dbt.constants import (
     DBT_CATALOG_FILE_NAME,
     DBT_MANIFEST_FILE_NAME,
     DBT_RUN_RESULTS_FILE_NAME,
+    DBT_SOURCES_FILE_NAME,
 )
 from metadata.ingestion.source.database.dbt.models import DbtFiles
 from metadata.readers.file.config_source_factory import get_reader
@@ -85,6 +86,7 @@ def _(config: DbtLocalConfig):
             config.dbtManifestFilePath,
             config.dbtCatalogFilePath,
             config.dbtRunResultsFilePath,
+            config.dbtSourcesFilePath,
         ]
         yield from download_dbt_files(
             blob_grouped_by_directory=blob_grouped_by_directory,
@@ -123,12 +125,22 @@ def _(config: DbtHttpConfig):
             dbt_catalog = requests.get(  # pylint: disable=missing-timeout
                 config.dbtCatalogHttpPath
             )
+
+        dbt_sources = None
+        if config.dbtSourcesHttpPath:
+            logger.debug(
+                f"Requesting [dbtSourcesHttpPath] to: {config.dbtSourcesHttpPath}"
+            )
+            dbt_sources = requests.get(  # pylint: disable=missing-timeout
+                config.dbtSourcesHttpPath
+            )
         if not dbt_manifest:
             raise DBTConfigException("Manifest file not found in file server")
         yield DbtFiles(
             dbt_catalog=dbt_catalog.json() if dbt_catalog else None,
             dbt_manifest=dbt_manifest.json(),
             dbt_run_results=[dbt_run_results.json()] if dbt_run_results else None,
+            dbt_sources=dbt_sources.json() if dbt_sources else None,
         )
     except DBTConfigException as exc:
         raise exc
@@ -164,7 +176,7 @@ def _(config: DbtCloudConfig):  # pylint: disable=too-many-locals
         logger.debug(
             "Requesting [dbt_catalog], [dbt_manifest] and [dbt_run_results] data"
         )
-        params_data = {"order_by": "-finished_at", "limit": "1", "status": "10"}
+        params_data = {"order_by": "-finished_at", "limit": "1"}
         if project_id:
             params_data["project_id"] = project_id
 
@@ -243,6 +255,7 @@ def get_blobs_grouped_by_dir(blobs: List[str]) -> Dict[str, List[str]]:
     return blob_grouped_by_directory
 
 
+# pylint: disable=too-many-locals, too-many-branches
 def download_dbt_files(
     blob_grouped_by_directory: Dict, config, client, bucket_name: Optional[str]
 ) -> Iterable[DbtFiles]:
@@ -255,6 +268,7 @@ def download_dbt_files(
     ) in blob_grouped_by_directory.items():
         dbt_catalog = None
         dbt_manifest = None
+        dbt_sources = None
         dbt_run_results = []
         kwargs = {}
         if bucket_name:
@@ -285,12 +299,16 @@ def download_dbt_files(
                             logger.warning(
                                 f"{DBT_RUN_RESULTS_FILE_NAME} not found in {key}: {exc}"
                             )
+                    if DBT_SOURCES_FILE_NAME == blob_file_name.lower():
+                        logger.debug(f"{DBT_SOURCES_FILE_NAME} found in {key}")
+                        dbt_sources = reader.read(path=blob, **kwargs)
             if not dbt_manifest:
                 raise DBTConfigException(f"Manifest file not found at: {key}")
             yield DbtFiles(
                 dbt_catalog=json.loads(dbt_catalog) if dbt_catalog else None,
                 dbt_manifest=json.loads(dbt_manifest),
                 dbt_run_results=dbt_run_results if dbt_run_results else None,
+                dbt_sources=json.loads(dbt_sources) if dbt_sources else None,
             )
         except DBTConfigException as exc:
             logger.warning(exc)

metadata/ingestion/source/database/dbt/dbt_service.py

@@ -15,7 +15,12 @@ DBT service Topology.
 from abc import ABC, abstractmethod
 from typing import Iterable, List
 
-from dbt_artifacts_parser.parser import parse_catalog, parse_manifest, parse_run_results
+from dbt_artifacts_parser.parser import (
+    parse_catalog,
+    parse_manifest,
+    parse_run_results,
+    parse_sources,
+)
 from pydantic import Field
 from typing_extensions import Annotated
 
@@ -209,11 +214,13 @@ class DbtServiceSource(TopologyRunnerMixin, Source, ABC):
             self.remove_run_result_non_required_keys(
                 run_results=self.context.get().dbt_file.dbt_run_results
             )
+
         dbt_objects = DbtObjects(
             dbt_catalog=parse_catalog(self.context.get().dbt_file.dbt_catalog)
             if self.context.get().dbt_file.dbt_catalog
             else None,
             dbt_manifest=parse_manifest(self.context.get().dbt_file.dbt_manifest),
+            dbt_sources=parse_sources(self.context.get().dbt_file.dbt_sources),
             dbt_run_results=[
                 parse_run_results(run_result_file)
                 for run_result_file in self.context.get().dbt_file.dbt_run_results

metadata/ingestion/source/database/dbt/dbt_utils.py

@@ -44,6 +44,20 @@ def create_test_case_parameter_definitions(dbt_test):
                 }
             ]
             return test_case_param_definition
+        if hasattr(dbt_test, "freshness"):
+            test_case_param_definition = [
+                {
+                    "name": "warn_after",
+                    "displayName": "warn_after",
+                    "required": False,
+                },
+                {
+                    "name": "error_after",
+                    "displayName": "error_after",
+                    "required": False,
+                },
+            ]
+            return test_case_param_definition
     except Exception as err:  # pylint: disable=broad-except
         logger.debug(traceback.format_exc())
         logger.error(
@@ -67,6 +81,21 @@ def create_test_case_parameter_values(dbt_test):
                 {"name": manifest_node.test_metadata.name, "value": dbt_test_values}
             ]
             return test_case_param_values
+        if hasattr(manifest_node, "freshness"):
+            warn_after = manifest_node.freshness.warn_after
+            error_after = manifest_node.freshness.error_after
+
+            test_case_param_values = [
+                {
+                    "name": "error_after",
+                    "value": f"{error_after.count} {error_after.period.value}",
+                },
+                {
+                    "name": "warn_after",
+                    "value": f"{warn_after.count} {warn_after.period.value}",
+                },
+            ]
+            return test_case_param_values
     except Exception as err:  # pylint: disable=broad-except
         logger.debug(traceback.format_exc())
         logger.error(