github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add test_mock to Postgres blocks

Browse Source
This commit is contained in:
abhi1992002
2025-03-29 18:27:01 +05:30
parent 79651855c2
commit ea03c404b1
1 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
autogpt_platform/backend/backend/blocks/postgres.py
View File

@@ -133,6 +133,12 @@ class PostgresDeleteBlock(Block):
("success", True),
("rows_affected", 1)
],
test_mock={
"run": lambda *args, **kwargs: [
("success", True),
("rows_affected", 1)
]
},
)
def run(
@@ -237,6 +243,12 @@ class PostgresExecuteQueryBlock(Block):
("success", True),
("result", [{"id": 1, "name": "Test User"}])
],
test_mock={
"run": lambda *args, **kwargs: [
("success", True),
("result", [{"id": 1, "name": "Test User"}])
]
},
)
def run(
@@ -315,6 +327,13 @@ class PostgresInsertBlock(Block):
("rows_affected", 1),
("inserted_rows", [{"id": 1, "name": "Test User", "email": "test@example.com"}])
],
test_mock={
"run": lambda *args, **kwargs: [
("success", True),
("rows_affected", 1),
("inserted_rows", [{"id": 1, "name": "Test User", "email": "test@example.com"}])
]
},
)
def run(
@@ -413,6 +432,13 @@ class PostgresInsertOrUpdateBlock(Block):
("rows_affected", 1),
("affected_rows", [{"id": 1, "name": "Updated User", "email": "updated@example.com"}])
],
test_mock={
"run": lambda *args, **kwargs: [
("success", True),
("rows_affected", 1),
("affected_rows", [{"id": 1, "name": "Updated User", "email": "updated@example.com"}])
]
},
)
def run(
@@ -524,6 +550,12 @@ class PostgresSelectBlock(Block):
("success", True),
("rows", [{"id": 1, "name": "Test User", "email": "test@example.com"}])
],
test_mock={
"run": lambda *args, **kwargs: [
("success", True),
("rows", [{"id": 1, "name": "Test User", "email": "test@example.com"}])
]
},
)
def run(
@@ -627,6 +659,13 @@ class PostgresUpdateBlock(Block):
("rows_affected", 1),
("updated_rows", [{"id": 1, "name": "Updated User", "email": "updated@example.com"}])
],
test_mock={
"run": lambda *args, **kwargs: [
("success", True),
("rows_affected", 1),
("updated_rows", [{"id": 1, "name": "Updated User", "email": "updated@example.com"}])
]
},
)
def run(