This repository has been archived by the owner on Mar 13, 2021. It is now read-only.
forked from cloudfoundry/guardian
-
Notifications
You must be signed in to change notification settings - Fork 1
/
fake_nstar_runner.go
137 lines (123 loc) · 4.06 KB
/
fake_nstar_runner.go
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
// This file was generated by counterfeiter
package rundmcfakes
import (
"io"
"sync"
"code.cloudfoundry.org/guardian/rundmc"
"code.cloudfoundry.org/lager"
)
type FakeNstarRunner struct {
StreamInStub func(log lager.Logger, pid int, path string, user string, tarStream io.Reader) error
streamInMutex sync.RWMutex
streamInArgsForCall []struct {
log lager.Logger
pid int
path string
user string
tarStream io.Reader
}
streamInReturns struct {
result1 error
}
StreamOutStub func(log lager.Logger, pid int, path string, user string) (io.ReadCloser, error)
streamOutMutex sync.RWMutex
streamOutArgsForCall []struct {
log lager.Logger
pid int
path string
user string
}
streamOutReturns struct {
result1 io.ReadCloser
result2 error
}
invocations map[string][][]interface{}
invocationsMutex sync.RWMutex
}
func (fake *FakeNstarRunner) StreamIn(log lager.Logger, pid int, path string, user string, tarStream io.Reader) error {
fake.streamInMutex.Lock()
fake.streamInArgsForCall = append(fake.streamInArgsForCall, struct {
log lager.Logger
pid int
path string
user string
tarStream io.Reader
}{log, pid, path, user, tarStream})
fake.recordInvocation("StreamIn", []interface{}{log, pid, path, user, tarStream})
fake.streamInMutex.Unlock()
if fake.StreamInStub != nil {
return fake.StreamInStub(log, pid, path, user, tarStream)
} else {
return fake.streamInReturns.result1
}
}
func (fake *FakeNstarRunner) StreamInCallCount() int {
fake.streamInMutex.RLock()
defer fake.streamInMutex.RUnlock()
return len(fake.streamInArgsForCall)
}
func (fake *FakeNstarRunner) StreamInArgsForCall(i int) (lager.Logger, int, string, string, io.Reader) {
fake.streamInMutex.RLock()
defer fake.streamInMutex.RUnlock()
return fake.streamInArgsForCall[i].log, fake.streamInArgsForCall[i].pid, fake.streamInArgsForCall[i].path, fake.streamInArgsForCall[i].user, fake.streamInArgsForCall[i].tarStream
}
func (fake *FakeNstarRunner) StreamInReturns(result1 error) {
fake.StreamInStub = nil
fake.streamInReturns = struct {
result1 error
}{result1}
}
func (fake *FakeNstarRunner) StreamOut(log lager.Logger, pid int, path string, user string) (io.ReadCloser, error) {
fake.streamOutMutex.Lock()
fake.streamOutArgsForCall = append(fake.streamOutArgsForCall, struct {
log lager.Logger
pid int
path string
user string
}{log, pid, path, user})
fake.recordInvocation("StreamOut", []interface{}{log, pid, path, user})
fake.streamOutMutex.Unlock()
if fake.StreamOutStub != nil {
return fake.StreamOutStub(log, pid, path, user)
} else {
return fake.streamOutReturns.result1, fake.streamOutReturns.result2
}
}
func (fake *FakeNstarRunner) StreamOutCallCount() int {
fake.streamOutMutex.RLock()
defer fake.streamOutMutex.RUnlock()
return len(fake.streamOutArgsForCall)
}
func (fake *FakeNstarRunner) StreamOutArgsForCall(i int) (lager.Logger, int, string, string) {
fake.streamOutMutex.RLock()
defer fake.streamOutMutex.RUnlock()
return fake.streamOutArgsForCall[i].log, fake.streamOutArgsForCall[i].pid, fake.streamOutArgsForCall[i].path, fake.streamOutArgsForCall[i].user
}
func (fake *FakeNstarRunner) StreamOutReturns(result1 io.ReadCloser, result2 error) {
fake.StreamOutStub = nil
fake.streamOutReturns = struct {
result1 io.ReadCloser
result2 error
}{result1, result2}
}
func (fake *FakeNstarRunner) Invocations() map[string][][]interface{} {
fake.invocationsMutex.RLock()
defer fake.invocationsMutex.RUnlock()
fake.streamInMutex.RLock()
defer fake.streamInMutex.RUnlock()
fake.streamOutMutex.RLock()
defer fake.streamOutMutex.RUnlock()
return fake.invocations
}
func (fake *FakeNstarRunner) recordInvocation(key string, args []interface{}) {
fake.invocationsMutex.Lock()
defer fake.invocationsMutex.Unlock()
if fake.invocations == nil {
fake.invocations = map[string][][]interface{}{}
}
if fake.invocations[key] == nil {
fake.invocations[key] = [][]interface{}{}
}
fake.invocations[key] = append(fake.invocations[key], args)
}
var _ rundmc.NstarRunner = new(FakeNstarRunner)