aboutsummaryrefslogtreecommitdiffstats
path: root/brillo/minijail/mock_minijail.h
blob: a8556320814c85a7eccb3d24d9b76e26e57f951b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
// Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef LIBBRILLO_BRILLO_MINIJAIL_MOCK_MINIJAIL_H_
#define LIBBRILLO_BRILLO_MINIJAIL_MOCK_MINIJAIL_H_

#include <vector>

#include <base/macros.h>
#include <gmock/gmock.h>

#include "brillo/minijail/minijail.h"

namespace brillo {

class MockMinijail : public brillo::Minijail {
 public:
  MockMinijail() {}
  virtual ~MockMinijail() {}

  MOCK_METHOD0(New, struct minijail*());
  MOCK_METHOD1(Destroy, void(struct minijail*));

  MOCK_METHOD3(DropRoot,
               bool(struct minijail* jail,
                    const char* user,
                    const char* group));
  MOCK_METHOD2(UseSeccompFilter, void(struct minijail* jail, const char* path));
  MOCK_METHOD2(UseCapabilities, void(struct minijail* jail, uint64_t capmask));
  MOCK_METHOD1(ResetSignalMask, void(struct minijail* jail));
  MOCK_METHOD1(Enter, void(struct minijail* jail));
  MOCK_METHOD3(Run,
               bool(struct minijail* jail,
                    std::vector<char*> args,
                    pid_t* pid));
  MOCK_METHOD3(RunSync,
               bool(struct minijail* jail,
                    std::vector<char*> args,
                    int* status));
  MOCK_METHOD3(RunAndDestroy,
               bool(struct minijail* jail,
                    std::vector<char*> args,
                    pid_t* pid));
  MOCK_METHOD3(RunSyncAndDestroy,
               bool(struct minijail* jail,
                    std::vector<char*> args,
                    int* status));
  MOCK_METHOD4(RunPipeAndDestroy,
               bool(struct minijail* jail,
                    std::vector<char*> args,
                    pid_t* pid,
                    int* stdin));
  MOCK_METHOD6(RunPipesAndDestroy,
               bool(struct minijail* jail,
                    std::vector<char*> args,
                    pid_t* pid,
                    int* stdin,
                    int* stdout,
                    int* stderr));

 private:
  DISALLOW_COPY_AND_ASSIGN(MockMinijail);
};

}  // namespace brillo

#endif  // LIBBRILLO_BRILLO_MINIJAIL_MOCK_MINIJAIL_H_