1/*
2 * Copyright (C) 2015, The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *     http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#ifndef AIDL_TESTS_FAKE_IO_DELEGATE_H_
18#define AIDL_TESTS_FAKE_IO_DELEGATE_H_
19
20#include <android-base/macros.h>
21
22#include <map>
23#include <memory>
24#include <set>
25#include <string>
26#include <vector>
27
28#include "io_delegate.h"
29
30namespace android {
31namespace aidl {
32namespace test {
33
34class FakeIoDelegate : public IoDelegate {
35 public:
36  FakeIoDelegate() = default;
37  virtual ~FakeIoDelegate() = default;
38
39  // Overrides from the real IoDelegate
40  std::unique_ptr<std::string> GetFileContents(
41      const std::string& filename,
42      const std::string& append_content_suffix = "") const override;
43  std::unique_ptr<LineReader> GetLineReader(
44      const std::string& file_path) const override;
45  bool FileIsReadable(const std::string& path) const override;
46  bool CreatedNestedDirs(
47      const std::string& base_dir,
48      const std::vector<std::string>& nested_subdirs) const override;
49  std::unique_ptr<CodeWriter> GetCodeWriter(
50      const std::string& file_path) const override;
51  void RemovePath(const std::string& file_path) const override;
52
53  // Methods added to facilitate testing.
54  void SetFileContents(const std::string& filename,
55                       const std::string& contents);
56  void AddStubParcelable(const std::string& canonical_name,
57                         const std::string& cpp_header);
58  void AddStubInterface(const std::string& canonical_name);
59  void AddCompoundParcelable(const std::string& canonical_name,
60                             const std::vector<std::string>& subclasses);
61  void AddBrokenFilePath(const std::string& path);
62  // Returns true iff we've previously written to |path|.
63  // When we return true, we'll set *contents to the written string.
64  bool GetWrittenContents(const std::string& path, std::string* content);
65
66  bool PathWasRemoved(const std::string& path);
67
68 private:
69  // Remove leading "./" from |path|.
70  std::string CleanPath(const std::string& path) const;
71
72  std::map<std::string, std::string> file_contents_;
73  // Normally, writing to files leaves the IoDelegate unchanged, so
74  // GetCodeWriter is a const method.  However, for tests, we break this
75  // intentionally by storing the written strings.
76  mutable std::map<std::string, std::string> written_file_contents_;
77
78  // We normally just write to strings in |written_file_contents_| but for
79  // files in this list, we simulate I/O errors.
80  std::set<std::string> broken_files_;
81  mutable std::set<std::string> removed_files_;
82
83  DISALLOW_COPY_AND_ASSIGN(FakeIoDelegate);
84};  // class FakeIoDelegate
85
86}  // namespace test
87}  // namespace android
88}  // namespace aidl
89
90#endif // AIDL_TESTS_FAKE_IO_DELEGATE_H_
91