mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 04:04:14 +00:00
adding batch inserts to sqlstore and authorized sqlstorealong with type adjustments and conversations
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
d2ce672d4b
commit
fe695ca475
5 changed files with 54 additions and 17 deletions
|
@ -368,6 +368,32 @@ async def test_where_operator_gt_and_update_delete():
|
|||
assert {r["id"] for r in rows_after} == {1, 3}
|
||||
|
||||
|
||||
async def test_batch_insert():
|
||||
with TemporaryDirectory() as tmp_dir:
|
||||
db_path = tmp_dir + "/test.db"
|
||||
store = SqlAlchemySqlStoreImpl(SqliteSqlStoreConfig(db_path=db_path))
|
||||
|
||||
await store.create_table(
|
||||
"batch_test",
|
||||
{
|
||||
"id": ColumnType.INTEGER,
|
||||
"name": ColumnType.STRING,
|
||||
"value": ColumnType.INTEGER,
|
||||
},
|
||||
)
|
||||
|
||||
batch_data = [
|
||||
{"id": 1, "name": "first", "value": 10},
|
||||
{"id": 2, "name": "second", "value": 20},
|
||||
{"id": 3, "name": "third", "value": 30},
|
||||
]
|
||||
|
||||
await store.insert("batch_test", batch_data)
|
||||
|
||||
result = await store.fetch_all("batch_test", order_by=[("id", "asc")])
|
||||
assert result.data == batch_data
|
||||
|
||||
|
||||
async def test_where_operator_edge_cases():
|
||||
with TemporaryDirectory() as tmp_dir:
|
||||
db_path = tmp_dir + "/test.db"
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue