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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// Copyright (c) 2009 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 <Cocoa/Cocoa.h>
#include "base/scoped_nsautorelease_pool.h"
#include "base/scoped_nsobject.h"
#include "chrome/browser/cocoa/browser_test_helper.h"
#import "chrome/browser/cocoa/browser_window_controller.h"
#import "chrome/browser/cocoa/cocoa_test_helper.h"
#import "chrome/browser/cocoa/infobar_container_controller.h"
#include "chrome/browser/cocoa/infobar_test_helper.h"
#include "testing/gtest/include/gtest/gtest.h"
// Objective-C classes must be defined outside the namespace.
@interface BrowserWindowControllerPong : BrowserWindowController {
BOOL pong_;
}
@property(readonly) BOOL pong;
@end
@implementation BrowserWindowControllerPong
@synthesize pong = pong_;
- (id)initWithBrowser:(Browser*)browser {
if ((self = [super initWithBrowser:browser takeOwnership:NO])) {
pong_ = NO;
}
return self;
}
- (void)infoBarResized:(float)newHeight {
pong_ = TRUE;
}
@end
namespace {
class InfoBarContainerControllerTest : public testing::Test {
virtual void SetUp() {
browserController_.reset([[BrowserWindowControllerPong alloc]
initWithBrowser:browser_helper_.browser()]);
TabStripModel* model = browser_helper_.browser()->tabstrip_model();
controller_.reset([[InfoBarContainerController alloc]
initWithTabStripModel:model
browserWindowController:browserController_]);
}
public:
// Order is very important here. We want the controller deleted
// before the pool, and want the pool deleted before
// BrowserTestHelper.
CocoaTestHelper cocoa_helper_;
BrowserTestHelper browser_helper_;
base::ScopedNSAutoreleasePool pool_;
scoped_nsobject<BrowserWindowControllerPong> browserController_;
scoped_nsobject<InfoBarContainerController> controller_;
};
TEST_F(InfoBarContainerControllerTest, Show) {
// Make sure the container's view is non-nil and draws without crashing.
NSView* view = [controller_ view];
EXPECT_TRUE(view != nil);
[cocoa_helper_.contentView() addSubview:view];
}
TEST_F(InfoBarContainerControllerTest, BWCPong) {
// Call positionInfoBarsAndResize and check that the BWC got a resize message.
[controller_ positionInfoBarsAndRedraw];
EXPECT_TRUE([browserController_ pong]);
}
TEST_F(InfoBarContainerControllerTest, AddAndRemoveInfoBars) {
NSView* view = [controller_ view];
[cocoa_helper_.contentView() addSubview:view];
// Add three infobars, one of each type, and then remove them.
// After each step check to make sure we have the correct number of
// infobar subviews.
MockAlertInfoBarDelegate alertDelegate;
MockLinkInfoBarDelegate linkDelegate;
MockConfirmInfoBarDelegate confirmDelegate;
[controller_ addInfoBar:&alertDelegate];
EXPECT_EQ(1U, [[view subviews] count]);
[controller_ addInfoBar:&linkDelegate];
EXPECT_EQ(2U, [[view subviews] count]);
[controller_ addInfoBar:&confirmDelegate];
EXPECT_EQ(3U, [[view subviews] count]);
// Just to mix things up, remove them in a different order.
[controller_ removeInfoBarsForDelegate:&linkDelegate];
EXPECT_EQ(2U, [[view subviews] count]);
[controller_ removeInfoBarsForDelegate:&confirmDelegate];
EXPECT_EQ(1U, [[view subviews] count]);
[controller_ removeInfoBarsForDelegate:&alertDelegate];
EXPECT_EQ(0U, [[view subviews] count]);
}
TEST_F(InfoBarContainerControllerTest, RemoveAllInfoBars) {
NSView* view = [controller_ view];
[cocoa_helper_.contentView() addSubview:view];
// Add three infobars and then remove them all.
MockAlertInfoBarDelegate alertDelegate;
MockLinkInfoBarDelegate linkDelegate;
MockConfirmInfoBarDelegate confirmDelegate;
[controller_ addInfoBar:&alertDelegate];
[controller_ addInfoBar:&linkDelegate];
[controller_ addInfoBar:&confirmDelegate];
EXPECT_EQ(3U, [[view subviews] count]);
[controller_ removeAllInfoBars];
EXPECT_EQ(0U, [[view subviews] count]);
}
} // namespace
|
