fix(Connections): specify connection workspace (#170)

BLSQ · Jun 25, 2024 · f8b2dda · f8b2dda
1 parent 841f47c
commit f8b2dda
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/openhexa/sdk/workspaces/current_workspace.py b/openhexa/sdk/workspaces/current_workspace.py
@@ -186,8 +186,8 @@ def get_connection(
         if self._connected:
             response = graphql(
                 """
-                query getConnection($slug: String!) {
-                    connectionBySlug(slug: $slug) {
+                query getConnection($workspaceSlug:String!, $connectionSlug: String!) {
+                    connectionBySlug(workspaceSlug:$workspaceSlug, connectionSlug: $connectionSlug) {
                         type
                         fields {
                             code
@@ -196,7 +196,7 @@ def get_connection(
                     }
                 }
             """,
-                {"slug": identifier},
+                {"workspaceSlug": self.slug, "connectionSlug": identifier},
             )
             data = response["connectionBySlug"]
             if data is None:

diff --git a/tests/test_workspace.py b/tests/test_workspace.py
@@ -269,6 +269,7 @@ def setup_method(self, monkeypatch):
         """Basics setup for the TestConnectedWorkspace."""
         # Mock the class variable
         monkeypatch.setenv("HEXA_SERVER_URL", "http://app.openhexa.test")
+        monkeypatch.setenv("HEXA_WORKSPACE", "workspace")
 
     def test_workspace_files_path(self, monkeypatch, workspace):
         """Basic checks for the Workspace.files_path() method."""