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