| OLD | NEW |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 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 | 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 #include "chrome/browser/extensions/extension_tab_util.h" | 5 #include "chrome/browser/extensions/extension_tab_util.h" |
| 6 | 6 |
| 7 #include "base/macros.h" | 7 #include "base/macros.h" |
| 8 #include "base/memory/ptr_util.h" | 8 #include "base/memory/ptr_util.h" |
| 9 #include "chrome/common/extensions/api/tabs.h" | 9 #include "chrome/common/extensions/api/tabs.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 | 11 |
| 12 namespace extensions { | 12 namespace extensions { |
| 13 | 13 |
| 14 namespace { | 14 namespace { |
| 15 | 15 |
| 16 const char kCustomUrl[] = "www.example.com/foo?bar=baz"; | 16 const char kCustomUrl[] = "www.example.com/foo?bar=baz"; |
| 17 | 17 |
| 18 class ExtensionTabUtilTestDelegate : public ExtensionTabUtil::Delegate { | 18 class ExtensionTabUtilTestDelegate : public ExtensionTabUtil::Delegate { |
| 19 public: | 19 public: |
| 20 ExtensionTabUtilTestDelegate() {} | 20 ExtensionTabUtilTestDelegate() {} |
| 21 ~ExtensionTabUtilTestDelegate() override {} | 21 ~ExtensionTabUtilTestDelegate() override {} |
| 22 | 22 |
| 23 // ExtensionTabUtil::Delegate | 23 // ExtensionTabUtil::Delegate |
| 24 void ScrubTabForExtension(const Extension* extension, | 24 void ScrubTabForExtension(const std::string& extension_id, |
| 25 content::WebContents* contents, | |
| 26 api::tabs::Tab* tab) override { | 25 api::tabs::Tab* tab) override { |
| 27 tab->url.reset(new std::string(kCustomUrl)); | 26 tab->url.reset(new std::string(kCustomUrl)); |
| 28 } | 27 } |
| 29 | 28 |
| 30 private: | 29 private: |
| 31 DISALLOW_COPY_AND_ASSIGN(ExtensionTabUtilTestDelegate); | 30 DISALLOW_COPY_AND_ASSIGN(ExtensionTabUtilTestDelegate); |
| 32 }; | 31 }; |
| 33 | 32 |
| 34 } // namespace | 33 } // namespace |
| 35 | 34 |
| 36 // Test that the custom ScrubTabForExtension delegate works - in this test it | 35 // Test that the custom ScrubTabForExtension delegate works - in this test it |
| 37 // sets URL to a custom string. | 36 // sets URL to a custom string. |
| 38 TEST(ExtensionTabUtilTest, Delegate) { | 37 TEST(ExtensionTabUtilTest, Delegate) { |
| 39 auto test_delegate = base::MakeUnique<ExtensionTabUtilTestDelegate>(); | 38 auto test_delegate = base::MakeUnique<ExtensionTabUtilTestDelegate>(); |
| 40 ExtensionTabUtil::SetPlatformDelegate(test_delegate.get()); | 39 ExtensionTabUtil::SetPlatformDelegate(test_delegate.get()); |
| 41 | 40 |
| 42 api::tabs::Tab tab; | 41 api::tabs::Tab tab; |
| 43 ExtensionTabUtil::ScrubTabForExtension(nullptr, nullptr, &tab); | 42 ExtensionTabUtil::ScrubTabForExtension(nullptr, nullptr, &tab); |
| 44 EXPECT_EQ(kCustomUrl, *tab.url); | 43 EXPECT_EQ(kCustomUrl, *tab.url); |
| 45 | 44 |
| 46 // Unset the delegate. | 45 // Unset the delegate. |
| 47 ExtensionTabUtil::SetPlatformDelegate(nullptr); | 46 ExtensionTabUtil::SetPlatformDelegate(nullptr); |
| 48 } | 47 } |
| 49 | 48 |
| 50 } // namespace extensions | 49 } // namespace extensions |
| OLD | NEW |