From adbf3df4e04caa0979b418ef04f2c6398335fe48 Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Tue, 12 Nov 2024 14:12:16 +0100 Subject: [PATCH] Move cognee_db for integration test --- cognee/.cognee_system/databases/cognee_db | Bin 139264 -> 0 bytes .../integration/run_toy_tasks/conftest.py | 10 ++++ .../integration/run_toy_tasks/data/cognee_db | 0 .../run_toy_tasks}/test_run_tasks.py | 1 - .../test_run_tasks_from_queue.py | 0 cognee/tests/unit/run_tasks/conftest.py | 27 ++--------- cognee/tests/unit/utils/get_mock_user.py | 45 ------------------ 7 files changed, 13 insertions(+), 70 deletions(-) delete mode 100644 cognee/.cognee_system/databases/cognee_db create mode 100644 cognee/tests/integration/run_toy_tasks/conftest.py create mode 100644 cognee/tests/integration/run_toy_tasks/data/cognee_db rename cognee/tests/{unit/run_tasks => integration/run_toy_tasks}/test_run_tasks.py (94%) rename cognee/tests/{unit/run_tasks => integration/run_toy_tasks}/test_run_tasks_from_queue.py (100%) delete mode 100644 cognee/tests/unit/utils/get_mock_user.py diff --git a/cognee/.cognee_system/databases/cognee_db b/cognee/.cognee_system/databases/cognee_db deleted file mode 100644 index 60455ad298370fbce1f08c2ea7554de6a85116b8..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 139264 zcmeI*4{#ILodx{BVGKE4`q0`A^vg!Qa z0{(CGF_nLk8GXV3md8C$_jJ0f`0lk?LS^DJp-w^MSGwvbhX4d1009U<00Izz00bZa z0SG|g`U}jSlFJ8`6DMT+A3Puc0SG_<0uX=z1Rwwb2tWV=5E%agMtPRrCY!Fgw|%|M zxpd#&C0_FAkmC|f(u{$k45URz85dI2LXs+`Ek?7++7;r0YJOaD;)0C-g9ij4009U< z00Izz00bZa0SG_<0^b0EsTq2eG&_Lf|8KyG#SkC>0SG_<0uX=z1Rwwb2tWV=aRRB1 z|Ap)S)w$g=;zi_1#&2JbvPflmSdeW2yD! z%Rf?_n6}0^^uqVc_y4)Zy(n7@uwz=o-Usm(W-Gyg1vduZE15mV;NeZA41MD$DIhsQN0uX=z1Rwwb2tWV= z5P$##ZdQS58G5-iYoitlrI{F&P$g+gg&BTFchW-Bsic&&gUJ7c;E=l{R< z6@m^x00Izz00bZa0SG_<0uX=z1ip&EHJ|?%?*Gr4a7soLX&Q6yoVbsFiU$NB009U< z00Izz00ba#GYAYgvQ+vtQ)drsvdN?qzK6Nqu(PM5r^DxN+tlk}yLvjid%dowVJ>%V?pk^ngPV8j>+AFdUA|UdIN)pXw{P6o)fepCbfbpVGsgR-9Tsh|ko>UZv0+Ik(HDp1X$iIm z108I*tHa&Py8K+HuO-mV`C3Drn^uFq0|F3$00bZa0SG_<0uX=z1Rwx`aU)PCSL)~4WcpV7=C>!ZjaTAt;7yyVd#$0gzXznq`?AG80Dn-z?XKmY;| zfB*y_009U<00Izz00h30z|;)AN}4sm@&8vMMPmp+00Izz00bZa0SG_<0uX?}xDybL z|E1^uh3o$nx$9+`r}-avKmY;|fB*y_009U<00Izzz?TvjT%pL;+wA(GQ12~M?kKsZ z?$LkQ_3!7mZn*MT()HOiQ*1Pv%?w#(bJ&{f4(E!-hDv)=Q*C{%r-2#)@;$ldb*faFK!x-F3gUHf{=YA^k)sg=AOHafKmY;| zfB*y_009U<00Li^fG~ZaH0dPH|0hkG#E$2X=s);6tfUOAWERBiE*3j+QQrnA~uLv!vb2kEM+v99X( zcvr2iWma0(RIT=PEOwBX|8M+mFX$NrAOHafKmY;|fB*y_009U<;L8Z4n*T3y{Qs_u zc=yXhL>mY|00Izz00bZa0SG_<0uX=z1jdWNZ5d^%bQc&1*Z<2VjhCLH9}s{51Rwwb z2tWV=5P$##AOHaf+z0{T{6CKWZ-gvjLI45~fB*y_009U<00Izz00hR50FM91&$>WQ zAOHafKmY;|fB*y_009U<00K8kK)C*2J@KTBc#2q0s5Q@PS~b&hkL3n(Z|4i~fB*y_ z009U<00Izz00bZ~P6bw|m3o^@ufO!n#e#n|Hut`Z`}QvJl1GOemja*c=o%v9NXK z;{%oFw!F7v?-rG-KXpXX%p^s8g89Iq{kZb3!^OV0GP=mmR^&t?Hp)M3Sf;Dm^5{#O zo~JD5XQYhCPxW4E9J7zvk%*t1zw9FKTi(3SK;8A|najUQ9g(z>7HPz!J}M&-*`fcv zc(t-R{1g467iLwKew;EQW3*VU(uhfpS&@j!pKbWrxArWXyIXdo;*E2`_S6w6(qxiG zOzLB1B;vu3j<0?sP+PWR|L!?Q>*oDi%7{kBXeyOPOzLAsB;wU`6N|sKbLj{B`b$p! z^MiX#sUuRRQc4;zsgH_C#Q$(LJ^SBu1^sOKs{!`m!t*I3(u~QRH0q>2$|Di;D-L@e zd$_K0+pF7mR-Ss9OC6CS$6nXM`F}N0A|oClE)dN`6){9SNxV*!5I*AAIQ^KTTM&Q% z1Rwwb2tWV=5P$##AOL|IEif%ZFPEm5)Iy;&L!=T4rAeL~p-`Hq$rcKwsTQSBD9x5+ z35C*xN2X9H%~@m!h0?TxLYV(wnmG{8|7U4lk`XTvTZjrm#y`aa0uX=z1Rwwb2tWV= z5P$##ATaI(%Cjaz)zqw3VVurR`VWSMlpTJ6FE<^wdS?CNDhm*K&E<_LSLTP3r&72Y;(H zw68mGU`F7>nN3ga5YGQ+5tTCH5b-8aNqC3{#@#ALhadm}2tWV=5P$##AOHafKmY>c zL112{UN4s(r%Uq;St)Btvk93gYe{nk87XT?GXsjbe7{uEeo2r2@&EsghfN55fB*y_ z009U<00Izz00bZa0SMe!0nGn@V|@`F0uX=z1Rwwb2tWV=5P$##ATS;T#QFbox5VB5 zui2uh(8zLM;tTPB00bZa0SG_<0uX=z1Rwx`>n2dHPJ92hnPEs;I@#QxUgI`j%&*8e zJoB9w9#`*wzg(5RJwMmKSvoIWcim(9Z}-%eAKmrC+PtHV-13~X?M+6jm6A?j_fg=Z zosDlT{@ebm@6P(;@e^N^XQyvZQ5Lgww!3wI*g5&dU5oPu`rhy~oc;VyfTY_Zl@83?6VYpgs&D$e|9`CG|J)qCY>M(z88MkSt~ru>O0#X^+qn-%&vQ{a6l5s)atO&?&m*0*J83GW100ba#vj_~9Ov%=2 zweqcHVYbD~b%nTKNc=Iive9mH*mVxu;ySxdEYi(at2}Pq;@Wy!WAl8k50i zIo;xhhB~{gKH+WnrJ=3>7ZiGwu0l5#^t5_7zN;w$tZ1xV&i7xp%-*bZa6f*zmC6JWg|v9`L1xGY<6jceCGIa~XkLivEo<8yPp zo?d6b6X3iaALk5q`9i`Er^lU8Hf317w!X@~R+pxQuAyF+&`LKy5~^sRipbU$7Rm>! z#kHAGDdEMWgteQnX!N!03UX|iUyp29SH*80j@spR0WcJ1!(E}+N~J9#You3k8;np@ zEcCVdbxWHX>V*R73Q-qXCn>ed3*|DAE!64tggGbM750m-C5Ti)=O^ExQEE;3^373k zHydU{TsY+P_xQLV&mAowq|(s^v6=#Nv`{xc+P-LjnV7B3&zBE=M>LMsi2lw^kd2m( zj_PAOIGw}3)**1D5$+1`v6A7V-I5?Do0jA%wUznlOr!mDQqgCcf$yrbwbQ4|hw@_M z6qZd$8s9aHZFsLB43Fy#bH0$r?~86MLScuegL8)a0$hAyknM2_@96aUU2Is=;_p*R z<|v$?)P86BXdDrWiR)BKZOI&YXLN(&cRd}RkT8Id@ZlYKrIv0`am)Cc;)iymwzxCd zn3JtF&XEsfrwT4?6mcO`!{Jxe*REV;*NyBTMq)a}?MT=;jFjo-3x!1kv@%;eXO4Vu zL1f1;QX#HxcEYNSTQK?>{`*#i*vH}RAz{a!ypBxSN^R|&bn7Ute@-gyMb3aZGg~`r zR`dfJ(M;Q$&toStlUM7#1%2I0Wv(k+>8ZzXhVo#N;Gn87x zYJ?XD2|k0PsaFN z!WbKiQK8gU&rXLo(mj4#i47)$|93n`__tiR{y#V8jEs1V7|^__c}mlqTa+uC@QVq9 z>P>1!MevL8fB*y_009U<00QG)z&dNRgJ{;}b;iC%I;n)0y1|0m#Wcd!4Ds7v!oq|X zvl6}oCM*)Z2BtYO5bCA>UL5V!C{+@^I!B`Eb#d*IaypSZ33(m9D3Z|;jz8n`Kk;$m z6G74PXdcLvKibiWke+h31^rzC{`-2|i`xgM%@mKshWw&!+@iRr`3bIZ3r0~DYo%uu z`*N+O!?E}>UR9`Zt*9f>^kQC#Zmv{i{2xcE(YxK~LbMC;_{>Y~ED(Xt`JM@8I4gk}sHZWWV4!|7sSiSRcs zK{xWz8ZB3xX2*~w#}zwZOOzHqxFg5?$;%S<5-F_X!zNJ07Np*o6W%ki;YeC#c(ZQO zDz!y<@=)~nJhIJ;iJHjMTZiSOX`Q%=!U~IiR>Wk^f?Jf@MS1ByE5j|R$Zzeyoax!x zygd2fY;k`RDUJNRC1I^a%0@qWzvkqPbf-F2QoV5+@1z}Z5_cs|@sxu2bb1=5xOxK9 Z@E%BPS5!1rsV&o{V~VsCcg>+`{|}Zl+Y 0: - apply_with_to_item(module_items[(i + 1) :], fn, fn_str) - - @pytest.fixture(autouse=True, scope="session") -def set_get_mock_user_wrapper(): - - def get_mock_user_wrapper(): - return get_mock_user(None, None) - - module_items = list(sys.modules.items()) - apply_with_to_item(module_items, get_mock_user_wrapper(), "get_default_user") - - yield +def copy_cognee_db_to_target_location(): + os.system("mv cognee/tests/integration/run_toy_tasks/data/cognee_db cognee/.cognee_system/databases") diff --git a/cognee/tests/unit/utils/get_mock_user.py b/cognee/tests/unit/utils/get_mock_user.py deleted file mode 100644 index 76ce62cd0..000000000 --- a/cognee/tests/unit/utils/get_mock_user.py +++ /dev/null @@ -1,45 +0,0 @@ -import warnings -from typing import List, Optional -from unittest.mock import Mock, create_autospec -from uuid import UUID, uuid4 - -from cognee.modules.users.models import User - - -def get_mock_user( - user_id: Optional[UUID] = None, - groups: Optional[List["Group"]] = None, - **additional_attributes -) -> Mock: - """ - Creates a mock User instance with configurable attributes. - - Args: - user_id: Optional UUID for the user. Generates random UUID if not provided. - groups: Optional list of group mocks to associate with the user. - **additional_attributes: Any additional attributes to set on the mock user. - - Returns: - Mock: A configured mock User instance. - """ - warnings.warn("\n\n\n---------------USING MOCK USER--------------------\n\n\n") - - # Generate a random UUID if none provided - user_id = user_id or uuid4() - - # Create base mock - mock_user = create_autospec(User, instance=True) - - # Configure basic attributes - mock_user.id = user_id - mock_user.__tablename__ = "users" - mock_user.groups = groups or [] - - # Set polymorphic identity - mock_user.__mapper_args__ = {"polymorphic_identity": "user"} - - # Add any additional attributes - for attr_name, attr_value in additional_attributes.items(): - setattr(mock_user, attr_name, attr_value) - - return mock_user