diff --git a/pyatlan/model/enums.py b/pyatlan/model/enums.py index ba10bc453..3762a3c95 100644 --- a/pyatlan/model/enums.py +++ b/pyatlan/model/enums.py @@ -250,7 +250,9 @@ def to_qualified_name(self): AIRFLOW_CLOUD_COMPOSER = ("airflow-cloud-composer", AtlanConnectionCategory.ELT) SPARK = ("spark", AtlanConnectionCategory.ELT) MPARTICLE = ("mparticle", AtlanConnectionCategory.DATABASE) + ESSBASE = ("essbase", AtlanConnectionCategory.DATABASE) GENERIC = ("genericdb", AtlanConnectionCategory.DATABASE) + FILE = ("file", AtlanConnectionCategory.OBJECT_STORE) class AtlanCustomAttributePrimitiveType(str, Enum): diff --git a/tests/integration/file_test.py b/tests/integration/file_test.py index cc21d85b7..d7fa0f8f7 100644 --- a/tests/integration/file_test.py +++ b/tests/integration/file_test.py @@ -18,7 +18,7 @@ MODULE_NAME = TestId.make_unique("File") -CONNECTOR_TYPE = AtlanConnectorType.API +CONNECTOR_TYPE = AtlanConnectorType.FILE FILE_NAME = f"{MODULE_NAME}-file.pdf" CERTIFICATE_STATUS = CertificateStatus.VERIFIED CERTIFICATE_MESSAGE = "Automated testing of the Python SDK."