OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "content/browser/trace_subscriber_stdio.h" | 5 #include "content/browser/trace_subscriber_stdio.h" |
6 | 6 |
| 7 #include "base/message_loop.h" |
7 #include "base/scoped_temp_dir.h" | 8 #include "base/scoped_temp_dir.h" |
| 9 #include "base/threading/sequenced_worker_pool.h" |
| 10 #include "content/public/browser/browser_thread.h" |
8 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
9 | 12 |
10 namespace { | 13 TEST(TraceSubscriberStdioTest, CanWriteDataToFile) { |
| 14 ScopedTempDir trace_dir; |
| 15 ASSERT_TRUE(trace_dir.CreateUniqueTempDir()); |
| 16 FilePath trace_file(trace_dir.path().AppendASCII("trace.txt")); |
| 17 { |
| 18 content::TraceSubscriberStdio subscriber(trace_file); |
11 | 19 |
12 class TraceSubscriberStdioTest : public testing::Test { | 20 std::string foo("foo"); |
13 public: | 21 subscriber.OnTraceDataCollected( |
14 virtual void SetUp() { | 22 make_scoped_refptr(base::RefCountedString::TakeString(&foo))); |
15 ASSERT_TRUE(trace_dir_.CreateUniqueTempDir()); | 23 |
16 trace_file_ = trace_dir_.path().AppendASCII("trace.txt"); | 24 std::string bar("bar"); |
| 25 subscriber.OnTraceDataCollected( |
| 26 make_scoped_refptr(base::RefCountedString::TakeString(&bar))); |
| 27 |
| 28 subscriber.OnEndTracingComplete(); |
17 } | 29 } |
18 | 30 content::BrowserThread::GetBlockingPool()->FlushForTesting(); |
19 std::string ReadTraceFile() { | 31 std::string result; |
20 std::string result; | 32 EXPECT_TRUE(file_util::ReadFileToString(trace_file, &result)); |
21 EXPECT_TRUE(file_util::ReadFileToString(trace_file_, &result)); | 33 EXPECT_EQ("[foo,bar]", result); |
22 return result; | |
23 } | |
24 | |
25 ScopedTempDir trace_dir_; | |
26 FilePath trace_file_; | |
27 }; | |
28 | |
29 } // namespace | |
30 | |
31 TEST_F(TraceSubscriberStdioTest, CanWriteDataToFile) { | |
32 TraceSubscriberStdio subscriber(trace_file_); | |
33 subscriber.OnTraceDataCollected("[foo]"); | |
34 subscriber.OnTraceDataCollected("[bar]"); | |
35 EXPECT_TRUE(subscriber.IsValid()); | |
36 | |
37 subscriber.OnEndTracingComplete(); | |
38 EXPECT_FALSE(subscriber.IsValid()); | |
39 } | 34 } |
40 | |
OLD | NEW |