j, d Z ddlmZ ddlZddlmc mZ ddl Z ddl
Z
ddl Z ddl m
Z
ddlmZmZ ddlmZ ddlmZ ddlmZ dd lmZ G d
d e
j ZdS ) zNUnit and integration tests for document upload and AI classification pipeline. ) annotationsN)Path) MagicMockpatch)
TestClient)get_current_user)
create_app)DocumentProcessingServicec e Zd Zd$dZd$dZd$dZd$dZ ed d%d
Z ed ed ed
ed ed ed d&d Z ed ed ed ed ed d'd! Z
ed d(d" Z d#S ))TestDocumentUploadAireturnNonec g | _ d S N)
temp_files)selfs 3C:\Projects\DSCons\tests\test_document_upload_ai.pysetUpzTestDocumentUploadAi.setUp s
&( c | j D ];} | r| ,# t $ r Y 8w xY wd S r )r existsunlink Exception)r ps r tearDownz TestDocumentUploadAi.tearDown s` A
88:: HHJJJ
s (4
A Ac > t }d}d}| || \ }}}}}t | } | j | | | | |t | | |d | | d | |
| | | d S )Ns Mock document contentz test_doc.txttxtz/v1/erp/documents/vault/) r
save_filer r append
assertTruer assertEquallen
startswith
read_bytesassertIsNotNone)
r service file_contentfilenamefile_url
file_path_str file_size file_format file_hash file_paths
r test_save_filez#TestDocumentUploadAi.test_save_file s +--/ !ELEVEV (F
F
B-K '' y))) ((**+++ C $5$5666 e,,, ++,FGGHHH --// >>> Y'''''r c t }t j }t j |d 5 }d}| d| d d d n # 1 swxY w Y | }| |d \ }}}} }
| j t | | || } | | d d S )Nwu
Biên bản nghiệm thu công trình số 456
zword/document.xmlztest_document.docxu. Biên bản nghiệm thu công trình số 456)
r
ioBytesIOzipfileZipFilewritestrgetvaluer r r r extract_textr! ) r r& docx_bufferz xml_content docx_datar) r* r+ r, r- extracted_texts r test_extract_text_docxz+TestDocumentUploadAi.test_extract_text_docx1 s, +--jll _[#
.
.
9!
JJ*K 8 8 8
9
9
9
9
9
9
9
9
9
9
9
9
9
9
9 ((** ELEVEV +F
F
B-K
tM 2 2333 --m[II )YZZZZZs A A #A z8app.modules.agents.application.llm_client.LLMClient.chat mock_chatr c dd l }dd
fd }||_ t fd }| | d S ) Nr u ```json
{
"document_group": "acceptance",
"document_type": "BB",
"document_code": "BB-NT-456",
"document_title": "Biên bản nghiệm thu hoàn thành",
"issue_date": "2026-08-19",
"issuer_name": "DSCons Corp",
"signer_name": "Nguyễn Văn A",
"summary_content": "Nghiệm thu hoàn thành hạng mục kết cấu tầng hầm.",
"ai_risk_assessment": "Không phát hiện rủi ro.",
"ai_legal_references": ["Luật Xây dựng 2025 (Điều 135)"]
}
```argsAnykwargsr
strc K S r )rA rC
mock_responses r fake_chatz.fake_chatc s
r c \ K dd d {V } | d d | d d | d d | d d
| d d d S )
NzMock document text contentztest.pdfdocument_group
acceptance
document_typeBB
document_codez BB-NT-456document_titleu% Biên bản nghiệm thu hoàn thành signer_nameu Nguyễn Văn A)analyze_with_air! )resultr r& s r run_testz;TestDocumentUploadAi.test_analyze_with_ai..run_testj s "22,j F
V$4 5| D D D VO 4d ; ; ; VO 4k B B B V$4 57^ _ _ _ VM 24E F F F F Fr )rA rB rC rB r
rD )asyncio side_effectr
run)r r? rT rH rS rG r& s ` @@r test_analyze_with_aiz)TestDocumentUploadAi.test_analyze_with_aiN s
" ! ! ! ! ! ! !* +-- G G G G G G HHJJ r zKapp.core.postgres.erp_client.ErpDatabaseClient.create_project_from_documentz;app.core.postgres.erp_client.ErpDatabaseClient.get_documentzhapp.modules.core.application.document_processing_service.DocumentProcessingService.run_cross_agent_auditzbapp.modules.core.application.document_processing_service.DocumentProcessingService.analyze_with_aiz_app.modules.core.application.document_processing_service.DocumentProcessingService.extract_textz>app.core.postgres.erp_client.ErpDatabaseClient.update_document mock_update mock_extract mock_analyzemock_cross_audit mock_get_docmock_create_projc dd l }ddd d|_ ddddd |_ d
_ d d g d
|_ ddddddddddgd
_ t fd}| | d S )Nr
doc-uuid-1zTEMP-123)idrN
project_idTzp-1DA)r` project_name) created_newprojectuj Cộng hòa Xã hội Chủ nghĩa Việt Nam. extracted sample text hợp đồng kinh tế đầy đủ._ u3 Hồ sơ trích xuất hoàn chỉnh, chuẩn xác.)confidence_scorecross_audit_summaryflagscontractHDHD-123 Hợp đồng thiết kếz
2026-08-19z
DSCons Ltdu Trần Bu4 Hợp đồng thiết kế kiến trúc cảnh quan.u Có rủi ro chậm thanh toán.u Nghị định 206/2026/NĐ-CP)
rJ rL rN rO
issue_date issuer_namerP summary_contentai_risk_assessmentai_legal_referencesc K dddd d {V dd j d j d } | d d | d } |d d |d d
|d d |d
d |d d |d d d S )Nr_ z/tmp/fake_path.pdfz
fake_path.pdfpdfr rN rl rO rm rJ rj rL rk statuspending_reviewverification_status) run_pipelineassert_called_once_with return_valueassert_called_once call_argsr! )r} payloadrZ rY rX r r& s r rS z8TestDocumentUploadAi.test_run_pipeline..run_test st && 2OU
0 01Eu M M M 0 0 )?
* * , , , $-a0I Yq\< 8 8 8 lG W_ 5x @ @ @ W%5 68T U U U W%5 6
C C C W_ 5t < < < WX .0@ A A A W%: ;=M N N N N Nr )rT r{ r
rV )
r rX rY rZ r[ r\ r] rT rS r& s
```` @r test_run_pipelinez&TestDocumentUploadAi.test_run_pipelinev s . ' %
%
! #T::)
)
% %Q ! "#X)
)
% ) ! % :&'% U"D$D#E %
%
! ,-- O O O O O O O O O* HHJJ r zGapp.core.postgres.erp_client.ErpDatabaseClient.check_duplicate_documentz>app.core.postgres.erp_client.ErpDatabaseClient.create_documentz=app.core.postgres.erp_client.ErpDatabaseClient.get_connectionz\app.modules.core.application.document_processing_service.DocumentProcessingService.save_file mock_save
mock_get_connmock_update_docmock_create_docmock_check_dupc t }t }ddiddig|j _ ||j j j _ ||j j _ d|_ d |_ ddi|_ ddd|_ t
}d |j t < t | } t d 5 }
| d
d d id
di } | | j
d | | d d | | |
d d d d S # 1 swxY w Y d S )Nr` zcompany-uuid-1zproject-uuid-1)z/static/uploads/file.pdfz/disk/file.pdfd rt
mock_hash_123 doc-uuid-101
processing)r` rv c ddddS Nzuser-1ztest@dscons.vn
superadmin)r` emailrolerF rF r r z?TestDocumentUploadAi.test_api_upload_document.. % >
>
r z fastapi.BackgroundTasks.add_taskz/v1/erp/documents/uploadfile)z
report.pdfs pdf content byteszapplication/pdf project_codezPRJ-999)filesdata )r fetchonerU cursorr{ __enter__r dependency_overridesr r r postr! status_codejsonr| ) r r r r r r mock_connmock_curappclient
mock_add_taskresponses r test_api_upload_documentz-TestDocumentUploadAi.test_api_upload_document s KK ;;
# $
# $)
% @H %/<. r r z&/v1/erp/documents/doc-uuid-101/approveu Đã phê duyệt trích xuấtz CV-OK-101)rO rN )r r rv rx r ru rO rN ) r{ r r r r r r! r r r| r} )r rX r r r r~ s r test_api_approve_documentz.TestDocumentUploadAi.test_api_approve_document so ! #-$
$
ll6
6
!12
C ;; 4"C!,
-s333 2G<<< )>?LLL&&((( '*1- *G444 !67DDD !124UVVV 1;?????r N)r
r )r? r r
r )rX r rY r rZ r r[ r r\ r r] r r
r ) r r r r r r r r r r r
r )rX r r
r ) __name__
__module__ __qualname__r r r/ r> r rW r r r rF r r r r s ) ) ) ) ( ( ( (&[ [ [ [: U EFF% % % GF% N UU U HII
Ur Ul Ui U KLL> > > ML JI > @ U TUU
U KLL
U KLL
U JKK
Uf 0/ 0/ 0/ LK ML ML VU0/d U KLL @ @ @ ML @ @ @r r )__doc__
__future__r builtins @py_builtins_pytest.assertion.rewrite assertionrewrite
@pytest_arr2 unittestr4 pathlibr
unittest.mockr r fastapi.testclientr "app.modules.auth.presentation.authr app.mainr 8app.modules.core.application.document_processing_servicer
TestCaser rF r r r s T T " " " " " " * * * * * * * * ) ) ) ) ) ) ? ? ? ? ? ? ^ ^ ^ ^ ^ ^K@ K@ K@ K@ K@8, K@ K@ K@ K@ K@r