chrome/browser/cocoa/autocomplete_text_field_unittest_helper.mm


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

// 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 "chrome/browser/cocoa/autocomplete_text_field_unittest_helper.h"

#import "chrome/browser/cocoa/autocomplete_text_field.h"
#import "chrome/browser/cocoa/autocomplete_text_field_editor.h"
#include "testing/gtest/include/gtest/gtest.h"

@implementation AutocompleteTextFieldWindowTestDelegate

- (id)windowWillReturnFieldEditor:(NSWindow *)sender toObject:(id)anObject {
  id editor = nil;
  if ([anObject isKindOfClass:[AutocompleteTextField class]]) {
    if (editor_ == nil) {
      editor_.reset([[AutocompleteTextFieldEditor alloc] init]);
    }
    EXPECT_TRUE(editor_ != nil);

    // This needs to be called every time, otherwise notifications
    // aren't sent correctly.
    [editor_ setFieldEditor:YES];
    editor = editor_.get();
  }
  return editor;
}

@end