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 "base/files/file_util.h" | 5 #include "base/files/file_util.h" |
6 #include "base/files/scoped_temp_dir.h" | 6 #include "base/files/scoped_temp_dir.h" |
7 #include "base/stl_util.h" | 7 #include "base/stl_util.h" |
8 #include "base/strings/string_util.h" | 8 #include "base/strings/string_util.h" |
9 #include "chrome/browser/sessions/session_backend.h" | 9 #include "chrome/browser/sessions/session_backend.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
11 | 11 |
12 namespace { | 12 namespace { |
13 | 13 |
14 typedef std::vector<SessionCommand*> SessionCommands; | 14 typedef ScopedVector<SessionCommand> SessionCommands; |
15 | 15 |
16 struct TestData { | 16 struct TestData { |
17 SessionCommand::id_type command_id; | 17 SessionCommand::id_type command_id; |
18 std::string data; | 18 std::string data; |
19 }; | 19 }; |
20 | 20 |
21 SessionCommand* CreateCommandFromData(const TestData& data) { | 21 SessionCommand* CreateCommandFromData(const TestData& data) { |
22 SessionCommand* command = | 22 SessionCommand* command = |
23 new SessionCommand( | 23 new SessionCommand( |
24 data.command_id, | 24 data.command_id, |
(...skipping 22 matching lines...) Expand all Loading... |
47 | 47 |
48 // Path used in testing. | 48 // Path used in testing. |
49 base::FilePath path_; | 49 base::FilePath path_; |
50 base::ScopedTempDir temp_dir_; | 50 base::ScopedTempDir temp_dir_; |
51 }; | 51 }; |
52 | 52 |
53 TEST_F(SessionBackendTest, SimpleReadWrite) { | 53 TEST_F(SessionBackendTest, SimpleReadWrite) { |
54 scoped_refptr<SessionBackend> backend( | 54 scoped_refptr<SessionBackend> backend( |
55 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); | 55 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); |
56 struct TestData data = { 1, "a" }; | 56 struct TestData data = { 1, "a" }; |
57 std::vector<SessionCommand*> commands; | 57 SessionCommands commands; |
58 commands.push_back(CreateCommandFromData(data)); | 58 commands.push_back(CreateCommandFromData(data)); |
59 backend->AppendCommands(new SessionCommands(commands), false); | 59 backend->AppendCommands(new ScopedVector<SessionCommand>(commands.Pass()), |
60 commands.clear(); | 60 false); |
| 61 ASSERT_TRUE(commands.empty()); |
61 | 62 |
62 // Read it back in. | 63 // Read it back in. |
63 backend = NULL; | 64 backend = NULL; |
64 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); | 65 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); |
65 backend->ReadLastSessionCommandsImpl(&commands); | 66 backend->ReadLastSessionCommandsImpl(&commands); |
66 | 67 |
67 ASSERT_EQ(1U, commands.size()); | 68 ASSERT_EQ(1U, commands.size()); |
68 AssertCommandEqualsData(data, commands[0]); | 69 AssertCommandEqualsData(data, commands[0]); |
69 | 70 |
70 STLDeleteElements(&commands); | 71 commands.clear(); |
71 | 72 |
72 backend = NULL; | 73 backend = NULL; |
73 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); | 74 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); |
74 backend->ReadLastSessionCommandsImpl(&commands); | 75 backend->ReadLastSessionCommandsImpl(&commands); |
75 | 76 |
76 ASSERT_EQ(0U, commands.size()); | 77 ASSERT_EQ(0U, commands.size()); |
77 | 78 |
78 // Make sure we can delete. | 79 // Make sure we can delete. |
79 backend->DeleteLastSession(); | 80 backend->DeleteLastSession(); |
80 backend->ReadLastSessionCommandsImpl(&commands); | 81 backend->ReadLastSessionCommandsImpl(&commands); |
(...skipping 13 matching lines...) Expand all Loading... |
94 { 9, "abcdefghi" }, | 95 { 9, "abcdefghi" }, |
95 { 10, "abcdefghij" }, | 96 { 10, "abcdefghij" }, |
96 { 11, "abcdefghijk" }, | 97 { 11, "abcdefghijk" }, |
97 { 12, "abcdefghijkl" }, | 98 { 12, "abcdefghijkl" }, |
98 { 13, "abcdefghijklm" }, | 99 { 13, "abcdefghijklm" }, |
99 }; | 100 }; |
100 | 101 |
101 for (size_t i = 0; i < arraysize(data); ++i) { | 102 for (size_t i = 0; i < arraysize(data); ++i) { |
102 scoped_refptr<SessionBackend> backend( | 103 scoped_refptr<SessionBackend> backend( |
103 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); | 104 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); |
104 std::vector<SessionCommand*> commands; | 105 SessionCommands commands; |
105 if (i != 0) { | 106 if (i != 0) { |
106 // Read previous data. | 107 // Read previous data. |
107 backend->ReadLastSessionCommandsImpl(&commands); | 108 backend->ReadLastSessionCommandsImpl(&commands); |
108 ASSERT_EQ(i, commands.size()); | 109 ASSERT_EQ(i, commands.size()); |
109 for (std::vector<SessionCommand*>::iterator j = commands.begin(); | 110 for (std::vector<SessionCommand*>::iterator j = commands.begin(); |
110 j != commands.end(); ++j) { | 111 j != commands.end(); ++j) { |
111 AssertCommandEqualsData(data[j - commands.begin()], *j); | 112 AssertCommandEqualsData(data[j - commands.begin()], *j); |
112 } | 113 } |
113 backend->AppendCommands(new SessionCommands(commands), false); | 114 backend->AppendCommands(new ScopedVector<SessionCommand>(commands.Pass()), |
114 commands.clear(); | 115 false); |
115 } | 116 } |
116 commands.push_back(CreateCommandFromData(data[i])); | 117 commands.push_back(CreateCommandFromData(data[i])); |
117 backend->AppendCommands(new SessionCommands(commands), false); | 118 backend->AppendCommands(new ScopedVector<SessionCommand>(commands.Pass()), |
| 119 false); |
118 } | 120 } |
119 } | 121 } |
120 | 122 |
121 TEST_F(SessionBackendTest, BigData) { | 123 TEST_F(SessionBackendTest, BigData) { |
122 struct TestData data[] = { | 124 struct TestData data[] = { |
123 { 1, "a" }, | 125 { 1, "a" }, |
124 { 2, "ab" }, | 126 { 2, "ab" }, |
125 }; | 127 }; |
126 | 128 |
127 scoped_refptr<SessionBackend> backend( | 129 scoped_refptr<SessionBackend> backend( |
128 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); | 130 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); |
129 std::vector<SessionCommand*> commands; | 131 ScopedVector<SessionCommand> commands; |
| 132 |
130 commands.push_back(CreateCommandFromData(data[0])); | 133 commands.push_back(CreateCommandFromData(data[0])); |
131 const SessionCommand::size_type big_size = | 134 const SessionCommand::size_type big_size = |
132 SessionBackend::kFileReadBufferSize + 100; | 135 SessionBackend::kFileReadBufferSize + 100; |
133 const SessionCommand::id_type big_id = 50; | 136 const SessionCommand::id_type big_id = 50; |
134 SessionCommand* big_command = new SessionCommand(big_id, big_size); | 137 SessionCommand* big_command = new SessionCommand(big_id, big_size); |
135 reinterpret_cast<char*>(big_command->contents())[0] = 'a'; | 138 reinterpret_cast<char*>(big_command->contents())[0] = 'a'; |
136 reinterpret_cast<char*>(big_command->contents())[big_size - 1] = 'z'; | 139 reinterpret_cast<char*>(big_command->contents())[big_size - 1] = 'z'; |
137 commands.push_back(big_command); | 140 commands.push_back(big_command); |
138 commands.push_back(CreateCommandFromData(data[1])); | 141 commands.push_back(CreateCommandFromData(data[1])); |
139 backend->AppendCommands(new SessionCommands(commands), false); | 142 backend->AppendCommands(new ScopedVector<SessionCommand>(commands.Pass()), |
140 commands.clear(); | 143 false); |
141 | 144 |
142 backend = NULL; | 145 backend = NULL; |
143 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); | 146 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); |
144 commands.clear(); | 147 |
145 backend->ReadLastSessionCommandsImpl(&commands); | 148 backend->ReadLastSessionCommandsImpl(&commands); |
146 ASSERT_EQ(3U, commands.size()); | 149 ASSERT_EQ(3U, commands.size()); |
147 AssertCommandEqualsData(data[0], commands[0]); | 150 AssertCommandEqualsData(data[0], commands[0]); |
148 AssertCommandEqualsData(data[1], commands[2]); | 151 AssertCommandEqualsData(data[1], commands[2]); |
149 | 152 |
150 EXPECT_EQ(big_id, commands[1]->id()); | 153 EXPECT_EQ(big_id, commands[1]->id()); |
151 ASSERT_EQ(big_size, commands[1]->size()); | 154 ASSERT_EQ(big_size, commands[1]->size()); |
152 EXPECT_EQ('a', reinterpret_cast<char*>(commands[1]->contents())[0]); | 155 EXPECT_EQ('a', reinterpret_cast<char*>(commands[1]->contents())[0]); |
153 EXPECT_EQ('z', | 156 EXPECT_EQ('z', |
154 reinterpret_cast<char*>(commands[1]->contents())[big_size - 1]); | 157 reinterpret_cast<char*>(commands[1]->contents())[big_size - 1]); |
155 STLDeleteElements(&commands); | 158 commands.clear(); |
156 } | 159 } |
157 | 160 |
158 TEST_F(SessionBackendTest, EmptyCommand) { | 161 TEST_F(SessionBackendTest, EmptyCommand) { |
159 TestData empty_command; | 162 TestData empty_command; |
160 empty_command.command_id = 1; | 163 empty_command.command_id = 1; |
161 scoped_refptr<SessionBackend> backend( | 164 scoped_refptr<SessionBackend> backend( |
162 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); | 165 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); |
163 std::vector<SessionCommand*>* empty_commands = | 166 SessionCommands* empty_commands = new SessionCommands(); |
164 new std::vector<SessionCommand*>(); | |
165 empty_commands->push_back(CreateCommandFromData(empty_command)); | 167 empty_commands->push_back(CreateCommandFromData(empty_command)); |
166 backend->AppendCommands(empty_commands, true); | 168 backend->AppendCommands(empty_commands, true); |
167 backend->MoveCurrentSessionToLastSession(); | 169 backend->MoveCurrentSessionToLastSession(); |
168 | 170 |
169 std::vector<SessionCommand*> commands; | 171 SessionCommands commands; |
170 backend->ReadLastSessionCommandsImpl(&commands); | 172 backend->ReadLastSessionCommandsImpl(&commands); |
171 ASSERT_EQ(1U, commands.size()); | 173 ASSERT_EQ(1U, commands.size()); |
172 AssertCommandEqualsData(empty_command, commands[0]); | 174 AssertCommandEqualsData(empty_command, commands[0]); |
173 STLDeleteElements(&commands); | 175 commands.clear(); |
174 } | 176 } |
175 | 177 |
176 // Writes a command, appends another command with reset to true, then reads | 178 // Writes a command, appends another command with reset to true, then reads |
177 // making sure we only get back the second command. | 179 // making sure we only get back the second command. |
178 TEST_F(SessionBackendTest, Truncate) { | 180 TEST_F(SessionBackendTest, Truncate) { |
179 scoped_refptr<SessionBackend> backend( | 181 scoped_refptr<SessionBackend> backend( |
180 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); | 182 new SessionBackend(BaseSessionService::SESSION_RESTORE, path_)); |
181 struct TestData first_data = { 1, "a" }; | 183 struct TestData first_data = { 1, "a" }; |
182 std::vector<SessionCommand*> commands; | 184 SessionCommands commands; |
183 commands.push_back(CreateCommandFromData(first_data)); | 185 commands.push_back(CreateCommandFromData(first_data)); |
184 backend->AppendCommands(new SessionCommands(commands), false); | 186 backend->AppendCommands(new ScopedVector<SessionCommand>(commands.Pass()), |
185 commands.clear(); | 187 false); |
186 | 188 |
187 // Write another command, this time resetting the file when appending. | 189 // Write another command, this time resetting the file when appending. |
188 struct TestData second_data = { 2, "b" }; | 190 struct TestData second_data = { 2, "b" }; |
189 commands.push_back(CreateCommandFromData(second_data)); | 191 commands.push_back(CreateCommandFromData(second_data)); |
190 backend->AppendCommands(new SessionCommands(commands), true); | 192 backend->AppendCommands(new ScopedVector<SessionCommand>(commands.Pass()), |
191 commands.clear(); | 193 true); |
192 | 194 |
193 // Read it back in. | 195 // Read it back in. |
194 backend = NULL; | 196 backend = NULL; |
195 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); | 197 backend = new SessionBackend(BaseSessionService::SESSION_RESTORE, path_); |
196 backend->ReadLastSessionCommandsImpl(&commands); | 198 backend->ReadLastSessionCommandsImpl(&commands); |
197 | 199 |
198 // And make sure we get back the expected data. | 200 // And make sure we get back the expected data. |
199 ASSERT_EQ(1U, commands.size()); | 201 ASSERT_EQ(1U, commands.size()); |
200 AssertCommandEqualsData(second_data, commands[0]); | 202 AssertCommandEqualsData(second_data, commands[0]); |
201 | 203 |
202 STLDeleteElements(&commands); | 204 commands.clear(); |
203 } | 205 } |
OLD | NEW |