| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
| 8 #include "base/file_path.h" | 8 #include "base/file_path.h" |
| 9 #include "base/process_util.h" | 9 #include "base/process_util.h" |
| 10 #include "content/browser/child_process_security_policy_impl.h" | 10 #include "content/browser/child_process_security_policy_impl.h" |
| 11 #include "content/browser/web_contents/web_contents_impl.h" | 11 #include "content/browser/web_contents/web_contents_impl.h" |
| 12 #include "content/public/browser/render_process_host.h" | 12 #include "content/public/browser/render_process_host.h" |
| 13 #include "content/public/common/result_codes.h" | 13 #include "content/public/common/result_codes.h" |
| 14 #include "content/shell/shell.h" | 14 #include "content/shell/shell.h" |
| 15 #include "content/test/content_browser_test.h" | 15 #include "content/test/content_browser_test.h" |
| 16 #include "content/test/content_browser_test_utils.h" | 16 #include "content/test/content_browser_test_utils.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 18 | 18 |
| 19 namespace content { | |
| 20 | |
| 21 class ChildProcessSecurityPolicyInProcessBrowserTest | 19 class ChildProcessSecurityPolicyInProcessBrowserTest |
| 22 : public ContentBrowserTest { | 20 : public content::ContentBrowserTest { |
| 23 public: | 21 public: |
| 24 virtual void SetUp() OVERRIDE { | 22 virtual void SetUp() { |
| 25 EXPECT_EQ( | 23 EXPECT_EQ( |
| 26 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size(), | 24 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size(), |
| 27 0U); | 25 0U); |
| 28 ContentBrowserTest::SetUp(); | 26 ContentBrowserTest::SetUp(); |
| 29 } | 27 } |
| 30 | 28 |
| 31 virtual void TearDown() OVERRIDE { | 29 virtual void TearDown() { |
| 32 EXPECT_EQ( | 30 EXPECT_EQ( |
| 33 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size(), | 31 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size(), |
| 34 0U); | 32 0U); |
| 35 ContentBrowserTest::TearDown(); | 33 ContentBrowserTest::TearDown(); |
| 36 } | 34 } |
| 37 }; | 35 }; |
| 38 | 36 |
| 39 #if !defined(NDEBUG) && defined(OS_MACOSX) | 37 #if !defined(NDEBUG) && defined(OS_MACOSX) |
| 40 IN_PROC_BROWSER_TEST_F(ChildProcessSecurityPolicyInProcessBrowserTest, DISABLED_
NoLeak) { | 38 IN_PROC_BROWSER_TEST_F(ChildProcessSecurityPolicyInProcessBrowserTest, DISABLED_
NoLeak) { |
| 41 #else | 39 #else |
| 42 IN_PROC_BROWSER_TEST_F(ChildProcessSecurityPolicyInProcessBrowserTest, NoLeak) { | 40 IN_PROC_BROWSER_TEST_F(ChildProcessSecurityPolicyInProcessBrowserTest, NoLeak) { |
| 43 #endif | 41 #endif |
| 44 GURL url = GetTestUrl("", "simple_page.html"); | 42 GURL url = content::GetTestUrl("", "simple_page.html"); |
| 45 | 43 |
| 46 NavigateToURL(shell(), url); | 44 content::NavigateToURL(shell(), url); |
| 47 EXPECT_EQ( | 45 EXPECT_EQ( |
| 48 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size(), | 46 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size(), |
| 49 1U); | 47 1U); |
| 50 | 48 |
| 51 WebContents* web_contents = shell()->web_contents(); | 49 content::WebContents* web_contents = shell()->web_contents(); |
| 52 base::KillProcess(web_contents->GetRenderProcessHost()->GetHandle(), | 50 base::KillProcess(web_contents->GetRenderProcessHost()->GetHandle(), |
| 53 RESULT_CODE_KILLED, true); | 51 content::RESULT_CODE_KILLED, true); |
| 54 | 52 |
| 55 web_contents->GetController().Reload(true); | 53 web_contents->GetController().Reload(true); |
| 56 EXPECT_EQ( | 54 EXPECT_EQ( |
| 57 1U, | 55 1U, |
| 58 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size()); | 56 ChildProcessSecurityPolicyImpl::GetInstance()->security_state_.size()); |
| 59 } | 57 } |
| 60 | |
| 61 } // namespace content | |
| OLD | NEW |