diff options
Diffstat (limited to 'test/unit/test_popup.py')
-rw-r--r-- | test/unit/test_popup.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/test/unit/test_popup.py b/test/unit/test_popup.py index 5319d72..bc53e6c 100644 --- a/test/unit/test_popup.py +++ b/test/unit/test_popup.py @@ -62,6 +62,10 @@ mocked_page_infos = { **unprivileged_page_info, 'mapping': 'm1', 'payload': {'identifier': 'res1'} + }, + 'error': { + **unprivileged_page_info, + 'error': True } } @@ -143,7 +147,7 @@ def test_popup_display(driver, execute_in_page, page_info_key): assert by_id['page_url'].text == mocked_page_infos[page_info_key]['url'] assert not by_id['repo_query_container'].is_displayed() - if 'blocked' in page_info_key or page_info_key == 'mapping': + if 'blocked' in page_info_key or page_info_key in ('mapping', 'error'): assert by_id['scripts_blocked'].text.lower() == 'yes' elif 'allowed' in page_info_key: assert by_id['scripts_blocked'].text.lower() == 'no' @@ -167,6 +171,9 @@ def test_popup_display(driver, execute_in_page, page_info_key): elif 'default' in page_info_key: 'by default_policy)' in mapping_text + if page_info_key == 'error': + assert mapping_text == 'None (error occured when determining policy)' + @pytest.mark.ext_data(popup_ext_data) @pytest.mark.usefixtures('webextension') def test_popup_repo_query(driver, execute_in_page): |