diff --git a/tests/functional/adapter/upload_file_tests/test_upload_file.py b/tests/functional/adapter/upload_file_tests/test_upload_file.py index 474db11ed..ba9a2289b 100644 --- a/tests/functional/adapter/upload_file_tests/test_upload_file.py +++ b/tests/functional/adapter/upload_file_tests/test_upload_file.py @@ -1,5 +1,6 @@ import pytest from dbt.tests.util import get_relation_columns, run_dbt, run_sql_with_adapter +from dbt.contracts.results import NodeStatus import datetime import yaml @@ -75,7 +76,7 @@ def test_bigquery_upload_file_csv(self, project): } ) upload_result = run_dbt(["run-operation", "upload_file", "--args", upload_args]) - assert upload_result.success + assert upload_result.results[0].status == NodeStatus.Success # Check if the uploaded table contains expected values and schema self.perform_uploaded_table_checks(project.test_schema, "TestUploadFileCSV", project) @@ -94,7 +95,7 @@ def test_bigquery_upload_file_ndjson(self, project): } ) upload_result = run_dbt(["run-operation", "upload_file", "--args", upload_args]) - assert upload_result.success + assert upload_result.results[0].status == NodeStatus.Success # Check if the uploaded table contains expected values and schema self.perform_uploaded_table_checks(project.test_schema, "TestUploadFileNDJSON", project) @@ -112,7 +113,7 @@ def test_bigquery_upload_file_parquet(self, project): } ) upload_result = run_dbt(["run-operation", "upload_file", "--args", upload_args]) - assert upload_result.success + assert upload_result.results[0].status == NodeStatus.Success # Check if the uploaded table contains expected values and schema self.perform_uploaded_table_checks(project.test_schema, "TestUploadFileParquet", project)