a
    cl_u                     @   s0   d Z ddlmZ ddlmZ G dd deZdS )zTests for samba.upgrade.    )import_wins)LdbTestCasec                   @   s   e Zd Zdd Zdd ZdS )WinsUpgradeTestsc                 C   sD   ddddgdfi}t | j| | dgdd | jjd	d
D  d S )NzFOO#20   z	127.0.0.1z	127.0.0.2`   zname=FOO,type=0x20c                 S   s   g | ]}t |jqS  )strZdn).0mr   r   5/usr/lib/python3/dist-packages/samba/tests/upgrade.py
<listcomp>"       z1WinsUpgradeTests.test_upgrade.<locals>.<listcomp>z(objectClass=winsRecord)Z
expression)r   ldbassertEqualsearch)selfZwinsdbr   r   r   test_upgrade   s    zWinsUpgradeTests.test_upgradec                 C   s2   t | ji  | dt| jjddd d  d S )NZVERSIONz(objectClass=winsMaxVersion)r   r   Zcn)r   r   r   r   r   )r   r   r   r   test_version%   s    zWinsUpgradeTests.test_versionN)__name__
__module____qualname__r   r   r   r   r   r   r      s   r   N)__doc__Zsamba.upgrader   Zsamba.testsr   r   r   r   r   r   <module>   s   