@@ -79,8 +79,10 @@ func (f *peerEndpointLeaseReconciler) SetKeys(servers []serverInfo) error {
7979
8080func TestPeerEndpointLeaseReconciler (t * testing.T ) {
8181// enable feature flags
82- featuregatetesting .SetFeatureGateDuringTest (t , utilfeature .DefaultFeatureGate , features .APIServerIdentity , true )
83- featuregatetesting .SetFeatureGateDuringTest (t , utilfeature .DefaultFeatureGate , features .StorageVersionAPI , true )
82+ featuregatetesting .SetFeatureGatesDuringTest (t , utilfeature .DefaultFeatureGate , featuregatetesting.FeatureOverrides {
83+ features .APIServerIdentity : true ,
84+ features .StorageVersionAPI : true ,
85+ })
8486
8587server , sc := etcd3testing .NewUnsecuredEtcd3TestClientServer (t )
8688t .Cleanup (func () { server .Terminate (t ) })
@@ -189,8 +191,10 @@ func TestPeerEndpointLeaseReconciler(t *testing.T) {
189191
190192func TestPeerLeaseRemoveEndpoints (t * testing.T ) {
191193// enable feature flags
192- featuregatetesting .SetFeatureGateDuringTest (t , utilfeature .DefaultFeatureGate , features .APIServerIdentity , true )
193- featuregatetesting .SetFeatureGateDuringTest (t , utilfeature .DefaultFeatureGate , features .StorageVersionAPI , true )
194+ featuregatetesting .SetFeatureGatesDuringTest (t , utilfeature .DefaultFeatureGate , featuregatetesting.FeatureOverrides {
195+ features .APIServerIdentity : true ,
196+ features .StorageVersionAPI : true ,
197+ })
194198
195199server , sc := etcd3testing .NewUnsecuredEtcd3TestClientServer (t )
196200t .Cleanup (func () { server .Terminate (t ) })
0 commit comments