
    bi                     ~    d dl Z d dlZd dlmZmZ d dlmZ ej                  d        Zej                  d        Z	d Z
y)    N)	MagicMockpatch)CPUResourceWebSocketHandlerc                 >    t               }| j                  d|       |S )NzDjupyterlab_nvdashboard.apps.cpu.CustomWebSocketHandler.write_message)r   setattr)monkeypatchmocks     i/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jupyterlab_nvdashboard/tests/test_cpu_handlers.pymock_handlerr      s$    ;DN K    c               #      K   t        d      5 } t        d      5 }ddi}|| _        | |f d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   y xY ww)Nztornado.web.Applicationz"tornado.httputil.HTTPServerRequestbase_url/)r   settings)mock_applicationmock_requestmock_settingss      r
   handler_argsr      sm      	'(	-,<23	-7C 
 %2!,,	- 	- 	- 	- 	- 	-s/   AA	=A		AA	A		AAc                     t        | }|j                          | j                  \  }}t        j                  |d         }d|v sJ d|v sJ d|v sJ d|v sJ d|v sJ d|v sJ d|v sJ y )	Nr   timecpu_utilizationmemory_usage	disk_read
disk_writenetwork_readnetwork_write)r   	send_data	call_argsjsonloads)r   r   handlerargs_datas         r
   test_cpu_resource_handlerr%       s    )<8G$$GD!::d1gDT>>$$$T!!!$4T!!!d"""r   )r   pytestunittest.mockr   r   jupyterlab_nvdashboard.apps.cpur   fixturer   r   r%    r   r
   <module>r+      sF      * G   
- 
-#r   