-
Notifications
You must be signed in to change notification settings - Fork 111
/
certRotate_mock.go
143 lines (140 loc) · 12.9 KB
/
certRotate_mock.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
138
139
140
141
142
143
package main
import (
"github.com/chef/automate/api/config/deployment"
"github.com/chef/automate/lib/sshutils"
"github.com/spf13/cobra"
)
type MockCertRotateFlowImpl struct {
CertRotateFunc func(cmd *cobra.Command, args []string, flagsObj *certRotateFlags) error
CertRotateFrontendFunc func(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates) error
CertRotatePGFunc func(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates, skipFrontend bool) error
CertRotateFrontendForPGFunc func(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, timestamp string) error
GetSkipIpsListForPgRootCAPatchingFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, certs *certificates) ([]string, error)
CertRotateOSFunc func(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates, skipFrontend bool) error
CertRotateFrontendForOSFunc func(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, nodesCn string, timestamp string) error
GetSkipIpsListForOsRootCACNPatchingFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, certs *certificates, nodesCn string, flagsObj *certRotateFlags) ([]string, error)
GetFrontIpsToSkipRootCAandCNPatchingForOsFunc func(automatesConfig map[string]*deployment.AutomateConfig, newRootCA string, newCn string, node string, infra *AutomateHAInfraDetails) []string
GetFrontendIPsToSkipRootCAPatchingForPgFunc func(automatesConfig map[string]*deployment.AutomateConfig, newRootCA string, infra *AutomateHAInfraDetails) []string
PatchConfigFunc func(param *patchFnParameters) error
CopyAndExecuteFunc func(ips []string, sshUtil SSHUtil, timestamp string, remoteService string, fileName string, scriptCommands string, flagsObj *certRotateFlags) error
CopyAndExecuteConcurrentlyToFrontEndNodesFunc func(ips []string, sshConfig sshutils.SSHConfig, timestamp string, remoteService string, fileName string, scriptCommands string, flagsObj *certRotateFlags) error
ValidateEachIpFunc func(remoteService string, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags) bool
GetSshDetailsFunc func(infra *AutomateHAInfraDetails) *SSHConfig
GetIpsFunc func(remoteService string, infra *AutomateHAInfraDetails) []string
IsIPInClusterFunc func(ip string, infra *AutomateHAInfraDetails) bool
GetAllIPsFunc func(infra *AutomateHAInfraDetails) []string
GetFilteredIpsFunc func(serviceIps, skipIpsList []string) []string
CompareCurrentCertsWithNewCertsFunc func(remoteService string, newCerts *certificates, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates) []string
ComparePublicCertAndPrivateCertFunc func(newCerts *certificates, certByIpList []CertByIP, isCertsSame bool, flagsObj *certRotateFlags) []string
GetFrontEndIpsForSkippingCnAndRootCaPatchingFunc func(newRootCA, newCn, oldCn, oldRootCA, node string) bool
SkipMessagePrinterFunc func(remoteService, skipIpsMsg, nodeFlag string, skipIpsList []string)
GetCertsFunc func(infra *AutomateHAInfraDetails, flagsObj *certRotateFlags) (*certificates, error)
IsRemotePathFunc func(path string) bool
GetIPV4Func func(path string) string
GetMergerFunc func(fileName string, timestamp string, remoteType string, config string, sshUtil SSHUtil) (string, error)
CertRotateFromTemplateFunc func(clusterCertificateFile string, sshUtil SSHUtil, infra *AutomateHAInfraDetails, currentCertsInfo *certShowCertificates) error
RotatePGNodeCertsFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, pgRootCA string, pgIps *IP, concurrent bool) error
RotateOSNodeCertsFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, oss *NodeCertficate, osIp *IP, concurrent bool) error
RotateAutomateNodeCertsFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, certToml *CertificateToml, a2Ip *IP) error
RotateChefServerNodeCertsFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, certToml *CertificateToml, csIp *IP) error
RotateClusterFrontendCertificatesFunc func(infra *AutomateHAInfraDetails, sshUtil SSHUtil, flagsObj certRotateFlags, currentCertsInfo *certShowCertificates, certToml *CertificateToml) error
}
func (mcrf *MockCertRotateFlowImpl) CertRotate(cmd *cobra.Command, args []string, flagsObj *certRotateFlags) error {
return mcrf.CertRotateFunc(cmd, args, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) CertRotateFrontend(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates) error {
return mcrf.CertRotateFrontendFunc(sshUtil, certs, infra, flagsObj, currentCertsInfo)
}
func (mcrf *MockCertRotateFlowImpl) CertRotatePG(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates, skipFrontend bool) error {
return mcrf.CertRotatePGFunc(sshUtil, certs, infra, flagsObj, currentCertsInfo, skipFrontend)
}
func (mcrf *MockCertRotateFlowImpl) CertRotateFrontendForPG(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, timestamp string) error {
return mcrf.CertRotateFrontendForPGFunc(sshUtil, certs, infra, flagsObj, timestamp)
}
func (mcrf *MockCertRotateFlowImpl) GetSkipIpsListForPgRootCAPatching(infra *AutomateHAInfraDetails, sshUtil SSHUtil, certs *certificates) ([]string, error) {
return mcrf.GetSkipIpsListForPgRootCAPatchingFunc(infra, sshUtil, certs)
}
func (mcrf *MockCertRotateFlowImpl) CertRotateOS(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates, skipFrontend bool) error {
return mcrf.CertRotateOSFunc(sshUtil, certs, infra, flagsObj, currentCertsInfo, skipFrontend)
}
func (mcrf *MockCertRotateFlowImpl) CertRotateFrontendForOS(sshUtil SSHUtil, certs *certificates, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags, nodesCn string, timestamp string) error {
return mcrf.CertRotateFrontendForOSFunc(sshUtil, certs, infra, flagsObj, nodesCn, timestamp)
}
func (mcrf *MockCertRotateFlowImpl) GetSkipIpsListForOsRootCACNPatching(infra *AutomateHAInfraDetails, sshUtil SSHUtil, certs *certificates, nodesCn string, flagsObj *certRotateFlags) ([]string, error) {
return mcrf.GetSkipIpsListForOsRootCACNPatchingFunc(infra, sshUtil, certs, nodesCn, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) GetFrontIpsToSkipRootCAandCNPatchingForOs(automatesConfig map[string]*deployment.AutomateConfig, newRootCA string, newCn string, node string, infra *AutomateHAInfraDetails) []string {
return mcrf.GetFrontIpsToSkipRootCAandCNPatchingForOsFunc(automatesConfig, newRootCA, newCn, node, infra)
}
func (mcrf *MockCertRotateFlowImpl) GetFrontendIPsToSkipRootCAPatchingForPg(automatesConfig map[string]*deployment.AutomateConfig, newRootCA string, infra *AutomateHAInfraDetails) []string {
return mcrf.GetFrontendIPsToSkipRootCAPatchingForPgFunc(automatesConfig, newRootCA, infra)
}
func (mcrf *MockCertRotateFlowImpl) PatchConfig(param *patchFnParameters) error {
return mcrf.PatchConfigFunc(param)
}
func (mcrf *MockCertRotateFlowImpl) CopyAndExecute(ips []string, sshUtil SSHUtil, timestamp string, remoteService string, fileName string, scriptCommands string, flagsObj *certRotateFlags) error {
return mcrf.CopyAndExecuteFunc(ips, sshUtil, timestamp, remoteService, fileName, scriptCommands, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) CopyAndExecuteConcurrentlyToFrontEndNodes(ips []string, sshConfig sshutils.SSHConfig, timestamp string, remoteService string, fileName string, scriptCommands string, flagsObj *certRotateFlags) error {
return mcrf.CopyAndExecuteConcurrentlyToFrontEndNodesFunc(ips, sshConfig, timestamp, remoteService, fileName, scriptCommands, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) ValidateEachIp(remoteService string, infra *AutomateHAInfraDetails, flagsObj *certRotateFlags) bool {
return mcrf.ValidateEachIpFunc(remoteService, infra, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) GetSshDetails(infra *AutomateHAInfraDetails) *SSHConfig {
return mcrf.GetSshDetailsFunc(infra)
}
func (mcrf *MockCertRotateFlowImpl) GetIps(remoteService string, infra *AutomateHAInfraDetails) []string {
return mcrf.GetIpsFunc(remoteService, infra)
}
func (mcrf *MockCertRotateFlowImpl) IsIPInCluster(ip string, infra *AutomateHAInfraDetails) bool {
return mcrf.IsIPInClusterFunc(ip, infra)
}
func (mcrf *MockCertRotateFlowImpl) GetAllIPs(infra *AutomateHAInfraDetails) []string {
return mcrf.GetAllIPsFunc(infra)
}
func (mcrf *MockCertRotateFlowImpl) GetFilteredIps(serviceIps, skipIpsList []string) []string {
return mcrf.GetFilteredIpsFunc(serviceIps, skipIpsList)
}
func (mcrf *MockCertRotateFlowImpl) CompareCurrentCertsWithNewCerts(remoteService string, newCerts *certificates, flagsObj *certRotateFlags, currentCertsInfo *certShowCertificates) []string {
return mcrf.CompareCurrentCertsWithNewCertsFunc(remoteService, newCerts, flagsObj, currentCertsInfo)
}
func (mcrf *MockCertRotateFlowImpl) ComparePublicCertAndPrivateCert(newCerts *certificates, certByIpList []CertByIP, isCertsSame bool, flagsObj *certRotateFlags) []string {
return mcrf.ComparePublicCertAndPrivateCertFunc(newCerts, certByIpList, isCertsSame, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) GetFrontEndIpsForSkippingCnAndRootCaPatching(newRootCA, newCn, oldCn, oldRootCA, node string) bool {
return mcrf.GetFrontEndIpsForSkippingCnAndRootCaPatchingFunc(newRootCA, newCn, oldCn, oldRootCA, node)
}
func (mcrf *MockCertRotateFlowImpl) SkipMessagePrinter(remoteService, skipIpsMsg, nodeFlag string, skipIpsList []string) {
mcrf.SkipMessagePrinterFunc(remoteService, skipIpsMsg, nodeFlag, skipIpsList)
}
func (mcrf *MockCertRotateFlowImpl) GetCerts(infra *AutomateHAInfraDetails, flagsObj *certRotateFlags) (*certificates, error) {
return mcrf.GetCertsFunc(infra, flagsObj)
}
func (mcrf *MockCertRotateFlowImpl) IsRemotePath(path string) bool {
return mcrf.IsRemotePathFunc(path)
}
func (mcrf *MockCertRotateFlowImpl) GetIPV4(path string) string {
return mcrf.GetIPV4Func(path)
}
func (mcrf *MockCertRotateFlowImpl) GetMerger(fileName string, timestamp string, remoteType string, config string, sshUtil SSHUtil) (string, error) {
return mcrf.GetMergerFunc(fileName, timestamp, remoteType, config, sshUtil)
}
func (mcrf *MockCertRotateFlowImpl) CertRotateFromTemplate(clusterCertificateFile string, sshUtil SSHUtil, infra *AutomateHAInfraDetails, currentCertsInfo *certShowCertificates) error {
return mcrf.CertRotateFromTemplateFunc(clusterCertificateFile, sshUtil, infra, currentCertsInfo)
}
func (mcrf *MockCertRotateFlowImpl) RotatePGNodeCerts(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, pgRootCA string, pgIps *IP, concurrent bool) error {
return mcrf.RotatePGNodeCertsFunc(infra, sshUtil, currentCertsInfo, pgRootCA, pgIps, concurrent)
}
func (mcrf *MockCertRotateFlowImpl) RotateOSNodeCerts(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, oss *NodeCertficate, osIp *IP, concurrent bool) error {
return mcrf.RotateOSNodeCertsFunc(infra, sshUtil, currentCertsInfo, oss, osIp, concurrent)
}
func (mcrf *MockCertRotateFlowImpl) RotateAutomateNodeCerts(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, certToml *CertificateToml, a2Ip *IP) error {
return mcrf.RotateAutomateNodeCertsFunc(infra, sshUtil, currentCertsInfo, certToml, a2Ip)
}
func (mcrf *MockCertRotateFlowImpl) RotateChefServerNodeCerts(infra *AutomateHAInfraDetails, sshUtil SSHUtil, currentCertsInfo *certShowCertificates, certToml *CertificateToml, csIp *IP) error {
return mcrf.RotateChefServerNodeCertsFunc(infra, sshUtil, currentCertsInfo, certToml, csIp)
}
func (mcrf *MockCertRotateFlowImpl) RotateClusterFrontendCertificates(infra *AutomateHAInfraDetails, sshUtil SSHUtil, flagsObj certRotateFlags, currentCertsInfo *certShowCertificates, certToml *CertificateToml) error {
return mcrf.RotateClusterFrontendCertificatesFunc(infra, sshUtil, flagsObj, currentCertsInfo, certToml)
}