P1ie b d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
mZ ddlmZmZmZ dd lmZ dd
lmZ dd lmZmZmZmZm Z e Z G d d
e Z ed G d de Z ed G d de Z G d de Z! G d de Z" G d de Z#y)z
Comprehensive unit tests for AI analysis reporting email system.
Tests cover:
- Admin notification emails
- User resolution notification emails
- Email template rendering
- Permission-based email recipients
- Email failure handling
)TestCaseoverride_settings)get_user_model)
Permission)mail)timezone)datetime)patch MagicMockcall)Person)liuyao)get_report_recipients send_admin_report_notification!send_user_resolution_notificationget_category_displayget_admin_action_displayc @ e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
d Z y
) EmailRecipientsTestsz/Test email recipient management and permissionsc t j j dddddd | _ t j j dddd dd | _ t j j d
d d d
dd | _ t j j d j t j j ddd | _
y)Set up test data 13800138001admin1@example.comAdminOneT)phoneemail
first_name last_nameis_staff is_active 13800138002admin2@example.comTwo 13800138003user@example.comRegularUserF can_receive_ai_report_emailscodename%Can receive AI analysis report emails r, namecontent_type_idN) r) objects create_useradmin1admin2 regular_userr filterdeletecreate
permissionselfs Z/home/cursorai/projects/iching/iching/tests/../../tests/ai_report/test_reporting_emails.pysetUpzEmailRecipientsTests.setUp# s ll.. &
/
ll.. &
/
!LL 4 4 $
5
!!+I!JQQS %,,3338 4
c ddl m} | yzClean up test datar )cleanup_test_permissionsNtests.test_utilsrB r< rB s r= tearDownz EmailRecipientsTests.tearDownJ = "r? c < t }| j |g y)z5Test getting recipients when no users have permissionN)r assertEqualr<
recipientss r= (test_get_report_recipients_no_permissionz=EmailRecipientsTests.test_get_report_recipients_no_permissionO s *,
R(r? c R | j j j | j | j j j | j t }| j
d| | j
d| | j t | d y)z2Test getting recipients when users have permissionr r$ N) r4 user_permissionsaddr: r5 r assertInrI lenrJ s r= *test_get_report_recipients_with_permissionz?EmailRecipientsTests.test_get_report_recipients_with_permissionT sv
$$(( 9 $$(( 9*,
*J7
*J7 Z !,r? c | j j j | j t }| j d| | j
t | d y)z;Test that non-staff users are excluded even with permissionr' r N)r6 rO rP r: r assertNotInrI rR rJ s r= -test_get_report_recipients_excludes_non_staffzBEmailRecipientsTests.test_get_report_recipients_excludes_non_staff_ sM **..t?*,
+Z8 Z !,r? c | j j j | j d| j _ | j j t
}| j d| y)z%Test that inactive users are excludedFr N)r4 rO rP r: r" saver rU rJ s r= ,test_get_report_recipients_excludes_inactivezAEmailRecipientsTests.test_get_report_recipients_excludes_inactiveg sS $$(( 9 % *,
-z:r? c & | j j j | j d| j _ | j j t
}| j d| | j t | d y)z*Test that users without email are excluded r r N)
r4 rO rP r: r rX r rU rI rR rJ s r= ,test_get_report_recipients_excludes_no_emailzAEmailRecipientsTests.test_get_report_recipients_excludes_no_emailp sg $$(( 9 *,
-z: Z !,r? c t j j d j t }| j |g y)z+Test handling when permission doesn't existr* r+ N)r r2 r7 r8 r rI rJ s r= 5test_get_report_recipients_handles_missing_permissionzJEmailRecipientsTests.test_get_report_recipients_handles_missing_permissionz s= !!+I!JQQS*,
R(r? N) __name__
__module__ __qualname____doc__r> rF rL rS rV rY r\ r^ r? r= r r s. 9%
N#
)
--;-)r? r -django.core.mail.backends.locmem.EmailBackend
EMAIL_BACKENDc \ e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
e d d
Z d Z
y )
AdminNotificationEmailTestsz%Test admin notification email sendingc $ ddl m} |dddd | _ t j j dd d
d | _ t j j d t d d
d
t dd | j dddit j | _
t j j d| j t j dddddddddit j t j
| _ y)r r )$create_admin_with_report_permissionsr admin@example.comr r) r r r r 13800138000r' Test 测试用户 r. completed
bazi_analysiszTest BaZi analysis content)r0
birth_date
birth_time
created_byanalysis_status ai_analysisbazi_report_timestamp 测试问题10responsezTest LiuYao analysis)
questionuserqdatey1y2y3y4y5y6rw rx report_timestampreport_resolved_atN)rD rj adminr) r2 r3 r r r9 r r
r nowpersonr liuyao_entry)r< rj s r= r> z!AdminNotificationEmailTests.setUp s J9 %
LL,, $ -
nn++ D!Q'B{yy '(*FG"*,,. ,
#NN 1 1#,,.sssss '#%;< %\\^ '||~ 2
r? c ddl m} | yrA rC rE s r= rF z$AdminNotificationEmailTests.tearDown rG r? c t | j ddd }| j | | j t t
j d t
j d }| j |j dg | j d|j | j d|j | j d |j | j d|j | j d
|j y ) z2Test successful admin notification for BaZi reportbaziinappropriate_contentz This analysis is inappropriater. r rk u 新的AI分析举报BAZI 内容不当ro N) r r
assertTruerI rR r outboxtorQ subjectbodyr< resultr s r= )test_send_admin_notification_bazi_successzEAdminNotificationEmailTests.test_send_admin_notification_bazi_success s / KK # ,
T[[)1- A $7#89
,emm<
femm,
nejj1
6
C
nejj1r? c v t | j ddd }| j | | j t t
j d t
j d }| j d|j | j d|j | j d|j y )
z4Test successful admin notification for LiuYao reportr inaccurate_analysisz Hexagram interpretation is wrongr. r LIUYAOrz 分析不准确N)
r r r rI rR r r rQ r r r s r= +test_send_admin_notification_liuyao_successzGAdminNotificationEmailTests.test_send_admin_notification_liuyao_success s / ! .
T[[)1- A
h
.
nejj1
'4r? c | j j j t | j dddd }| j | | j
t t j d y)z4Test admin notification when no recipients availabler technical_error Test messageT)silentr N)
r rO clearr r assertFalserI rR r r r< r s r= *test_send_admin_notification_no_recipientszFAdminNotificationEmailTests.test_send_admin_notification_no_recipients s`
##))+/ KK
T[[)1-r? c t | j ddd }| j | t j d }| j d|j y)z/Test admin notification with empty user messager otherr[ r u 用户未留言Nr r r r r rQ r r s r= *test_send_admin_notification_empty_messagezFAdminNotificationEmailTests.test_send_admin_notification_empty_message sJ / KK
A
'4r? c d}t | j dd| }| j | t j d }| j ||j y)z,Test admin notification with Unicode contentu. 这是一个包含中文和emoji的消息 🚨r r r Nr )r< unicode_messager r s r= ,test_send_admin_notification_unicode_contentzHAdminNotificationEmailTests.test_send_admin_notification_unicode_content sN J/ KK #
A
ouzz2r? api.utils.send_mailc v t d |_ t | j ddd }| j | y)z&Test handling of email sending failurezSMTP server downr r r N) Exception side_effectr r r r< mock_send_mailr s r= *test_send_admin_notification_email_failurezFAdminNotificationEmailTests.test_send_admin_notification_email_failure s< &//A%B"/ KK
r? c t | j ddd }| j | t j d }| j |j | j t |d y)z4Test that both HTML and plain text versions are sentr r zTest HTML emailr alternativesN)r r r r r assertIsNotNoner hasattrr s r= *test_send_admin_notification_html_and_textzFAdminNotificationEmailTests.test_send_admin_notification_html_and_text s^ / KK #
A UZZ( ~67r? N)r_ r` ra rb r> rF r r r r r r r r rc r? r= rh rh sH /&
P#
2(5". 5
3 ! ! " !8r? rh c V e Zd ZdZd Zd Zd Zd Zd Zd Z d Z
e d d
Z y ) UserResolutionEmailTestsz(Test user resolution notification emailsc t j j dddd | _ t j j dddd d
| _ t
j j
d t d
dd | j dddid
ddd| j t j t j | _
y)r rm r' 测试 用户rl r rk r r) Tr r r r r! u 测试人员rp r. rr rs Test contentr z Test reportresolved) r0 rt rv rw rx bazi_analysis_reportedbazi_report_categorybazi_report_messagebazi_report_statusbazi_report_resolved_bybazi_report_resolved_atry N) r) r2 r3 r r r r9 r r r r r; s r= r> z UserResolutionEmailTests.setUp) s LL,, $ -
\\-- % .
nn++ D!Q'yy '(.9#'!8 - )$(JJ$,LLN"*,,. ,
r? c ddl m} | yrA rC rE s r= rF z!UserResolutionEmailTests.tearDownI rG r? c t | j | j ddd }| j | | j t t j d t j d }| j |j dg | j d|j | j d|j | j d|j | j d |j y
) z,Test successful user resolution notificationr r z:We have reviewed your report and taken appropriate action.r. r r' 举报处理结果
测试 用户zWe have reviewedN) r r r r rI rR r r r rQ r r r s r= .test_send_user_resolution_notification_successzGUserResolutionEmailTests.test_send_user_resolution_notification_successN s 2 II KK H
T[[)1- A $6#78
*EMM:
*EMM:
ouzz2
(%**5r? c d| j _ | j j t | j | j ddd }| j | | j
t t j d y)z8Test user resolution notification when user has no emailr[ r r z Admin notesr N)
r r rX r r r rI rR r r r s r= /test_send_user_resolution_notification_no_emailzHUserResolutionEmailTests.test_send_user_resolution_notification_no_emailb se 2 II KK
T[[)1-r? c t d| j ddd }| j | | j t t
j d y)z3Test user resolution notification when user is NoneNr dismisszReport dismissedr )r r r rI rR r r r s r= .test_send_user_resolution_notification_no_userzGUserResolutionEmailTests.test_send_user_resolution_notification_no_userr sH 2 KK
T[[)1-r? c g d}|D ] }t j j t | j | j
d|d| }| j
| | j t t j d t j j t | j | j
ddd }| j | | j t t j d y) z7Test user resolution notification for all admin actions)r
regeneratecontent_warningescalater zAdmin notes for r r zAdmin notes for resolvedr. N)
r r r r r r r rI rR r )r< actionsactionr s r= 2test_send_user_resolution_notification_all_actionszKUserResolutionEmailTests.test_send_user_resolution_notification_all_actions s J F KK 6 "6(+ F
V $ S -q 1
2 II KK &
T[[)1-r? c t j j d| j t j
dddddddddiddd | j t j
}t | j |d d d }| j | t j d
}| j d|j | j d|j y)z5Test user resolution notification for LiuYao analysisz
Test questionr{ r| rr r} r Tr r )r~ r r r r r r r r rw rx analysis_reportedreport_category
report_statusreport_resolved_byr r z We have regenerated the analysisr r N)r r2 r9 r r r r r r r r rQ r r )r< r r r s r= -test_send_user_resolution_notification_liuyaozFUserResolutionEmailTests.test_send_user_resolution_notification_liuyao s ~~,,$,,.sssss '#^4 " 1$ #zz '||~ -
3 II .
A
*EMM:
ouzz2r? r c t d |_ t | j | j ddd }| j | y)z0Test handling of user notification email failurezEmail service unavailabler resolvezTest admin notesN)r r r r r r r s r= 4test_send_user_resolution_notification_email_failurezMUserResolutionEmailTests.test_send_user_resolution_notification_email_failure sC &//J%K"2 II KK
r? N)
r_ r` ra rb r> rF r r r r r r r rc r? r= r r % sC 2
@#
6(. . .<3: ! ! " !r? r c e Zd ZdZd Zd Zy)EmailUtilityFunctionTestsz Test email utility functionsc ddddddd}|j D ]" \ }}t | }| j || $ y) z#Test category display name functionr r u 语言冒犯u 技术错误u 其他invalid_category)r r offensive_languager r r N)itemsr rI )r<
test_casescategoryexpectedr s r= test_get_category_displayz3EmailUtilityFunctionTests.test_get_category_display sQ &4#4"0 - 2
#-"2"2"4 Hh)(3F VX . #5r? c ddddddd}|j D ]" \ }}t | }| j || $ y) z'Test admin action display name functionu 已忽略 已解决r r r invalid_action)r r r r r r N)r r rI )r< r r r r s r= test_get_admin_action_displayz7EmailUtilityFunctionTests.test_get_admin_action_display sQ #"& 0" .
!+ 0 0 2 FH-f5F VX . !3r? N)r_ r` ra rb r r rc r? r= r r s &
/
/r? r c e Zd ZdZd Zd Z ed d Z ed d Z ed d Z ed d Z
y
) EmailTemplateTestsz)Test email template rendering and contentc t j j dddd | _ t j j ddddd
| _ t
j j
d j t
j j d d
d }| j j j | y)r rm r' r r rl r rk u 管理员Tr r* r+ r- r. r/ N) r) r2 r3 r r r r7 r8 r9 rO rP )r< r: s r= r> zEmailTemplateTests.setUp s LL,, $ -
\\-- %" .
!!+I!JQQS ''..38 /
##''
3r? c ddl m} | yrA rC rE s r= rF zEmailTemplateTests.tearDown rG r? rd re c X t j j dt ddd d| j dddit j t j }t |d
d d t j d
}g d}|D ] }| j ||j y)z?Test that admin email template receives all necessary variables 模板测试rp Frr rs zTemplate test content)r0 rt genderrv rw rx ry r r r 测试模板变量渲染r )r r r r z May 15, 1990r r' N r r2 r9 r r r r r r r rQ r r< r r expected_contentcontents r= #test_admin_email_template_variablesz6EmailTemplateTests.test_admin_email_template_variables s && D!R(yy '(*AB"*,,.$,LLN '
' # &
A
(G MM'5:: . (r? c t j j dt ddd | j dddiddt j t j | j
}t | j |d
d d t j d
}g d}|D ] }| j ||j y)zITest that user resolution email template receives all necessary variables 解决测试rp r. rr rs zResolution testTr )
r0 rt rv rw rx r r ry r r r r ' 我们已经解决了您报告的问题r )r u 八字分析r r r N)
r r2 r9 r r r r r r r r rQ r r s r= -test_user_resolution_email_template_variablesz@EmailTemplateTests.test_user_resolution_email_template_variables5 s && D!Q'yy '(*;<#'!2"*,,.$,LLN$(JJ '
* II 5
A
(G MM'5:: . (r? c
t j j dt ddd | j dddit j t j }d}t |d d
| t j d }| j d|j | j ||j | j d |j y
)z4Test proper encoding of Chinese characters in emailsu 中文测试用户名 🎯rp r. rr rs u 中文分析内容r0 rt rv rw rx ry r uP 这是一个包含中文字符、标点符号！和emoji 🚨 的测试消息。r r r r Nr )r< r chinese_messager s r= &test_email_encoding_chinese_charactersz9EmailTemplateTests.test_email_encoding_chinese_charactersZ s &&-D!Q'yy '(*>?"*,,.$,LLN '
m& #
A
2EJJ?
ouzz2
nejj1r? c
D t j j dt ddd | j dddit j t j }d}t |d d
| t j d }| j d |j y
)z4Test that HTML content is properly handled in emailsz HTML Testrp r. rr rs zHTML test contentr z-Bold textr r r zO<script>alert("XSS")</script><b>Bold text</b>Nr )r< r html_messager s r= test_email_html_content_safetyz1EmailTemplateTests.test_email_html_content_safetyu s && D!Q'yy '(*=>"*,,.$,LLN '
G &
A
ginisistr? N) r_ r` ra rb r> rF r r r r r rc r? r= r r s 348#
%TU"/ V"/H %TU"/ V"/H %TU2 V24 %TUu Vur? r c D e Zd ZdZd Zd Z ed d Zd Zd Z y) EmailEdgeCaseTestsz4Test edge cases and error scenarios for email systemc P t j j dd | _ y)r rm r' )r r N)r) r2 r3 r r; s r= r> zEmailEdgeCaseTests.setUp s# LL,, $ -
r? c ddl m} | yrA rC rE s r= rF zEmailEdgeCaseTests.tearDown rG r? zapi.utils.loggerc t j j ddddd }t j j d j t j j
dd d
}|j j | t j j
d t d
d
d
| j dddit j }t dt d 5 t! |ddd }| j# | |j$ j' ddd y# 1 sw Y yxY w)z,Test that email failures are properly loggedr rk r r) Tr r* r+ r- r. r/ z Logging Testrp rr testdata)r0 rt rv rw rx ry r z
SMTP Error)r r r z Logging testN)r) r2 r3 r r7 r8 r9 rO rP r r r r r r r r r error
assert_called)r< mock_loggerr r: r r s r= test_email_logging_on_failurez0EmailEdgeCaseTests.test_email_logging_on_failure s# (( % )
!!+I!JQQS ''..38 /
"":. && D!Q'yy ' ("*,,.
'
(i 6M
N3! F
V $ + + - O
N
Ns 4:D77E c B t dddd }| j | y)z,Test email sending with None analysis objectNr r zNone object test)r r r s r= $test_email_with_none_analysis_objectz7EmailEdgeCaseTests.test_email_with_none_analysis_object s( /
r? c G d d } | } t |ddd }y# t $ r }| j d| Y d}~yd}~ww xY w)z>Test email sending when analysis object has missing attributesc e Zd Zd Zy)KEmailEdgeCaseTests.test_email_with_missing_attributes..MockAnalysisc d| _ d| _ y )Nr. z
Mock Analysis)idr0 r; s r= __init__zTEmailEdgeCaseTests.test_email_with_missing_attributes..MockAnalysis.__init__ s + r? N)r_ r` ra r rc r? r= MockAnalysisr s
,r? r r r zMissing attributes testz/Email sending crashed with missing attributes: N)r r fail)r< r mock_objr es r= "test_email_with_missing_attributesz5EmailEdgeCaseTests.test_email_with_missing_attributes s\ , , > M3!) F M IIGsK L L Ms " A AA N)
r_ r` ra rb r> rF r r r r# rc r? r= r
r
s5 >
#
'. '.R
!Mr? r
N)$rb django.testr r django.contrib.authr django.contrib.auth.modelsr django.corer django.utilsr datetimer r
unittest.mockr r r
bazi.modelsr
liuyao.modelsr api.utilsr r r r r r) r rh r r r r
rc r? r= r. s 4 . 1 ! 0 0 `)8 `)F !PQ^8( ^8 R^8B !PQa!x a! Ra!H / /Dau auHZM ZMr?