o
    iv                     @   s6   d dl Z d dl mZ d dlmZ G dd de jZdS )    N)mock)logc                   @   s   e Zd Zeddd ZdS )TestLogRequestz"notebook.log.prometheus_log_methodc                 C   s   ddi}t jt jdd|dddd}t j|t jd	dd
}t  }tj||dd | d|jj |jt j	dt
t
|ddt j	dt
| |j|j|jd|d dddg || d S )NReferertest   )return_valueGETz1.2.3.4z/notebooks/foo/bar)request_timeheadersmethod	remote_ipurii  )request
get_statusT)r   log_json    )props)extrag     @@)statusr   ipr   r
   referer)r   Mock	MagicMockr   log_requestassertEqualerror
call_countassert_has_callscalldictr   r   r   r   assert_called_once_with)selfmock_prometheusr   r   handlerlogger r'   R/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/tests/test_log.pytest_log_request_json	   s6   

z$TestLogRequest.test_log_request_jsonN)__name__
__module____qualname__r   patchr)   r'   r'   r'   r(   r      s    r   )unittestr   notebookr   TestCaser   r'   r'   r'   r(   <module>   s    