OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/files/important_file_writer.h" | 5 #include "base/files/important_file_writer.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
9 #include "base/files/file_path.h" | 9 #include "base/files/file_path.h" |
10 #include "base/files/file_util.h" | 10 #include "base/files/file_util.h" |
(...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
92 protected: | 92 protected: |
93 SuccessfulWriteObserver successful_write_observer_; | 93 SuccessfulWriteObserver successful_write_observer_; |
94 FilePath file_; | 94 FilePath file_; |
95 MessageLoop loop_; | 95 MessageLoop loop_; |
96 | 96 |
97 private: | 97 private: |
98 ScopedTempDir temp_dir_; | 98 ScopedTempDir temp_dir_; |
99 }; | 99 }; |
100 | 100 |
101 TEST_F(ImportantFileWriterTest, Basic) { | 101 TEST_F(ImportantFileWriterTest, Basic) { |
102 ImportantFileWriter writer(file_, ThreadTaskRunnerHandle::Get()); | 102 ImportantFileWriterImpl writer(file_, ThreadTaskRunnerHandle::Get()); |
103 EXPECT_FALSE(PathExists(writer.path())); | 103 EXPECT_FALSE(PathExists(writer.path())); |
104 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); | 104 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); |
105 writer.WriteNow(make_scoped_ptr(new std::string("foo"))); | 105 writer.WriteNow(make_scoped_ptr(new std::string("foo"))); |
106 RunLoop().RunUntilIdle(); | 106 RunLoop().RunUntilIdle(); |
107 | 107 |
108 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); | 108 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); |
109 ASSERT_TRUE(PathExists(writer.path())); | 109 ASSERT_TRUE(PathExists(writer.path())); |
110 EXPECT_EQ("foo", GetFileContent(writer.path())); | 110 EXPECT_EQ("foo", GetFileContent(writer.path())); |
111 } | 111 } |
112 | 112 |
113 TEST_F(ImportantFileWriterTest, BasicWithSuccessfulWriteObserver) { | 113 TEST_F(ImportantFileWriterTest, BasicWithSuccessfulWriteObserver) { |
114 ImportantFileWriter writer(file_, ThreadTaskRunnerHandle::Get()); | 114 ImportantFileWriterImpl writer(file_, ThreadTaskRunnerHandle::Get()); |
115 EXPECT_FALSE(PathExists(writer.path())); | 115 EXPECT_FALSE(PathExists(writer.path())); |
116 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); | 116 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); |
117 successful_write_observer_.ObserveNextSuccessfulWrite(&writer); | 117 successful_write_observer_.ObserveNextSuccessfulWrite(&writer); |
118 writer.WriteNow(make_scoped_ptr(new std::string("foo"))); | 118 writer.WriteNow(make_scoped_ptr(new std::string("foo"))); |
119 RunLoop().RunUntilIdle(); | 119 RunLoop().RunUntilIdle(); |
120 | 120 |
121 // Confirm that the observer is invoked. | 121 // Confirm that the observer is invoked. |
122 EXPECT_TRUE(successful_write_observer_.GetAndResetObservationState()); | 122 EXPECT_TRUE(successful_write_observer_.GetAndResetObservationState()); |
123 ASSERT_TRUE(PathExists(writer.path())); | 123 ASSERT_TRUE(PathExists(writer.path())); |
124 EXPECT_EQ("foo", GetFileContent(writer.path())); | 124 EXPECT_EQ("foo", GetFileContent(writer.path())); |
(...skipping 13 matching lines...) Expand all Loading... |
138 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); | 138 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); |
139 writer.WriteNow(make_scoped_ptr(new std::string("baz"))); | 139 writer.WriteNow(make_scoped_ptr(new std::string("baz"))); |
140 RunLoop().RunUntilIdle(); | 140 RunLoop().RunUntilIdle(); |
141 | 141 |
142 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); | 142 EXPECT_FALSE(successful_write_observer_.GetAndResetObservationState()); |
143 ASSERT_TRUE(PathExists(writer.path())); | 143 ASSERT_TRUE(PathExists(writer.path())); |
144 EXPECT_EQ("baz", GetFileContent(writer.path())); | 144 EXPECT_EQ("baz", GetFileContent(writer.path())); |
145 } | 145 } |
146 | 146 |
147 TEST_F(ImportantFileWriterTest, ScheduleWrite) { | 147 TEST_F(ImportantFileWriterTest, ScheduleWrite) { |
148 ImportantFileWriter writer(file_, ThreadTaskRunnerHandle::Get()); | 148 ImportantFileWriterImpl writer(file_, ThreadTaskRunnerHandle::Get()); |
149 writer.set_commit_interval(TimeDelta::FromMilliseconds(25)); | 149 writer.set_commit_interval(TimeDelta::FromMilliseconds(25)); |
150 EXPECT_FALSE(writer.HasPendingWrite()); | 150 EXPECT_FALSE(writer.HasPendingWrite()); |
151 DataSerializer serializer("foo"); | 151 DataSerializer serializer("foo"); |
152 writer.ScheduleWrite(&serializer); | 152 writer.ScheduleWrite(&serializer); |
153 EXPECT_TRUE(writer.HasPendingWrite()); | 153 EXPECT_TRUE(writer.HasPendingWrite()); |
154 ThreadTaskRunnerHandle::Get()->PostDelayedTask( | 154 ThreadTaskRunnerHandle::Get()->PostDelayedTask( |
155 FROM_HERE, MessageLoop::QuitWhenIdleClosure(), | 155 FROM_HERE, MessageLoop::QuitWhenIdleClosure(), |
156 TimeDelta::FromMilliseconds(100)); | 156 TimeDelta::FromMilliseconds(100)); |
157 MessageLoop::current()->Run(); | 157 MessageLoop::current()->Run(); |
158 EXPECT_FALSE(writer.HasPendingWrite()); | 158 EXPECT_FALSE(writer.HasPendingWrite()); |
159 ASSERT_TRUE(PathExists(writer.path())); | 159 ASSERT_TRUE(PathExists(writer.path())); |
160 EXPECT_EQ("foo", GetFileContent(writer.path())); | 160 EXPECT_EQ("foo", GetFileContent(writer.path())); |
161 } | 161 } |
162 | 162 |
163 TEST_F(ImportantFileWriterTest, DoScheduledWrite) { | 163 TEST_F(ImportantFileWriterTest, DoScheduledWrite) { |
164 ImportantFileWriter writer(file_, ThreadTaskRunnerHandle::Get()); | 164 ImportantFileWriterImpl writer(file_, ThreadTaskRunnerHandle::Get()); |
165 EXPECT_FALSE(writer.HasPendingWrite()); | 165 EXPECT_FALSE(writer.HasPendingWrite()); |
166 DataSerializer serializer("foo"); | 166 DataSerializer serializer("foo"); |
167 writer.ScheduleWrite(&serializer); | 167 writer.ScheduleWrite(&serializer); |
168 EXPECT_TRUE(writer.HasPendingWrite()); | 168 EXPECT_TRUE(writer.HasPendingWrite()); |
169 writer.DoScheduledWrite(); | 169 writer.DoScheduledWrite(); |
170 ThreadTaskRunnerHandle::Get()->PostDelayedTask( | 170 ThreadTaskRunnerHandle::Get()->PostDelayedTask( |
171 FROM_HERE, MessageLoop::QuitWhenIdleClosure(), | 171 FROM_HERE, MessageLoop::QuitWhenIdleClosure(), |
172 TimeDelta::FromMilliseconds(100)); | 172 TimeDelta::FromMilliseconds(100)); |
173 MessageLoop::current()->Run(); | 173 MessageLoop::current()->Run(); |
174 EXPECT_FALSE(writer.HasPendingWrite()); | 174 EXPECT_FALSE(writer.HasPendingWrite()); |
175 ASSERT_TRUE(PathExists(writer.path())); | 175 ASSERT_TRUE(PathExists(writer.path())); |
176 EXPECT_EQ("foo", GetFileContent(writer.path())); | 176 EXPECT_EQ("foo", GetFileContent(writer.path())); |
177 } | 177 } |
178 | 178 |
179 TEST_F(ImportantFileWriterTest, BatchingWrites) { | 179 TEST_F(ImportantFileWriterTest, BatchingWrites) { |
180 ImportantFileWriter writer(file_, ThreadTaskRunnerHandle::Get()); | 180 ImportantFileWriterImpl writer(file_, ThreadTaskRunnerHandle::Get()); |
181 writer.set_commit_interval(TimeDelta::FromMilliseconds(25)); | 181 writer.set_commit_interval(TimeDelta::FromMilliseconds(25)); |
182 DataSerializer foo("foo"), bar("bar"), baz("baz"); | 182 DataSerializer foo("foo"), bar("bar"), baz("baz"); |
183 writer.ScheduleWrite(&foo); | 183 writer.ScheduleWrite(&foo); |
184 writer.ScheduleWrite(&bar); | 184 writer.ScheduleWrite(&bar); |
185 writer.ScheduleWrite(&baz); | 185 writer.ScheduleWrite(&baz); |
186 ThreadTaskRunnerHandle::Get()->PostDelayedTask( | 186 ThreadTaskRunnerHandle::Get()->PostDelayedTask( |
187 FROM_HERE, MessageLoop::QuitWhenIdleClosure(), | 187 FROM_HERE, MessageLoop::QuitWhenIdleClosure(), |
188 TimeDelta::FromMilliseconds(100)); | 188 TimeDelta::FromMilliseconds(100)); |
189 MessageLoop::current()->Run(); | 189 MessageLoop::current()->Run(); |
190 ASSERT_TRUE(PathExists(writer.path())); | 190 ASSERT_TRUE(PathExists(writer.path())); |
191 EXPECT_EQ("baz", GetFileContent(writer.path())); | 191 EXPECT_EQ("baz", GetFileContent(writer.path())); |
192 } | 192 } |
193 | 193 |
194 } // namespace base | 194 } // namespace base |
OLD | NEW |