
    [i                         d Z ddlZddlmc mZ ddlZddlZej                  j                  dd       ddlmZmZ d Zd Zej                   j"                  d        Zy)u   企业微信集成测试    Nz!/home/jang/Projects/ai-researcher)WeChatWorkClientWeChatWorkMessageHandlerc                  >   t               } d}| j                  |      }|d   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}|d
   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}|d   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}|d   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}|d   }d}||k(  }|slt        j                  d|fd||f      t        j                  |      t        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}x}}y	)u   测试解析文本消息u  <xml>
<ToUserName><![CDATA[corp_id]]></ToUserName>
<FromUserName><![CDATA[user123]]></FromUserName>
<CreateTime>123456789</CreateTime>
<MsgType><![CDATA[text]]></MsgType>
<Content><![CDATA[你好]]></Content>
<MsgId>123456</MsgId>
<AgentID>1000002</AgentID>
</xml>to_user_namecorp_id)==)z%(py1)s == %(py4)s)py1py4zassert %(py6)spy6Nfrom_user_nameuser123msg_typetextcontentu   你好msg_id123456)r   parse_message
@pytest_ar_call_reprcompare	_safereprAssertionError_format_explanation)clientxmlresult@py_assert0@py_assert3@py_assert2@py_format5@py_format7s           @/home/jang/Projects/ai-researcher/tests/unit/test_wechat_work.pytest_parse_text_messager"   	   s   F
C !!#&F.!.Y.!Y....!Y...!...Y......."#0y0#y0000#y000#000y0000000*''''''''''''''''''')(((((((((((((((((((('x'x''''x''''''x'''''''    c                  f   t               } | j                  ddd      }d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }dd	|iz  }t        t        j                  |            d
x}}d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }dd	|iz  }t        t        j                  |            d
x}}d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }dd	|iz  }t        t        j                  |            d
x}}d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }dd	|iz  }t        t        j                  |            d
x}}y
)u   测试构建回复r   r   u   回复消息inz%(py1)s in %(py3)sr   r	   py3assert %(py5)spy5Nz#<MsgType><![CDATA[text]]></MsgType>)
r   build_replyr   r   r   @py_builtinslocals_should_repr_global_namer   r   )r   r   r   r   @py_format4@py_format6s         r!   test_build_replyr2      s   F


Y	>
BC999999 >S    >S   >      S   S       070C77770C7770777777C777C7777777r#   c                  `  K   t               } t        |       }dddddddd}|j                  |       d	{   }d	}||u}|st        j                  d
|fd||f      dt        j                         v st        j                  |      rt        j                  |      ndt        j                  |      dz  }dd|iz  }t        t        j                  |            d	x}}d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }dd|iz  }t        t        j                  |            d	x}}d}||v }|st        j                  d|fd||f      t        j                  |      dt        j                         v st        j                  |      rt        j                  |      nddz  }dd|iz  }t        t        j                  |            d	x}}y	7 w)u   测试处理文本消息r   r   	123456789r   u   查询技术部员工r   1000002)r   r   create_timer   r   r   agent_idN)is not)z%(py0)s is not %(py3)sreply)py0r)   r*   r+   u   收到消息r%   r'   r(   )r   r   handle_messager   r   r-   r.   r/   r   r   r   )	r   handlermessager9   r   @py_assert1r0   r1   r   s	            r!   test_handle_text_messager?   *   s`     F&v.G "#"*G ((11E5555">U"""">U""">""""""U"""U""""""""+"e++++"e+++"++++++e+++e+++++++	 2s   4H.H+G5H.)__doc__builtinsr-   _pytest.assertion.rewrite	assertionrewriter   pytestsyspathinsertsrc.integrations.wechat_workr   r   r"   r2   markasyncior?    r#   r!   <module>rM      sP        
 6 7 S(,8 , ,r#   