fake_file_access.h

Go to the documentation of this file.

// Copyright 2017 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
 
#ifndef TESTING_FAKE_FILE_ACCESS_H_
#define TESTING_FAKE_FILE_ACCESS_H_
 
#include <memory>
 
#include "core/fxcrt/unowned_ptr.h"
#include "public/fpdf_dataavail.h"
#include "public/fpdfview.h"
#include "testing/range_set.h"
 
class DownloadHintsImpl;
class FileAccessWrapper;
class FileAvailImpl;
 
class FakeFileAccess {
 public:
  explicit FakeFileAccess(FPDF_FILEACCESS* file_access);
  ~FakeFileAccess();
 
  FPDF_FILEACCESS* GetFileAccess() const;
  FX_FILEAVAIL* GetFileAvail() const;
  FX_DOWNLOADHINTS* GetDownloadHints() const;
 
  FPDF_BOOL IsDataAvail(size_t offset, size_t size) const;
  void AddSegment(size_t offset, size_t size);
 
  unsigned long GetFileSize();
 
  int GetBlock(unsigned long position, unsigned char* pBuf, unsigned long size);
  void SetRequestedDataAvailable();
  void SetWholeFileAvailable();
 
 private:
  UnownedPtr<FPDF_FILEACCESS> file_access_;
  std::unique_ptr<FileAccessWrapper> file_access_wrapper_;
  std::unique_ptr<FileAvailImpl> file_avail_;
  std::unique_ptr<DownloadHintsImpl> download_hints_;
  RangeSet available_data_;
  RangeSet requested_data_;
};
 
#endif  // TESTING_FAKE_FILE_ACCESS_H_