1// Copyright 2014 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 <string> 6 7#include "base/files/file_path.h" 8#include "base/files/file_util.h" 9#include "chrome/browser/extensions/extension_apitest.h" 10#include "chrome/common/chrome_switches.h" 11#include "chromeos/dbus/dbus_thread_manager.h" 12#include "chromeos/dbus/fake_debug_daemon_client.h" 13#include "content/public/browser/browser_thread.h" 14#include "extensions/common/extension_builder.h" 15#include "net/dns/mock_host_resolver.h" 16#include "net/test/embedded_test_server/embedded_test_server.h" 17#include "net/test/embedded_test_server/http_request.h" 18#include "net/test/embedded_test_server/http_response.h" 19 20using net::test_server::BasicHttpResponse; 21using net::test_server::HttpResponse; 22using net::test_server::HttpRequest; 23 24namespace { 25 26class TestDebugDaemonClient : public chromeos::FakeDebugDaemonClient { 27 public: 28 explicit TestDebugDaemonClient(const base::FilePath& test_file) 29 : test_file_(test_file) {} 30 31 virtual ~TestDebugDaemonClient() {} 32 33 virtual void DumpDebugLogs(bool is_compressed, 34 base::File file, 35 scoped_refptr<base::TaskRunner> task_runner, 36 const GetDebugLogsCallback& callback) OVERRIDE { 37 base::File* file_param = new base::File(file.Pass()); 38 task_runner->PostTaskAndReply( 39 FROM_HERE, 40 base::Bind( 41 &GenerateTestLogDumpFile, test_file_, base::Owned(file_param)), 42 base::Bind(callback, true)); 43 } 44 45 static void GenerateTestLogDumpFile(const base::FilePath& test_tar_file, 46 base::File* file) { 47 std::string test_file_content; 48 EXPECT_TRUE(base::ReadFileToString(test_tar_file, &test_file_content)) 49 << "Cannot read content of file " << test_tar_file.value(); 50 const int data_size = static_cast<int>(test_file_content.size()); 51 EXPECT_EQ(data_size, file->Write(0, test_file_content.data(), data_size)); 52 EXPECT_TRUE(file->SetLength(data_size)); 53 file->Close(); 54 } 55 56 private: 57 base::FilePath test_file_; 58}; 59 60} // namespace 61 62namespace extensions { 63 64class LogPrivateApiTest : public ExtensionApiTest { 65 public: 66 LogPrivateApiTest() {} 67 68 virtual ~LogPrivateApiTest() {} 69 70 virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { 71 base::FilePath tar_file_path = 72 test_data_dir_.Append("log_private/dump_logs/system_logs.tar"); 73 chromeos::DBusThreadManager::GetSetterForTesting()->SetDebugDaemonClient( 74 scoped_ptr<chromeos::DebugDaemonClient>( 75 new TestDebugDaemonClient(tar_file_path))); 76 ExtensionApiTest::SetUpInProcessBrowserTestFixture(); 77 } 78 79 scoped_ptr<HttpResponse> HandleRequest(const HttpRequest& request) { 80 scoped_ptr<BasicHttpResponse> response(new BasicHttpResponse); 81 response->set_code(net::HTTP_OK); 82 response->set_content( 83 "<html><head><title>LogPrivateTest</title>" 84 "</head><body>Hello!</body></html>"); 85 return response.PassAs<HttpResponse>(); 86 } 87}; 88 89IN_PROC_BROWSER_TEST_F(LogPrivateApiTest, DumpLogsAndCaptureEvents) { 90 // Setup dummy HTTP server. 91 host_resolver()->AddRule("www.test.com", "127.0.0.1"); 92 ASSERT_TRUE(StartEmbeddedTestServer()); 93 embedded_test_server()->RegisterRequestHandler( 94 base::Bind(&LogPrivateApiTest::HandleRequest, base::Unretained(this))); 95 96 ASSERT_TRUE(RunExtensionTest("log_private/dump_logs")); 97} 98 99} // namespace extensions 100