From 2e3db639d2a2ef0b6b27e15eca0970457e6822bb Mon Sep 17 00:00:00 2001
From: Cheikh Gueye Wane <cwane@bluesquarehub.com>
Date: Mon, 4 Dec 2023 16:47:41 +0000
Subject: [PATCH] feat(Connections): import and re-export connection classes in
 openhexa.sdk (#93)

---
 openhexa/sdk/__init__.py | 13 ++++++++++++-
 tests/test_parameter.py  |  2 +-
 tests/test_pipeline.py   |  2 +-
 3 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/openhexa/sdk/__init__.py b/openhexa/sdk/__init__.py
index 25e8aba..d7d1a71 100644
--- a/openhexa/sdk/__init__.py
+++ b/openhexa/sdk/__init__.py
@@ -1,4 +1,15 @@
 from .pipelines import current_run, parameter, pipeline
 from .workspaces import workspace
+from .workspaces.connection import DHIS2Connection, IASOConnection, PostgreSQLConnection, GCSConnection, S3Connection
 
-__all__ = ["workspace", "pipeline", "parameter", "current_run"]
+__all__ = [
+    "workspace",
+    "pipeline",
+    "parameter",
+    "current_run",
+    "DHIS2Connection",
+    "IASOConnection",
+    "PostgreSQLConnection",
+    "GCSConnection",
+    "S3Connection",
+]
diff --git a/tests/test_parameter.py b/tests/test_parameter.py
index 2dc4ac6..4f6c99f 100644
--- a/tests/test_parameter.py
+++ b/tests/test_parameter.py
@@ -2,7 +2,7 @@
 import os
 import stringcase
 
-from openhexa.sdk.workspaces.connection import (
+from openhexa.sdk import (
     DHIS2Connection,
     IASOConnection,
     PostgreSQLConnection,
diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
index ebb96b2..b82aeb7 100644
--- a/tests/test_pipeline.py
+++ b/tests/test_pipeline.py
@@ -4,7 +4,7 @@
 import stringcase
 import os
 
-from openhexa.sdk.workspaces.connection import (
+from openhexa.sdk import (
     DHIS2Connection,
     IASOConnection,
     PostgreSQLConnection,