sql/test/scoped_error_ignorer.cc

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