1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// Copyright (c) 2010 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/net/url_request_mock_http_job.h"
#include "chrome/test/automation/tab_proxy.h"
#include "chrome/test/ui/ui_test.h"
class CookieModalDialogTest : public UITest {
public:
void RunBasicTest(MessageBoxFlags::DialogButton button_to_press,
const std::wstring& expected_title) {
scoped_refptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
ASSERT_TRUE(browser);
ASSERT_TRUE(browser->SetDefaultContentSetting(CONTENT_SETTINGS_TYPE_COOKIES,
CONTENT_SETTING_ASK));
GURL url(URLRequestMockHTTPJob::GetMockUrl(
FilePath(FILE_PATH_LITERAL("cookie2.html"))));
scoped_refptr<TabProxy> tab_proxy(GetActiveTab());
ASSERT_TRUE(tab_proxy);
int64 last_navigation_time;
ASSERT_TRUE(tab_proxy->GetLastNavigationTime(&last_navigation_time));
ASSERT_TRUE(tab_proxy->NavigateToURLAsync(url));
// cookie2.html sets a cookie and then reads back the cookie within onload,
// which means that the navigation will not complete until the cookie
// prompt is dismissed.
bool modal_dialog_showing = false;
MessageBoxFlags::DialogButton available_buttons;
ASSERT_TRUE(automation()->WaitForAppModalDialog());
ASSERT_TRUE(automation()->GetShowingAppModalDialog(&modal_dialog_showing,
&available_buttons));
ASSERT_TRUE(modal_dialog_showing);
ASSERT_NE((button_to_press & available_buttons), 0);
ASSERT_TRUE(automation()->ClickAppModalDialogButton(button_to_press));
// Now, the cookie prompt is dismissed, and we can wait for the navigation
// to complete. Before returning from onload, the test updates the title.
// We can therefore be sure that upon return from WaitForNavigation that
// the title has been updated with the final test result.
ASSERT_TRUE(tab_proxy->WaitForNavigation(last_navigation_time));
std::wstring title;
ASSERT_TRUE(tab_proxy->GetTabTitle(&title));
EXPECT_EQ(expected_title, title);
}
};
// TODO(port): Enable these once the cookie dialogs are fully implemented.
#if defined(OS_WIN)
TEST_F(CookieModalDialogTest, AllowCookies) {
RunBasicTest(MessageBoxFlags::DIALOGBUTTON_OK, L"cookie allowed");
}
TEST_F(CookieModalDialogTest, BlockCookies) {
RunBasicTest(MessageBoxFlags::DIALOGBUTTON_CANCEL, L"cookie blocked");
}
#endif
|
