1 //===--------------------- TildeExpressionResolver.h ------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef LLDB_UNITTESTS_UTILITY_MOCKS_MOCK_TILDE_EXPRESSION_RESOLVER_H
11 #define LLDB_UNITTESTS_UTILITY_MOCKS_MOCK_TILDE_EXPRESSION_RESOLVER_H
12 
13 #include "lldb/Utility/TildeExpressionResolver.h"
14 
15 #include "llvm/ADT/SmallString.h"
16 #include "llvm/ADT/StringMap.h"
17 
18 namespace lldb_private {
19 class MockTildeExpressionResolver : public TildeExpressionResolver {
20   llvm::StringRef CurrentUser;
21   llvm::StringMap<llvm::StringRef> UserDirectories;
22 
23 public:
24   MockTildeExpressionResolver(llvm::StringRef CurrentUser,
25                               llvm::StringRef HomeDir);
26 
27   void AddKnownUser(llvm::StringRef User, llvm::StringRef HomeDir);
28   void Clear();
29   void SetCurrentUser(llvm::StringRef User);
30 
31   bool ResolveExact(llvm::StringRef Expr,
32                     llvm::SmallVectorImpl<char> &Output) override;
33   bool ResolvePartial(llvm::StringRef Expr, llvm::StringSet<> &Output) override;
34 };
35 } // namespace lldb_private
36 
37 #endif
38