1# META: timeout=long
2
3import pytest
4
5from tests.support.asserts import assert_dialog_handled, assert_error, assert_success
6from tests.support.helpers import document_hidden
7
8
9def minimize(session):
10    return session.transport.send(
11        "POST", "session/{session_id}/window/minimize".format(**vars(session)))
12
13
14@pytest.fixture
15def check_user_prompt_closed_without_exception(session, create_dialog):
16    def check_user_prompt_closed_without_exception(dialog_type, retval):
17        assert not document_hidden(session)
18        create_dialog(dialog_type, text=dialog_type)
19
20        response = minimize(session)
21        assert_success(response)
22
23        assert_dialog_handled(session, expected_text=dialog_type, expected_retval=retval)
24        assert document_hidden(session)
25
26    return check_user_prompt_closed_without_exception
27
28
29@pytest.fixture
30def check_user_prompt_closed_with_exception(session, create_dialog):
31    def check_user_prompt_closed_with_exception(dialog_type, retval):
32        assert not document_hidden(session)
33        create_dialog(dialog_type, text=dialog_type)
34
35        response = minimize(session)
36        assert_error(response, "unexpected alert open")
37
38        assert_dialog_handled(session, expected_text=dialog_type, expected_retval=retval)
39        assert not document_hidden(session)
40
41    return check_user_prompt_closed_with_exception
42
43
44@pytest.fixture
45def check_user_prompt_not_closed_but_exception(session, create_dialog):
46    def check_user_prompt_not_closed_but_exception(dialog_type):
47        assert not document_hidden(session)
48        create_dialog(dialog_type, text=dialog_type)
49
50        response = minimize(session)
51        assert_error(response, "unexpected alert open")
52
53        assert session.alert.text == dialog_type
54        session.alert.dismiss()
55
56        assert not document_hidden(session)
57
58    return check_user_prompt_not_closed_but_exception
59
60
61@pytest.mark.capabilities({"unhandledPromptBehavior": "accept"})
62@pytest.mark.parametrize("dialog_type, retval", [
63    ("alert", None),
64    ("confirm", True),
65    ("prompt", ""),
66])
67def test_accept(check_user_prompt_closed_without_exception, dialog_type, retval):
68    check_user_prompt_closed_without_exception(dialog_type, retval)
69
70
71@pytest.mark.capabilities({"unhandledPromptBehavior": "accept and notify"})
72@pytest.mark.parametrize("dialog_type, retval", [
73    ("alert", None),
74    ("confirm", True),
75    ("prompt", ""),
76])
77def test_accept_and_notify(check_user_prompt_closed_with_exception, dialog_type, retval):
78    check_user_prompt_closed_with_exception(dialog_type, retval)
79
80
81@pytest.mark.capabilities({"unhandledPromptBehavior": "dismiss"})
82@pytest.mark.parametrize("dialog_type, retval", [
83    ("alert", None),
84    ("confirm", False),
85    ("prompt", None),
86])
87def test_dismiss(check_user_prompt_closed_without_exception, dialog_type, retval):
88    check_user_prompt_closed_without_exception(dialog_type, retval)
89
90
91@pytest.mark.capabilities({"unhandledPromptBehavior": "dismiss and notify"})
92@pytest.mark.parametrize("dialog_type, retval", [
93    ("alert", None),
94    ("confirm", False),
95    ("prompt", None),
96])
97def test_dismiss_and_notify(check_user_prompt_closed_with_exception, dialog_type, retval):
98    check_user_prompt_closed_with_exception(dialog_type, retval)
99
100
101@pytest.mark.capabilities({"unhandledPromptBehavior": "ignore"})
102@pytest.mark.parametrize("dialog_type", ["alert", "confirm", "prompt"])
103def test_ignore(check_user_prompt_not_closed_but_exception, dialog_type):
104    check_user_prompt_not_closed_but_exception(dialog_type)
105
106
107@pytest.mark.parametrize("dialog_type, retval", [
108    ("alert", None),
109    ("confirm", False),
110    ("prompt", None),
111])
112def test_default(check_user_prompt_closed_with_exception, dialog_type, retval):
113    check_user_prompt_closed_with_exception(dialog_type, retval)
114