Skip to content

Commit 5be3f58

Browse files
committed
fixed var names, assertion err
1 parent 8bc0967 commit 5be3f58

3 files changed

Lines changed: 67 additions & 1 deletion

File tree

agentstack/_tools/py_sql/config.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
{
2-
"name": "sql-queries",
2+
"name": "py_sql",
33
"url": "https://pypi.org/project/psycopg2/",
44
"category": "database",
55
"env": {

agentstack/_tools/py_sql/test.db

12 KB
Binary file not shown.

agentstack/_tools/py_sql/test.py

Lines changed: 66 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,66 @@
1+
from sql import Table
2+
from __init__ import construct_sql_query
3+
4+
def test_query_construction():
5+
"""Test query construction without a real database"""
6+
7+
# Define our test table structure
8+
users = Table('users')
9+
10+
print("\n=== Testing SELECT queries ===")
11+
# Test basic select
12+
query, params = construct_sql_query(
13+
"select",
14+
"users",
15+
columns=[users.name, users.age],
16+
where=users.age > 18
17+
)
18+
print("Select users over 18:")
19+
print(f"Query: {query}")
20+
print(f"Params: {params}")
21+
22+
# Test select with multiple conditions
23+
query, params = construct_sql_query(
24+
"select",
25+
"users",
26+
columns=[users.name, users.email],
27+
where=(users.age > 18) & (users.active == True)
28+
)
29+
print("\nSelect active users over 18:")
30+
print(f"Query: {query}")
31+
print(f"Params: {params}")
32+
33+
print("\n=== Testing INSERT queries ===")
34+
query, params = construct_sql_query(
35+
"insert",
36+
"users",
37+
values=[["John Doe", 25, "john@example.com", True]]
38+
)
39+
print("Insert new user:")
40+
print(f"Query: {query}")
41+
print(f"Params: {params}")
42+
43+
print("\n=== Testing UPDATE queries ===")
44+
query, params = construct_sql_query(
45+
"update",
46+
"users",
47+
columns=[users.active],
48+
values=[False],
49+
where=users.age < 18
50+
)
51+
print("Deactivate users under 18:")
52+
print(f"Query: {query}")
53+
print(f"Params: {params}")
54+
55+
print("\n=== Testing DELETE queries ===")
56+
query, params = construct_sql_query(
57+
"delete",
58+
"users",
59+
where=users.active == False
60+
)
61+
print("Delete inactive users:")
62+
print(f"Query: {query}")
63+
print(f"Params: {params}")
64+
65+
if __name__ == "__main__":
66+
test_query_construction()

0 commit comments

Comments
 (0)