OLD | NEW |
| (Empty) |
1 // Copyright 2013 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 "sql/test/scoped_error_ignorer.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "testing/gtest/include/gtest/gtest.h" | |
9 | |
10 namespace sql { | |
11 | |
12 // static | |
13 int ScopedErrorIgnorer::SQLiteLibVersionNumber() { | |
14 return sqlite3_libversion_number(); | |
15 } | |
16 | |
17 ScopedErrorIgnorer::ScopedErrorIgnorer() | |
18 : checked_(false) { | |
19 callback_ = | |
20 base::Bind(&ScopedErrorIgnorer::ShouldIgnore, base::Unretained(this)); | |
21 Connection::SetErrorIgnorer(&callback_); | |
22 } | |
23 | |
24 ScopedErrorIgnorer::~ScopedErrorIgnorer() { | |
25 EXPECT_TRUE(checked_) << " Test must call CheckIgnoredErrors()"; | |
26 Connection::ResetErrorIgnorer(); | |
27 } | |
28 | |
29 void ScopedErrorIgnorer::IgnoreError(int err) { | |
30 EXPECT_EQ(0u, ignore_errors_.count(err)) | |
31 << " Error " << err << " is already ignored"; | |
32 ignore_errors_.insert(err); | |
33 } | |
34 | |
35 bool ScopedErrorIgnorer::CheckIgnoredErrors() { | |
36 checked_ = true; | |
37 return errors_ignored_ == ignore_errors_; | |
38 } | |
39 | |
40 bool ScopedErrorIgnorer::ShouldIgnore(int err) { | |
41 // Look for extended code. | |
42 if (ignore_errors_.count(err) > 0) { | |
43 // Record that the error was seen and ignore it. | |
44 errors_ignored_.insert(err); | |
45 return true; | |
46 } | |
47 | |
48 // Trim extended codes and check again. | |
49 int base_err = err & 0xff; | |
50 if (ignore_errors_.count(base_err) > 0) { | |
51 // Record that the error was seen and ignore it. | |
52 errors_ignored_.insert(base_err); | |
53 return true; | |
54 } | |
55 | |
56 // Unexpected error. | |
57 ADD_FAILURE() << " Unexpected SQLite error " << err; | |
58 | |
59 // TODO(shess): If it never makes sense to pass through an error | |
60 // under the test harness, then perhaps the ignore callback | |
61 // signature should be changed. | |
62 return true; | |
63 } | |
64 | |
65 } // namespace sql | |
OLD | NEW |