| OLD | NEW |
| (Empty) | |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "ios/chrome/browser/translate/language_model_factory.h" |
| 6 |
| 7 #include "ios/chrome/browser/browser_state/test_chrome_browser_state.h" |
| 8 #include "ios/web/public/test/test_web_thread_bundle.h" |
| 9 #include "testing/gmock/include/gmock/gmock.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 |
| 12 using testing::IsNull; |
| 13 using testing::Not; |
| 14 |
| 15 class LanguageModelFactoryTest : public testing::Test { |
| 16 public: |
| 17 LanguageModelFactoryTest() { |
| 18 TestChromeBrowserState::Builder browser_state_builder; |
| 19 chrome_browser_state_ = browser_state_builder.Build(); |
| 20 } |
| 21 |
| 22 ~LanguageModelFactoryTest() override { chrome_browser_state_.reset(); } |
| 23 |
| 24 ios::ChromeBrowserState* chrome_browser_state() { |
| 25 return chrome_browser_state_.get(); |
| 26 } |
| 27 |
| 28 private: |
| 29 web::TestWebThreadBundle thread_bundle_; |
| 30 std::unique_ptr<TestChromeBrowserState> chrome_browser_state_; |
| 31 }; |
| 32 |
| 33 TEST_F(LanguageModelFactoryTest, NotCreatedInIncognito) { |
| 34 EXPECT_THAT(LanguageModelFactory::GetForBrowserState(chrome_browser_state()), |
| 35 Not(IsNull())); |
| 36 |
| 37 ios::ChromeBrowserState* otr_browser_state = |
| 38 chrome_browser_state()->GetOffTheRecordChromeBrowserState(); |
| 39 translate::LanguageModel* language_model = |
| 40 LanguageModelFactory::GetForBrowserState(otr_browser_state); |
| 41 EXPECT_THAT(language_model, IsNull()); |
| 42 } |
| OLD | NEW |