blob: d1a22f4fdd74f0dbf0c8ef56a73646a88bf9ee4e (
plain)
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
66
67
68
69
70
71
72
73
74
75
76
|
// Copyright 2015 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.
#import "ios/web/web_view_counter_impl.h"
#import "base/mac/scoped_nsobject.h"
#include "base/memory/scoped_ptr.h"
#include "ios/web/public/browser_state.h"
#include "ios/web/public/test/test_browser_state.h"
#import "ios/web/public/test/test_web_client.h"
#include "ios/web/public/test/test_web_thread_bundle.h"
#include "ios/web/public/test/web_test_util.h"
#include "ios/web/public/web_client.h"
#import "ios/web/public/web_view_counter.h"
#import "ios/web/public/web_view_creation_util.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"
namespace web {
namespace {
// A test fixture that sets up web-related classes for testing a
// |WebViewCounter|.
class WebViewCounterTest : public PlatformTest {
protected:
void SetUp() override {
PlatformTest::SetUp();
SetWebClient(&test_web_client_);
browser_state_.reset(new TestBrowserState());
}
void TearDown() override {
// The BrowserState needs to be destroyed first so that it is outlived by
// the WebThreadBundle.
browser_state_.reset();
SetWebClient(nullptr);
PlatformTest::TearDown();
}
// The BrowserState used for testing purposes.
scoped_ptr<BrowserState> browser_state_;
private:
// Used to create TestWebThreads.
TestWebThreadBundle thread_bundle_;
// Required by web::CreateWebView/web::CreateWKWebView functions.
web::TestWebClient test_web_client_;
};
} // namespace
// Tests that WebViewCounter correctly maintains the count of WKWebViews.
TEST_F(WebViewCounterTest, WKWebViewCount) {
CR_TEST_REQUIRES_WK_WEB_VIEW();
WebViewCounterImpl* web_view_counter =
WebViewCounterImpl::FromBrowserState(browser_state_.get());
ASSERT_TRUE(web_view_counter);
base::scoped_nsobject<WKWebView> web_view_1(
web::CreateWKWebView(CGRectZero, browser_state_.get()));
web_view_counter->InsertWKWebView(web_view_1);
base::scoped_nsobject<WKWebView> web_view_2(
web::CreateWKWebView(CGRectZero, browser_state_.get()));
web_view_counter->InsertWKWebView(web_view_2);
EXPECT_EQ(2U, web_view_counter->GetWKWebViewCount());
// Deallocate WKWebViews.
web_view_2.reset();
EXPECT_EQ(1U, web_view_counter->GetWKWebViewCount());
web_view_1.reset();
EXPECT_EQ(0U, web_view_counter->GetWKWebViewCount());
}
} // namespace web
|
