| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #import <Cocoa/Cocoa.h> | 5 #import <Cocoa/Cocoa.h> |
| 6 | 6 |
| 7 #import "base/mac/scoped_nsobject.h" | 7 #import "base/mac/scoped_nsobject.h" |
| 8 #import "base/strings/string_util.h" | 8 #import "base/strings/string_util.h" |
| 9 #include "base/strings/utf_string_conversions.h" | 9 #include "base/strings/utf_string_conversions.h" |
| 10 #import "chrome/app/chrome_command_ids.h" // For translate menu command ids. | 10 #import "chrome/app/chrome_command_ids.h" // For translate menu command ids. |
| 11 #include "chrome/browser/infobars/infobar_service.h" | 11 #include "chrome/browser/infobars/infobar_service.h" |
| 12 #import "chrome/browser/translate/translate_infobar_delegate.h" | 12 #import "chrome/browser/translate/translate_infobar_delegate.h" |
| 13 #import "chrome/browser/translate/translate_tab_helper.h" | 13 #import "chrome/browser/translate/translate_tab_helper.h" |
| 14 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h" | 14 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h" |
| 15 #import "chrome/browser/ui/cocoa/infobars/before_translate_infobar_controller.h" | 15 #import "chrome/browser/ui/cocoa/infobars/before_translate_infobar_controller.h" |
| 16 #import "chrome/browser/ui/cocoa/infobars/infobar_cocoa.h" | 16 #import "chrome/browser/ui/cocoa/infobars/infobar_cocoa.h" |
| 17 #import "chrome/browser/ui/cocoa/infobars/translate_infobar_base.h" | 17 #import "chrome/browser/ui/cocoa/infobars/translate_infobar_base.h" |
| 18 #include "chrome/test/base/testing_profile.h" | 18 #include "chrome/test/base/testing_profile.h" |
| 19 #include "components/translate/core/browser/translate_language_list.h" | 19 #include "components/translate/core/browser/translate_language_list.h" |
| 20 #include "components/translate/core/browser/translate_manager.h" |
| 20 #import "content/public/browser/web_contents.h" | 21 #import "content/public/browser/web_contents.h" |
| 21 #include "ipc/ipc_message.h" | 22 #include "ipc/ipc_message.h" |
| 22 #import "testing/gmock/include/gmock/gmock.h" | 23 #import "testing/gmock/include/gmock/gmock.h" |
| 23 #import "testing/gtest/include/gtest/gtest.h" | 24 #import "testing/gtest/include/gtest/gtest.h" |
| 24 #import "testing/platform_test.h" | 25 #import "testing/platform_test.h" |
| 25 | 26 |
| 26 using content::WebContents; | 27 using content::WebContents; |
| 27 | 28 |
| 28 namespace { | 29 namespace { |
| 29 | 30 |
| 30 // All states the translate toolbar can assume. | 31 // All states the translate toolbar can assume. |
| 31 translate::TranslateStep kTranslateToolbarStates[] = { | 32 translate::TranslateStep kTranslateToolbarStates[] = { |
| 32 translate::TRANSLATE_STEP_BEFORE_TRANSLATE, | 33 translate::TRANSLATE_STEP_BEFORE_TRANSLATE, |
| 33 translate::TRANSLATE_STEP_AFTER_TRANSLATE, | 34 translate::TRANSLATE_STEP_AFTER_TRANSLATE, |
| 34 translate::TRANSLATE_STEP_TRANSLATING, | 35 translate::TRANSLATE_STEP_TRANSLATING, |
| 35 translate::TRANSLATE_STEP_TRANSLATE_ERROR}; | 36 translate::TRANSLATE_STEP_TRANSLATE_ERROR}; |
| 36 | 37 |
| 37 class MockTranslateInfoBarDelegate : public TranslateInfoBarDelegate { | 38 class MockTranslateInfoBarDelegate : public TranslateInfoBarDelegate { |
| 38 public: | 39 public: |
| 39 MockTranslateInfoBarDelegate(content::WebContents* web_contents, | 40 MockTranslateInfoBarDelegate(content::WebContents* web_contents, |
| 40 translate::TranslateStep step, | 41 translate::TranslateStep step, |
| 41 TranslateErrors::Type error, | 42 TranslateErrors::Type error, |
| 42 PrefService* prefs) | 43 PrefService* prefs) |
| 43 : TranslateInfoBarDelegate(web_contents, | 44 : TranslateInfoBarDelegate( |
| 44 step, | 45 TranslateTabHelper::GetManagerFromWebContents( |
| 45 NULL, | 46 web_contents)->GetWeakPtr(), |
| 46 "en", | 47 false, |
| 47 "es", | 48 step, |
| 48 error, | 49 NULL, |
| 49 prefs, | 50 "en", |
| 50 false) {} | 51 "es", |
| 52 error, |
| 53 prefs, |
| 54 false) {} |
| 51 | 55 |
| 52 MOCK_METHOD0(Translate, void()); | 56 MOCK_METHOD0(Translate, void()); |
| 53 MOCK_METHOD0(RevertTranslation, void()); | 57 MOCK_METHOD0(RevertTranslation, void()); |
| 54 | 58 |
| 55 MOCK_METHOD0(TranslationDeclined, void()); | 59 MOCK_METHOD0(TranslationDeclined, void()); |
| 56 | 60 |
| 57 virtual bool IsTranslatableLanguageByPrefs() OVERRIDE { return true; } | 61 virtual bool IsTranslatableLanguageByPrefs() OVERRIDE { return true; } |
| 58 MOCK_METHOD0(ToggleTranslatableLanguageByPrefs, void()); | 62 MOCK_METHOD0(ToggleTranslatableLanguageByPrefs, void()); |
| 59 virtual bool IsSiteBlacklisted() OVERRIDE { return false; } | 63 virtual bool IsSiteBlacklisted() OVERRIDE { return false; } |
| 60 MOCK_METHOD0(ToggleSiteBlacklist, void()); | 64 MOCK_METHOD0(ToggleSiteBlacklist, void()); |
| (...skipping 200 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 261 translate_prefs->ResetTranslationDeniedCount("en"); | 265 translate_prefs->ResetTranslationDeniedCount("en"); |
| 262 for (int i = 0; i < 4; ++i) { | 266 for (int i = 0; i < 4; ++i) { |
| 263 translate_prefs->IncrementTranslationDeniedCount("en"); | 267 translate_prefs->IncrementTranslationDeniedCount("en"); |
| 264 } | 268 } |
| 265 CreateInfoBar(translate::TRANSLATE_STEP_BEFORE_TRANSLATE); | 269 CreateInfoBar(translate::TRANSLATE_STEP_BEFORE_TRANSLATE); |
| 266 BeforeTranslateInfobarController* controller = | 270 BeforeTranslateInfobarController* controller = |
| 267 (BeforeTranslateInfobarController*)infobar_controller_.get(); | 271 (BeforeTranslateInfobarController*)infobar_controller_.get(); |
| 268 EXPECT_TRUE([[controller alwaysTranslateButton] superview] == nil); | 272 EXPECT_TRUE([[controller alwaysTranslateButton] superview] == nil); |
| 269 EXPECT_TRUE([[controller neverTranslateButton] superview] != nil); | 273 EXPECT_TRUE([[controller neverTranslateButton] superview] != nil); |
| 270 } | 274 } |
| OLD | NEW |