This repository has been archived by the owner on Feb 23, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
mockexec.go
78 lines (61 loc) · 2.61 KB
/
mockexec.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
/*
Copyright 2016 The Rook Authors. All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package test
import (
"os/exec"
"time"
)
// MockExecutor mocks all the exec commands
type MockExecutor struct {
MockExecuteCommand func(command string, arg ...string) error
MockExecuteCommandWithEnv func(env []string, command string, arg ...string) error
MockStartExecuteCommand func(command string, arg ...string) (*exec.Cmd, error)
MockExecuteCommandWithOutput func(command string, arg ...string) (string, error)
MockExecuteCommandWithCombinedOutput func(command string, arg ...string) (string, error)
MockExecuteCommandWithTimeout func(timeout time.Duration, command string, arg ...string) (string, error)
}
// ExecuteCommand mocks ExecuteCommand
func (e *MockExecutor) ExecuteCommand(command string, arg ...string) error {
if e.MockExecuteCommand != nil {
return e.MockExecuteCommand(command, arg...)
}
return nil
}
// ExecuteCommandWithEnv mocks ExecuteCommandWithEnv
func (e *MockExecutor) ExecuteCommandWithEnv(env []string, command string, arg ...string) error {
if e.MockExecuteCommandWithEnv != nil {
return e.MockExecuteCommandWithEnv(env, command, arg...)
}
return nil
}
// ExecuteCommandWithOutput mocks ExecuteCommandWithOutput
func (e *MockExecutor) ExecuteCommandWithOutput(command string, arg ...string) (string, error) {
if e.MockExecuteCommandWithOutput != nil {
return e.MockExecuteCommandWithOutput(command, arg...)
}
return "", nil
}
// ExecuteCommandWithTimeout mocks ExecuteCommandWithTimeout
func (e *MockExecutor) ExecuteCommandWithTimeout(timeout time.Duration, command string, arg ...string) (string, error) {
if e.MockExecuteCommandWithTimeout != nil {
return e.MockExecuteCommandWithTimeout(time.Second, command, arg...)
}
return "", nil
}
// ExecuteCommandWithCombinedOutput mocks ExecuteCommandWithCombinedOutput
func (e *MockExecutor) ExecuteCommandWithCombinedOutput(command string, arg ...string) (string, error) {
if e.MockExecuteCommandWithCombinedOutput != nil {
return e.MockExecuteCommandWithCombinedOutput(command, arg...)
}
return "", nil
}