Sindbad~EG File Manager
�
'Μg� � � � d dl Z d dlZd dlmZmZmZ ej ddg�� Zd� Z G d� dej � Z
y) � N)�
import_helper�requires_resource�LOOPBACK_TIMEOUT�_wmi�win)� required_onc �� � t j | � S # t $ r Y n%t $ r}|j dk7 r� Y d }~nd }~ww xY wt j t � t j | � S )Ni )r �
exec_query�BrokenPipeError� WindowsError�winerror�time�sleepr )�query�es �*/usr/local/lib/python3.12/test/test_wmi.py�wmi_exec_queryr
sc � �����u�%�%�� �
� � � � �:�:�� � � ���� �J�J� � � �?�?�5� !�!s � � A�A�?�Ac �R � e Zd Zd� Zd� Zd� Zd� Zd� Z ed� d� � Z d� Z
d � Z y
) �WmiTestsc �" � t d� j d� }| j dt |� � |d j d� \ }}}| j d||d � | j d||d � | j |d|d � y )Nz)SELECT Version FROM Win32_OperatingSystem� � r �=�Versionz \d+\.\d+.+$)r �split� assertEqual�len� partition� assertRegex)�self�r�k�eq�vs r �test_wmi_query_os_versionz"WmiTests.test_wmi_query_os_version s� � � �F� G� M� M�d� S�� ����C��F�#��Q�4�>�>�#�&���2�q� ����b�!�A�$�'� ����A�q��t�,�
����N�A�a�D�1� c �D � t d� D ] }| j � � y �N�
)�ranger% �r �_s r �test_wmi_query_repeatedz WmiTests.test_wmi_query_repeated% s � ��r��A� � *� *� ,� r& c � � t d� | j d� y # t $ r!}|j dz dk( rY d }~y Y d }~�7d }~ww xY w)Nz.SELECT InvalidColumnName FROM InvalidTableNamel �� l zExpected OSError)r �OSErrorr
�fail)r �exs r �test_wmi_query_errorz WmiTests.test_wmi_query_error* sJ � � � �K� L�
� � �$�%�� � ��{�{�Z�'�:�5�� 6�� �s � � A �A�A c �D � t d� D ] }| j � � y r( )r* r2 r+ s r � test_wmi_query_repeated_errorz&WmiTests.test_wmi_query_repeated_error4 s � ��r��A� � %� %� '� r& c �n � | j t � 5 t d� d d d � y # 1 sw Y y xY w)Nz0not select, just in case someone tries something)� assertRaises�
ValueErrorr )r s r �test_wmi_query_not_selectz"WmiTests.test_wmi_query_not_select8 s% � �
�
�
�z�
*� �M� N� +�
*�
*�s � +�4�cpuc � � t d� D ]+ }| j t � 5 t d� d d d � �- y # 1 sw Y �8xY w)N� zSELECT * FROM CIM_DataFile)r* r6 r/ r r+ s r �test_wmi_query_overflowz WmiTests.test_wmi_query_overflow= s8 � � �q��A��"�"�7�+� � ;�<� ,�+� �+�+�s � ;�A c �` � t d� }| j |j d� |� | j |j d� |� t |j d� � } | j
t |� d� | j dt |� � �7# t $ r Y y w xY w)N�:SELECT ProcessId FROM Win32_Process WHERE ProcessId < 1000r z
ProcessId=\d+� )
r � assertFalse�
startswith�endswith�iterr r �nextr �
StopIteration)r r! �its r � test_wmi_query_multiple_rowsz%WmiTests.test_wmi_query_multiple_rowsE s� � � �W� X�� ���� � �d�+�Q�/� ������D�)�1�-�
�!�'�'�$�-�
�� ��� � ��b��+;�<�� � ��T�"�X�.� �� � � � �s �)8B! �! B-�,B-c � � ddl m} d} |d� 5 }t d� D �cg c] }|j t |� �� }}|D ]" }| j |j
� d� �$ d d d � y c c}w # 1 sw Y y xY w)Nr )�ThreadPoolExecutorr> � � z
ProcessId=)�concurrent.futuresrI r* �submitr r �result)r rI r �poolr, �task�ts r �test_wmi_query_threadsz WmiTests.test_wmi_query_threadsR sk � �9�L��
��
"�d�@E�b� �J� �1�D�K�K���6� �D�J���� � �����\�:� � #�
"��J� #�
"�s �A4� A/�)A4�/A4�4A=N) �__name__�
__module__� __qualname__r% r- r2 r4 r8 r r< rG rR � r& r r r s? � �2�-�
&�(�O�
�u� �=� �=� �;r& r ) r �unittest� test.supportr r r �
import_moduler r �TestCaser rV r& r �<module>r[ sB �� � � K� K� #�}�"�"�6��w�?��
"�>;�x� � � >;r&
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists