OLD | NEW |
| (Empty) |
1 // Copyright (c) 2009 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 "base/file_util.h" | |
6 #include "base/message_loop.h" | |
7 #include "base/path_service.h" | |
8 #include "chrome/browser/chrome_thread.h" | |
9 #include "chrome/browser/net/file_reader.h" | |
10 #include "chrome/common/chrome_paths.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 namespace { | |
14 | |
15 class FileReaderTest : public testing::Test { | |
16 public: | |
17 FileReaderTest() : file_thread_(ChromeThread::FILE) { | |
18 file_thread_.Start(); | |
19 } | |
20 private: | |
21 MessageLoop message_loop_; | |
22 ChromeThread file_thread_; | |
23 }; | |
24 | |
25 class Receiver { | |
26 public: | |
27 Receiver() : succeeded_(false) { | |
28 } | |
29 | |
30 FileReader::Callback* NewCallback() { | |
31 return ::NewCallback(this, &Receiver::DidReadFile); | |
32 } | |
33 | |
34 bool succeeded() const { return succeeded_; } | |
35 const std::string& data() const { return data_; } | |
36 | |
37 private: | |
38 void DidReadFile(bool success, const std::string& data) { | |
39 succeeded_ = success; | |
40 data_ = data; | |
41 MessageLoop::current()->Quit(); | |
42 } | |
43 | |
44 bool succeeded_; | |
45 std::string data_; | |
46 }; | |
47 | |
48 void RunBasicTest(const char* filename) { | |
49 FilePath path; | |
50 PathService::Get(chrome::DIR_TEST_DATA, &path); | |
51 path = path.AppendASCII(filename); | |
52 | |
53 std::string file_contents; | |
54 bool file_exists = file_util::ReadFileToString(path, &file_contents); | |
55 | |
56 Receiver receiver; | |
57 | |
58 scoped_refptr<FileReader> file_reader( | |
59 new FileReader(path, receiver.NewCallback())); | |
60 file_reader->Start(); | |
61 | |
62 MessageLoop::current()->Run(); | |
63 | |
64 EXPECT_EQ(file_exists, receiver.succeeded()); | |
65 EXPECT_EQ(file_contents, receiver.data()); | |
66 } | |
67 | |
68 TEST_F(FileReaderTest, SmallFile) { | |
69 RunBasicTest("title1.html"); | |
70 } | |
71 | |
72 TEST_F(FileReaderTest, BiggerFile) { | |
73 RunBasicTest("download-test1.lib"); | |
74 } | |
75 | |
76 TEST_F(FileReaderTest, NonExistantFile) { | |
77 FilePath path; | |
78 PathService::Get(chrome::DIR_TEST_DATA, &path); | |
79 path = path.AppendASCII("file_that_does_not_exist"); | |
80 | |
81 Receiver receiver; | |
82 | |
83 scoped_refptr<FileReader> file_reader( | |
84 new FileReader(path, receiver.NewCallback())); | |
85 file_reader->Start(); | |
86 | |
87 MessageLoop::current()->Run(); | |
88 | |
89 EXPECT_FALSE(receiver.succeeded()); | |
90 } | |
91 | |
92 } // namespace | |
OLD | NEW |