@@ -50,15 +50,12 @@ var (
50
50
cloudtopHost = flag .Bool ("cloudtop-host" , false , "The local host is cloudtop, a kind of googler machine with special requirements to access GCP" )
51
51
extraDriverFlags = flag .String ("extra-driver-flags" , "" , "Extra flags to pass to the driver" )
52
52
enableConfidentialCompute = flag .Bool ("enable-confidential-compute" , false , "Create VMs with confidential compute mode. This uses NVMe devices" )
53
- hdMachineType = flag .String ("hyperdisk-machine-type" , "c3-standard-4" , "Type of machine to provision instance on" )
54
- hdMinCpuPlatform = flag .String ("hyperdisk-min-cpu-platform" , "sapphirerapids" , "Minimum CPU architecture" )
55
-
56
- testContexts = []* remote.TestContext {}
57
- hyperdiskTestContexts = []* remote.TestContext {}
58
- computeService * compute.Service
59
- computeAlphaService * computealpha.Service
60
- computeBetaService * computebeta.Service
61
- kmsClient * cloudkms.KeyManagementClient
53
+
54
+ testContexts = []* remote.TestContext {}
55
+ computeService * compute.Service
56
+ computeAlphaService * computealpha.Service
57
+ computeBetaService * computebeta.Service
58
+ kmsClient * cloudkms.KeyManagementClient
62
59
)
63
60
64
61
const localSSDCount int64 = 2
@@ -76,9 +73,7 @@ func TestE2E(t *testing.T) {
76
73
var _ = BeforeSuite (func () {
77
74
var err error
78
75
tcc := make (chan * remote.TestContext )
79
- hdtcc := make (chan * remote.TestContext )
80
76
defer close (tcc )
81
- defer close (hdtcc )
82
77
83
78
zones := strings .Split (* zones , "," )
84
79
// Create 2 instances for each zone as we need 2 instances each zone for certain test cases
@@ -113,11 +108,7 @@ var _ = BeforeSuite(func() {
113
108
for _ , zone := range zones {
114
109
go func (curZone string ) {
115
110
defer GinkgoRecover ()
116
- tcc <- NewDefaultTestContext (curZone , strconv .Itoa (randInt ))
117
- }(zone )
118
- go func (curZone string ) {
119
- defer GinkgoRecover ()
120
- hdtcc <- NewTestContext (curZone , * hdMinCpuPlatform , * hdMachineType , strconv .Itoa (randInt ))
111
+ tcc <- NewTestContext (curZone , strconv .Itoa (randInt ))
121
112
}(zone )
122
113
}
123
114
}
@@ -129,9 +120,6 @@ var _ = BeforeSuite(func() {
129
120
tc := <- tcc
130
121
testContexts = append (testContexts , tc )
131
122
klog .Infof ("Added TestContext for node %s" , tc .Instance .GetName ())
132
- tc = <- hdtcc
133
- hyperdiskTestContexts = append (hyperdiskTestContexts , tc )
134
- klog .Infof ("Added TestContext for node %s" , tc .Instance .GetName ())
135
123
}
136
124
})
137
125
@@ -156,21 +144,17 @@ func getDriverConfig() testutils.DriverConfig {
156
144
}
157
145
}
158
146
159
- func NewDefaultTestContext (zone string , instanceNumber string ) * remote.TestContext {
160
- return NewTestContext (zone , * minCpuPlatform , * machineType , instanceNumber )
161
- }
162
-
163
- func NewTestContext (zone , minCpuPlatform , machineType string , instanceNumber string ) * remote.TestContext {
164
- nodeID := fmt .Sprintf ("%s-%s-%s-%s" , * vmNamePrefix , zone , machineType , instanceNumber )
147
+ func NewTestContext (zone string , instanceNumber string ) * remote.TestContext {
148
+ nodeID := fmt .Sprintf ("%s-%s-%s" , * vmNamePrefix , zone , instanceNumber )
165
149
klog .Infof ("Setting up node %s" , nodeID )
166
150
167
151
instanceConfig := remote.InstanceConfig {
168
152
Project : * project ,
169
153
Architecture : * architecture ,
170
- MinCpuPlatform : minCpuPlatform ,
154
+ MinCpuPlatform : * minCpuPlatform ,
171
155
Zone : zone ,
172
156
Name : nodeID ,
173
- MachineType : machineType ,
157
+ MachineType : * machineType ,
174
158
ServiceAccount : * serviceAccount ,
175
159
ImageURL : * imageURL ,
176
160
CloudtopHost : * cloudtopHost ,
@@ -200,12 +184,12 @@ func NewTestContext(zone, minCpuPlatform, machineType string, instanceNumber str
200
184
pkgs := []string {"lvm2" , "mdadm" , "grep" , "coreutils" }
201
185
err = testutils .InstallDependencies (i , pkgs )
202
186
if err != nil {
203
- klog .Fatalf ("Failed to install dependency package on node %v: error : %v" , i .GetNodeID (), err )
187
+ klog .Errorf ("Failed to install dependency package on node %v: error : %v" , i .GetNodeID (), err )
204
188
}
205
189
206
190
err = testutils .SetupDataCachingConfig (i )
207
191
if err != nil {
208
- klog .Fatalf ("Failed to setup data cache required config error %v" , err )
192
+ klog .Errorf ("Failed to setup data cache required config error %v" , err )
209
193
}
210
194
klog .Infof ("Creating new driver and client for node %s" , i .GetName ())
211
195
tc , err := testutils .GCEClientAndDriverSetup (i , getDriverConfig ())
0 commit comments