Home | History | Annotate | Download | only in log_private 
      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 
     20 using net::test_server::BasicHttpResponse;
     21 using net::test_server::HttpResponse;
     22 using net::test_server::HttpRequest;
     23 
     24 namespace {
     25 
     26 class 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 
     62 namespace extensions {
     63 
     64 class 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 
     89 IN_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