U
    {®Þh  ã                   @   sB   d dl mZmZmZ d dlmZ d dlmZ G dd„ deeƒZdS )é    )ÚRequestFactoryÚTestCaseÚoverride_settings)ÚUpgradeNotificationPanel)ÚWagtailTestUtilsc                   @   sâ   e Zd ZdZdZedd„ ƒZdd„ Zeddd	d
„ ƒZ	edddd„ ƒZ
edddd„ ƒZdd„ Zedddd„ ƒZedddd„ ƒZedddd„ ƒZdd„ Zdd„ Zedddd„ ƒZedddd „ ƒZeddd!d"„ ƒZd#S )$ÚTestUpgradeNotificationPanelzdata-w-upgradezdata-w-upgrade-lts-onlyc                 C   s>   t ƒ | _tƒ | _| jdd| _| jdd| _| j d¡| _	d S )NÚtester)ÚusernameZsupertesterú/)
r   Úpanelr   Zrequest_factoryZcreate_userÚuserZcreate_superuserÚ	superuserÚgetÚrequest)Úcls© r   úu/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/test_upgrade_notification.pyÚsetUpTestData   s
    z*TestUpgradeNotificationPanel.setUpTestDatac                 C   s   |   | j ¡ ¡ d S ©N)Ú
assertTruer   Úget_upgrade_check_setting©Úselfr   r   r   Ú&test_get_upgrade_check_setting_default   s    zCTestUpgradeNotificationPanel.test_get_upgrade_check_setting_defaultF)ZWAGTAIL_ENABLE_UPDATE_CHECKc                 C   s   |   | j ¡ ¡ d S r   )ÚassertFalser   r   r   r   r   r   Ú$test_get_upgrade_check_setting_false   s    zATestUpgradeNotificationPanel.test_get_upgrade_check_setting_falseÚLTSc                 C   s   |   | j ¡ d¡ d S )Nr   ©ÚassertEqualr   r   r   r   r   r   Ú"test_get_upgrade_check_setting_LTS   s    z?TestUpgradeNotificationPanel.test_get_upgrade_check_setting_LTSÚltsc                 C   s   |   | j ¡ d¡ d S )Nr    r   r   r   r   r   Ú"test_get_upgrade_check_setting_lts   s    z?TestUpgradeNotificationPanel.test_get_upgrade_check_setting_ltsc                 C   s   |   | j ¡ ¡ d S r   ©r   r   Úupgrade_check_lts_onlyr   r   r   r   Ú#test_upgrade_check_lts_only_default"   s    z@TestUpgradeNotificationPanel.test_upgrade_check_lts_only_defaultc                 C   s   |   | j ¡ ¡ d S r   r"   r   r   r   r   Ú(test_upgrade_check_lts_only_setting_true%   s    zETestUpgradeNotificationPanel.test_upgrade_check_lts_only_setting_truec                 C   s   |   | j ¡ ¡ d S r   ©r   r   r#   r   r   r   r   Ú'test_upgrade_check_lts_only_setting_LTS)   s    zDTestUpgradeNotificationPanel.test_upgrade_check_lts_only_setting_LTSc                 C   s   |   | j ¡ ¡ d S r   r&   r   r   r   r   Ú'test_upgrade_check_lts_only_setting_lts-   s    zDTestUpgradeNotificationPanel.test_upgrade_check_lts_only_setting_ltsc                 C   s0   | j | j_ d| ji}| j |¡}|  |d¡ d S ©Nr   Ú )r   r   r   Úrender_htmlr   ©r   Zparent_contextÚresultr   r   r   Útest_render_html_normal_user1   s    

z9TestUpgradeNotificationPanel.test_render_html_normal_userc                 C   s@   | j | j_d| ji}| j |¡}|  | j|¡ |  | j|¡ d S ©Nr   )	r   r   r   r   r+   ÚassertInÚDATA_ATTRIBUTE_UPGRADE_CHECKZassertNotInÚ DATA_ATTRIBUTE_UPGRADE_CHECK_LTSr,   r   r   r   Útest_render_html_superuser9   s
    

z7TestUpgradeNotificationPanel.test_render_html_superuserc                 C   s0   | j | j_d| ji}| j |¡}|  |d¡ d S r)   )r   r   r   r   r+   r   r,   r   r   r   Útest_render_html_setting_falseB   s    

z;TestUpgradeNotificationPanel.test_render_html_setting_falsec                 C   s@   | j | j_d| ji}| j |¡}|  | j|¡ |  | j|¡ d S r/   ©r   r   r   r   r+   r0   r1   r2   r,   r   r   r   Útest_render_html_setting_LTSK   s
    

z9TestUpgradeNotificationPanel.test_render_html_setting_LTSc                 C   s@   | j | j_d| ji}| j |¡}|  | j|¡ |  | j|¡ d S r/   r5   r,   r   r   r   Útest_render_html_setting_ltsU   s
    

z9TestUpgradeNotificationPanel.test_render_html_setting_ltsN)Ú__name__Ú
__module__Ú__qualname__r1   r2   Úclassmethodr   r   r   r   r   r!   r$   r%   r'   r(   r.   r3   r4   r6   r7   r   r   r   r   r      s4   






	

	r   N)	Zdjango.testr   r   r   Zwagtail.admin.views.homer   Zwagtail.test.utilsr   r   r   r   r   r   Ú<module>   s   