diff --git a/docs/internals.rst b/docs/internals.rst index 43944de9..7b4c1755 100644 --- a/docs/internals.rst +++ b/docs/internals.rst @@ -166,7 +166,7 @@ Example usage: .. code-block:: python - def get_version(conn); + def get_version(conn): return conn.execute( "select sqlite_version()" ).fetchall()[0][0] diff --git a/tests/test_database.py b/tests/test_database.py index d4055776..1f1a3a7e 100644 --- a/tests/test_database.py +++ b/tests/test_database.py @@ -7,7 +7,7 @@ import uuid @pytest.mark.asyncio -async def test_execute1(app_client): +async def test_execute(app_client): db = app_client.ds.databases["fixtures"] results = await db.execute("select * from facetable") assert isinstance(results, Results) @@ -42,6 +42,16 @@ async def test_results_single_value(app_client, query, expected): results.single_value() +@pytest.mark.asyncio +async def test_execute_fn(app_client): + db = app_client.ds.databases["fixtures"] + + def get_1_plus_1(conn): + return conn.execute("select 1 + 1").fetchall()[0][0] + + assert 2 == await db.execute_fn(get_1_plus_1) + + @pytest.mark.parametrize( "tables,exists", (