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 }}
|
||||
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:
|
||||
name: Delete custom graph data test ${{ matrix.python-version }} 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