KVM MCP Server

� �h����
�SS
Kr
SS
KJs Jr SS
KrSSKJrJ r SSK JrJrJ r Jr \RR S5r\RR S5r\RR S5r\RR S5r\RR S5r\RR S 5r\RR S 5rg
)�N)�patch� MagicMock)� create_vm�start_vm�stop_vm� reboot_vmc��# �[
5n[
5n
S
URlXRlSU
Rl[SUS9 [S[
SS9
S9 [ SSS S SSS .5IS
hv
�NnUSnSoCU:HoU(do[R"SU4
SX445[R"U5
[R"U5
S.-nSSU0
-n[[R"U5
5
e
S
=
n=
pTSo2SoCU;oU(do[R"SU4
SX445[R"U5
[R"U5
S.-nSSU0
-n[[R"U5
5
e
S
=
n=
pTS
S
S
5 S
S
S
5 g
G
N!,(d
f  N=f
!,(d
f  g
=f
7f
)zTest successful VM creationNr�libvirt.open�
�return_valuezsubprocess.run)
� returncoder�test-vm���z/path/to/image.qcow2)�name�memory�vcpus� disk_size�master_image�status�success�
�==�
z%(py1)s == %(py4)s��py1�py4�assert %(py6)s�py6�message�
�in�
z%(py1)s in %(py4)s)r�lookupByNamer� defineXML�createrr� @pytest_ar�_call_reprcompare� _saferepr�AssertionError�_format_explanation�� mock_conn�mock_domain�result�@py_assert0�@py_assert3�@py_assert2�@py_format5�@py_format7s �7/home/steve/projects/kvm-mcp/tests/test_vm_lifecycle.py�test_create_vm_successr7sA
�����I��+�K�*.�I���'�'2���$�&'�K���#� �~�I� 6� ��i�1�.E� F� ������2�/ �����h��,�9�,�9�,�,�,�,�,��,�,�,��,�,�,�9�,�,�,�,�,�,�,��-�9�-�-�-�-�-�-�-�-�y�-�-�-�y�-�-�-�-�-�-�-�-�-�-�-� G
� 7� 6�� G
� F�� 7� 6�sI�AF?
�F.�%F�;
F�<D F� F.� F?
�
F� F+ �'F.�. F<�8F?
c���# �[
5n[
5n
XRl[S
US9 [ SSSSS.5IShv
�NnUS nS oCU:HoU(do[ R"SU4
SX445[ R"U5
[ R"U5
S .-nSSU0
-n[[ R"U5
5
e
S=
n=
pTSo2SoCU;oU(do[ R"SU4
SX445[ R"U5
[ R"U5
S .-nSSU0
-n[[ R"U5
5
e
S=
n=
pTSSS5 gG
N!,(d
f  g=f
7f
)z'Test VM creation when VM already existsr rrrrr)rrrNr�errorrrrrr zalready existsr!r"r$) rr%rrrr(r)r*r+r,r-s r6�test_create_vm_already_existsr:s

�����I��+�K�*5���'� �~�I� 6� �����/ �����h��*�7�*�7�*�*�*�*�*��*�*�*��*�*�*�7�*�*�*�*�*�*�*��4�)�#4�4�#4�4�4�4�4�4��4�4�4��4�4�4�#4�4�4�4�4�4�4�4� 7� 6�� 7� 6�s/�/E0
�E�
E�D E� E0
�
E� E-�)E0
c��# �[
5n[
5n
XRlS
U
Rl[ SUS9 [SSS0
5IShv
�NnUSnS oCU:HoU(do[R"S U4
SX445[R"U5
[R"U5
S.-nS SU0
-n[[R"U5
5
e
S=
n=
pTSo2SoCU;oU(do[R"SU4
SX445[R"U5
[R"U5
S.-nS SU0
-n[[R"U5
5
e
S=
n=
pTSSS5 gG
N!,(d
f  g=f
7f
)zTest successful VM startrr rrrrNrrrrrrr zstarted successfullyr!r"r$)rr%rr'rrr(r)r*r+r,r-s r6�test_start_vm_successr<.s
�����I��+�K�*5���'�&'�K���#� �~�I� 6�� �V�Y�,?�@�@���h��,�9�,�9�,�,�,�,�,��,�,�,��,�,�,�9�,�,�,�,�,�,�,�%�:� �):�:�):�:�:�:�:�:�%�:�:�:�%�:�:�:�):�:�:�:�:�:�:�:� 7� 6�@� 7� 6��1�AE?
�E.�
E+�D E.�" E?
�+
E.�. E<�8E?
c��# �[
5n[
5n
XRlS
U
Rl[ SUS9 [SSS0
5IShv
�NnUSnS oCU:HoU(do[R"S U4
SX445[R"U5
[R"U5
S.-nS SU0
-n[[R"U5
5
e
S=
n=
pTSo2SoCU;oU(do[R"SU4
SX445[R"U5
[R"U5
S.-nS SU0
-n[[R"U5
5
e
S=
n=
pTSSS5 gG
N!,(d
f  g=f
7f
)zTest successful VM stoprr rrrrNrrrrrrr zstopped successfullyr!r"r$)rr%r�shutdownrrr(r)r*r+r,r-s r6�test_stop_vm_successr@=s
�����I��+�K�*5���'�()�K���%� �~�I� 6��y�6�9�*=�>�>���h��,�9�,�9�,�,�,�,�,��,�,�,��,�,�,�9�,�,�,�,�,�,�,�%�:� �):�:�):�:�:�:�:�:�%�:�:�:�%�:�:�:�):�:�:�:�:�:�:�:� 7� 6�>� 7� 6�r=c��# �[
5n[
5n
XRlS
U
Rl[ SUS9 [SSS0
5IShv
�NnUSnS oCU:HoU(do[R"S U4
SX445[R"U5
[R"U5
S.-nS SU0
-n[[R"U5
5
e
S=
n=
pTSo2SoCU;oU(do[R"SU4
SX445[R"U5
[R"U5
S.-nS SU0
-n[[R"U5
5
e
S=
n=
pTSSS5 gG
N!,(d
f  g=f
7f
)zTest successful VM rebootrr rrrrNrrrrrrr zrebooted successfullyr!r"r$)rr%r�rebootrrr(r)r*r+r,r-s r6�test_reboot_vm_successrCLs
�����I��+�K�*5���'�&'�K���#� �~�I� 6� ��v�y�.A�B�B���h��,�9�,�9�,�,�,�,�,��,�,�,��,�,�,�9�,�,�,�,�,�,�,�&�;��*;�;�*;�;�;�;�;�;�&�;�;�;�&�;�;�;�*;�;�;�;�;�;�;�;� 7� 6�B� 7� 6�r=c��z# �[[[4G
HnU"S
05IShv
�Nn
U
SnSo2U:HoD(do[R"SU4
SX#45[R "U5
[R "U5
S.-nSS U0
-n[ [R"U5
5
e
S=
n=
pCS o!So2U;oD(do[R"SU4
S X#45[R "U5
[R "U5
S.-nSS U0
-n[ [R"U5
5
e
S=
n=
pCG
M! gG
N7f
)z$Test VM operations with invalid name� operationNrr9rrrrr zname not providedr!r"r$)rrrr(r)r*r+r,�rEr0r1r2r3r4r5s r6�test_vm_operations_invalid_namerG[s�������3� � ��b�1�1���h��*�7�*�7�*�*�*�*�*��*�*�*��*�*�*�7�*�*�*�*�*�*�*�"�7�Y�&7�7�&7�7�7�7�7�7�"�7�7�7�"�7�7�7�&7�7�7�7�7�7�7�7�4�1�s�"D;
�
D8�DD;
c ���# �[
S
SS9 [[[4G
H nU"SSS0
5IShv
�Nn
U
SnSo2U:HoD(do[R "S U4
S X#45[R"U5
[R"U5
S.-nSS U0
-n[[R"U5
5
e
S=
n=
pCSo!So2U;oD(do[R "SU4
SX#45[R"U5
[R"U5
S.-nSS U0
-n[[R"U5
5
e
S=
n=
pCG
M# SSS5 gG
N!,(d
f  g=f
7f
)z(Test VM operations with connection errorr NrrErrrr9rrrrr zFailed to connectr!r"r$) rrrrr(r)r*r+r,rFs r6�#test_vm_operations_connection_errorrIcs���� �~�D� 1�"�G�Y�7�I�$�[�6�9�2E�F�F�F��(�#�.�w�.�w�.�.�.�.�.�#�.�.�.�#�.�.�.�w�.�.�.�.�.�.�.�&�;��*;�;�*;�;�;�;�;�;�&�;�;�;�&�;�;�;�*;�;�;�;�;�;�;�;�8� 2� 1�F� 2� 1�s-�E!
�$E�
E �DE� E!
�
E� E�E!
)�builtins�@py_builtins�_pytest.assertion.rewrite� assertion�rewriter(�pytest� unittest.mockrr�kvm_mcp_serverrrrr�mark�asyncior7r:r<r@rCrGrI��r6�<module>rV
s��


��� �*�B�B�
���
�
�
.��
.�,���
�
�
5��
5�"���
�
�
;��
;����
�
�
;��
;����
�
�
<��
<����
�
�
8��
8����
�
�
<��
<rU