src/net/base/mock_file_stream.h

   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
   3 // found in the LICENSE file.
   4
   5 // This file defines MockFileStream, a test class for FileStream.
   6
   7 #ifndef NET_BASE_MOCK_FILE_STREAM_H_
   8 #define NET_BASE_MOCK_FILE_STREAM_H_
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/files/file_path.h"
  13 #include "net/base/file_stream.h"
  14 #include "net/base/net_errors.h"
  15
  16 namespace net {
  17
  18 class IOBuffer;
  19
  20 namespace testing {
  21
  22 class MockFileStream : public net::FileStream {
  23  public:
  24   MockFileStream(net::NetLog* net_log)
  25       : net::FileStream(net_log), forced_error_(net::OK) {}
  26
  27   MockFileStream(base::PlatformFile file, int flags, net::NetLog* net_log)
  28       : net::FileStream(file, flags, net_log), forced_error_(net::OK) {}
  29
  30   // FileStream methods.
  31   virtual int OpenSync(const base::FilePath& path, int open_flags) OVERRIDE;
  32   virtual int Seek(net::Whence whence, int64 offset,
  33                    const Int64CompletionCallback& callback) OVERRIDE;
  34   virtual int64 SeekSync(net::Whence whence, int64 offset) OVERRIDE;
  35   virtual int64 Available() OVERRIDE;
  36   virtual int Read(IOBuffer* buf,
  37                    int buf_len,
  38                    const CompletionCallback& callback) OVERRIDE;
  39   virtual int ReadSync(char* buf, int buf_len) OVERRIDE;
  40   virtual int ReadUntilComplete(char *buf, int buf_len) OVERRIDE;
  41   virtual int Write(IOBuffer* buf,
  42                     int buf_len,
  43                     const CompletionCallback& callback) OVERRIDE;
  44   virtual int WriteSync(const char* buf, int buf_len) OVERRIDE;
  45   virtual int64 Truncate(int64 bytes) OVERRIDE;
  46   virtual int Flush(const CompletionCallback& callback) OVERRIDE;
  47   virtual int FlushSync() OVERRIDE;
  48
  49   void set_forced_error(int error) { forced_error_ = error; }
  50   void clear_forced_error() { forced_error_ = net::OK; }
  51   int forced_error() const { return forced_error_; }
  52   const base::FilePath& get_path() const { return path_; }
  53
  54  private:
  55   int ReturnError(int function_error) {
  56     if (forced_error_ != net::OK) {
  57       int ret = forced_error_;
  58       clear_forced_error();
  59       return ret;
  60     }
  61
  62     return function_error;
  63   }
  64
  65   int64 ReturnError64(int64 function_error) {
  66     if (forced_error_ != net::OK) {
  67       int64 ret = forced_error_;
  68       clear_forced_error();
  69       return ret;
  70     }
  71
  72     return function_error;
  73   }
  74
  75   int forced_error_;
  76   base::FilePath path_;
  77 };
  78
  79 }  // namespace testing
  80
  81 }  // namespace net
  82
  83 #endif  // NET_BASE_MOCK_FILE_STREAM_H_