diff --git a/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/conftest.py b/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/conftest.py index b70a2736e6be3..a9e8c93c31d2b 100644 --- a/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/conftest.py +++ b/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/conftest.py @@ -273,12 +273,12 @@ def all_api_mocks_fixture( @pytest.fixture(name="airbyte_cloud_sync_and_poll") def sync_and_poll_fixture(): with patch("dagster_airbyte.resources.AirbyteCloudClient.sync_and_poll") as mocked_function: - # Airbyte output where all sync'd tables match the workspace data that was used to create the assets def + # Airbyte output where all synced tables match the workspace data that was used to create the assets def expected_airbyte_output = AirbyteOutput( connection_details=SAMPLE_CONNECTION_DETAILS, job_details=get_job_details_sample(status=AirbyteJobStatusType.SUCCEEDED), ) - # Airbyte output where a table is missing and an unexpected table is sync'd, + # Airbyte output where a table is missing and an unexpected table is synced, # compared to the workspace data that was used to create the assets def unexpected_airbyte_output = AirbyteOutput( connection_details=UNEXPECTED_SAMPLE_CONNECTION_DETAILS, diff --git a/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/test_resources.py b/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/test_resources.py index d3378be34e07c..82fde4a0a2f0c 100644 --- a/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/test_resources.py +++ b/python_modules/libraries/dagster-airbyte/dagster_airbyte_tests/experimental/test_resources.py @@ -7,6 +7,7 @@ import pytest import responses from dagster import AssetExecutionContext, AssetKey, EnvVar, Failure, materialize +from dagster._core.events import DagsterEventType from dagster._core.test_utils import environ from dagster_airbyte import AirbyteCloudWorkspace, airbyte_assets from dagster_airbyte.resources import ( @@ -404,7 +405,7 @@ def my_airbyte_assets(context: AssetExecutionContext, airbyte: AirbyteCloudWorks asset_materializations = [ event for event in result.events_for_node(cleaned_connection_id) - if event.event_type_value == "ASSET_MATERIALIZATION" + if event.event_type_value == DagsterEventType.ASSET_MATERIALIZATION ] assert len(asset_materializations) == 2 materialized_asset_keys = { @@ -424,7 +425,7 @@ def my_airbyte_assets(context: AssetExecutionContext, airbyte: AirbyteCloudWorks asset_materializations = [ event for event in result.events_for_node(cleaned_connection_id) - if event.event_type_value == "ASSET_MATERIALIZATION" + if event.event_type_value == DagsterEventType.ASSET_MATERIALIZATION ] assert len(asset_materializations) == 2 materialized_asset_keys = {