fix: remove unused set_current_user and add missing delete test to e2e_tests
This commit is contained in:
parent
8562eec85f
commit
fa450fa727
2 changed files with 31 additions and 8 deletions
31
.github/workflows/e2e_tests.yml
vendored
31
.github/workflows/e2e_tests.yml
vendored
|
|
@ -325,6 +325,37 @@ jobs:
|
||||||
EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
|
EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
|
||||||
run: uv run python ./cognee/tests/test_delete_default_graph.py
|
run: uv run python ./cognee/tests/test_delete_default_graph.py
|
||||||
|
|
||||||
|
test-deletion-on-default-graph-non-mocked:
|
||||||
|
name: Delete default graph data test ${{ matrix.python-version }} on ubuntu-22.04
|
||||||
|
runs-on: ubuntu-22.04
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
python-version: ${{ fromJSON(inputs.supported-python-versions) }}
|
||||||
|
fail-fast: false
|
||||||
|
steps:
|
||||||
|
- name: Check out
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
fetch-depth: 0
|
||||||
|
|
||||||
|
- name: Cognee Setup
|
||||||
|
uses: ./.github/actions/cognee_setup
|
||||||
|
with:
|
||||||
|
python-version: ${{ matrix.python-version }}
|
||||||
|
|
||||||
|
- name: Run deletion on default graph non mocked
|
||||||
|
env:
|
||||||
|
ENV: 'dev'
|
||||||
|
LLM_MODEL: ${{ secrets.LLM_MODEL }}
|
||||||
|
LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
|
||||||
|
LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
|
||||||
|
LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }}
|
||||||
|
EMBEDDING_MODEL: ${{ secrets.EMBEDDING_MODEL }}
|
||||||
|
EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }}
|
||||||
|
EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }}
|
||||||
|
EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
|
||||||
|
run: uv run python ./cognee/tests/test_delete_default_graph_non_mocked.py
|
||||||
|
|
||||||
test-deletion-on-custom-graph:
|
test-deletion-on-custom-graph:
|
||||||
name: Delete custom graph data test ${{ matrix.python-version }} on ubuntu-22.04
|
name: Delete custom graph data test ${{ matrix.python-version }} on ubuntu-22.04
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-22.04
|
||||||
|
|
|
||||||
|
|
@ -1,8 +0,0 @@
|
||||||
import uuid
|
|
||||||
from sqlalchemy import text
|
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
|
||||||
|
|
||||||
|
|
||||||
async def set_current_user(session: AsyncSession, user_id: uuid.UUID, local: bool = False):
|
|
||||||
scope = "LOCAL " if local else ""
|
|
||||||
await session.execute(text(f"SET {scope}app.current_user_id = '{user_id}'"))
|
|
||||||
Loading…
Add table
Reference in a new issue