src/leveldb/util/testutil.h

   1 // Copyright (c) 2011 The LevelDB 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. See the AUTHORS file for names of contributors.
   4
   5 #ifndef STORAGE_LEVELDB_UTIL_TESTUTIL_H_
   6 #define STORAGE_LEVELDB_UTIL_TESTUTIL_H_
   7
   8 #include "leveldb/env.h"
   9 #include "leveldb/slice.h"
  10 #include "util/random.h"
  11
  12 namespace leveldb {
  13 namespace test {
  14
  15 // Store in *dst a random string of length "len" and return a Slice that
  16 // references the generated data.
  17 extern Slice RandomString(Random* rnd, int len, std::string* dst);
  18
  19 // Return a random key with the specified length that may contain interesting
  20 // characters (e.g. \x00, \xff, etc.).
  21 extern std::string RandomKey(Random* rnd, int len);
  22
  23 // Store in *dst a string of length "len" that will compress to
  24 // "N*compressed_fraction" bytes and return a Slice that references
  25 // the generated data.
  26 extern Slice CompressibleString(Random* rnd, double compressed_fraction,
  27                                 int len, std::string* dst);
  28
  29 // A wrapper that allows injection of errors.
  30 class ErrorEnv : public EnvWrapper {
  31  public:
  32   bool writable_file_error_;
  33   int num_writable_file_errors_;
  34
  35   ErrorEnv() : EnvWrapper(Env::Default()),
  36                writable_file_error_(false),
  37                num_writable_file_errors_(0) { }
  38
  39   virtual Status NewWritableFile(const std::string& fname,
  40                                  WritableFile** result) {
  41     if (writable_file_error_) {
  42       ++num_writable_file_errors_;
  43       *result = NULL;
  44       return Status::IOError(fname, "fake error");
  45     }
  46     return target()->NewWritableFile(fname, result);
  47   }
  48 };
  49
  50 }  // namespace test
  51 }  // namespace leveldb
  52
  53 #endif  // STORAGE_LEVELDB_UTIL_TESTUTIL_H_