@@ -8,51 +8,18 @@ package azsecrets_test
88
99import (
1010 "context"
11- "encoding/json"
12- "errors"
1311 "fmt"
1412 "strings"
1513 "testing"
1614 "time"
1715
18- "github.com/Azure/azure-sdk-for-go/sdk/azcore"
1916 "github.com/Azure/azure-sdk-for-go/sdk/azcore/policy"
2017 "github.com/Azure/azure-sdk-for-go/sdk/azcore/to"
2118 "github.com/Azure/azure-sdk-for-go/sdk/internal/mock"
22- "github.com/Azure/azure-sdk-for-go/sdk/internal/recording"
2319 "github.com/Azure/azure-sdk-for-go/sdk/security/keyvault/azsecrets"
2420 "github.com/stretchr/testify/require"
2521)
2622
27- // pollStatus calls a function until it stops returning a response error with the given status code.
28- // If this takes more than 2 minutes, it fails the test.
29- func pollStatus (t * testing.T , expectedStatus int , fn func () error ) {
30- var err error
31- for i := 0 ; i < 12 ; i ++ {
32- err = fn ()
33- var respErr * azcore.ResponseError
34- if ! (errors .As (err , & respErr ) && respErr .StatusCode == expectedStatus ) {
35- break
36- }
37- if i < 11 {
38- recording .Sleep (10 * time .Second )
39- }
40- }
41- require .NoError (t , err )
42- }
43-
44- type serdeModel interface {
45- json.Marshaler
46- json.Unmarshaler
47- }
48-
49- func testSerde [T serdeModel ](t * testing.T , model T ) {
50- data , err := model .MarshalJSON ()
51- require .NoError (t , err )
52- err = model .UnmarshalJSON (data )
53- require .NoError (t , err )
54- }
55-
5623func TestBackupRestore (t * testing.T ) {
5724 client := startTest (t )
5825
@@ -114,7 +81,7 @@ func TestCRUD(t *testing.T) {
11481 require .Equal (t , setParams .Value , setResp .Value )
11582 require .Equal (t , name , setResp .ID .Name ())
11683 require .NotEmpty (t , setResp .ID .Version ())
117- testSerde (t , & setResp .SecretBundle )
84+ testSerde (t , & setResp .Secret )
11885
11986 getResp , err := client .GetSecret (context .Background (), setResp .ID .Name (), "" , nil )
12087 require .NoError (t , err )
@@ -128,13 +95,13 @@ func TestCRUD(t *testing.T) {
12895 require .Equal (t , setParams .Tags , getResp .Tags )
12996 require .Equal (t , setParams .Value , getResp .Value )
13097
131- updateParams := azsecrets.UpdateSecretParameters {
98+ updateParams := azsecrets.UpdateSecretPropertiesParameters {
13299 SecretAttributes : & azsecrets.SecretAttributes {
133100 Expires : to .Ptr (time .Date (2040 , 1 , 1 , 1 , 1 , 1 , 0 , time .UTC )),
134101 },
135102 }
136103 testSerde (t , & updateParams )
137- updateResp , err := client .UpdateSecret (context .Background (), name , setResp .ID .Version (), updateParams , nil )
104+ updateResp , err := client .UpdateSecretProperties (context .Background (), name , setResp .ID .Version (), updateParams , nil )
138105 require .NoError (t , err )
139106 require .Equal (t , setParams .ContentType , updateResp .ContentType )
140107 require .Equal (t , setResp .ID , updateResp .ID )
@@ -153,7 +120,7 @@ func TestCRUD(t *testing.T) {
153120 require .Equal (t , setParams .Tags , deleteResp .Tags )
154121 require .Equal (t , name , deleteResp .ID .Name ())
155122 require .Equal (t , updateResp .ID .Version (), deleteResp .ID .Version ())
156- testSerde (t , & deleteResp .DeletedSecretBundle )
123+ testSerde (t , & deleteResp .DeletedSecret )
157124 pollStatus (t , 404 , func () error {
158125 _ , err := client .GetDeletedSecret (context .Background (), name , nil )
159126 return err
@@ -212,7 +179,7 @@ func TestDisableChallengeResourceVerification(t *testing.T) {
212179 }
213180 client , err := azsecrets .NewClient (vaultURL , & FakeCredential {}, options )
214181 require .NoError (t , err )
215- pager := client .NewListSecretsPager (nil )
182+ pager := client .NewListSecretPropertiesPager (nil )
216183 _ , err = pager .NextPage (context .Background ())
217184 if test .err {
218185 require .Error (t , err )
@@ -270,11 +237,11 @@ func TestListDeletedSecrets(t *testing.T) {
270237 })
271238
272239 expected := map [string ]struct {}{secret1 : {}, secret2 : {}}
273- pager := client .NewListDeletedSecretsPager (& azsecrets.ListDeletedSecretsOptions {MaxResults : to .Ptr (int32 (1 ))})
240+ pager := client .NewListDeletedSecretPropertiesPager (& azsecrets.ListDeletedSecretPropertiesOptions {MaxResults : to .Ptr (int32 (1 ))})
274241 for pager .More () && len (expected ) > 0 {
275242 page , err := pager .NextPage (context .Background ())
276243 require .NoError (t , err )
277- testSerde (t , & page .DeletedSecretListResult )
244+ testSerde (t , & page .DeletedSecretPropertiesListResult )
278245 for _ , secret := range page .Value {
279246 testSerde (t , secret )
280247 delete (expected , secret .ID .Name ())
@@ -298,11 +265,11 @@ func TestListSecrets(t *testing.T) {
298265 defer cleanUpSecret (t , client , name )
299266 }
300267
301- pager := client .NewListSecretsPager (& azsecrets.ListSecretsOptions {MaxResults : to .Ptr (int32 (1 ))})
268+ pager := client .NewListSecretPropertiesPager (& azsecrets.ListSecretPropertiesOptions {MaxResults : to .Ptr (int32 (1 ))})
302269 for pager .More () {
303270 page , err := pager .NextPage (context .Background ())
304271 require .NoError (t , err )
305- testSerde (t , & page .SecretListResult )
272+ testSerde (t , & page .SecretPropertiesListResult )
306273 for _ , secret := range page .Value {
307274 testSerde (t , secret )
308275 if strings .HasPrefix (secret .ID .Name (), "listsecrets" ) {
@@ -335,11 +302,11 @@ func TestListSecretVersions(t *testing.T) {
335302 }
336303 defer cleanUpSecret (t , client , name )
337304
338- pager := client .NewListSecretVersionsPager (name , & azsecrets.ListSecretVersionsOptions {MaxResults : to .Ptr (int32 (1 ))})
305+ pager := client .NewListSecretPropertiesVersionsPager (name , & azsecrets.ListSecretPropertiesVersionsOptions {MaxResults : to .Ptr (int32 (1 ))})
339306 for pager .More () {
340307 page , err := pager .NextPage (context .Background ())
341308 require .NoError (t , err )
342- testSerde (t , & page .SecretListResult )
309+ testSerde (t , & page .SecretPropertiesListResult )
343310 for i , secret := range page .Value {
344311 testSerde (t , secret )
345312 if i > 0 {
@@ -377,7 +344,7 @@ func TestNameRequired(t *testing.T) {
377344 require .EqualError (t , err , expected )
378345 _ , err = client .SetSecret (context .Background (), "" , azsecrets.SetSecretParameters {}, nil )
379346 require .EqualError (t , err , expected )
380- _ , err = client .UpdateSecret (context .Background (), "" , "" , azsecrets.UpdateSecretParameters {}, nil )
347+ _ , err = client .UpdateSecretProperties (context .Background (), "" , "" , azsecrets.UpdateSecretPropertiesParameters {}, nil )
381348 require .EqualError (t , err , expected )
382349}
383350
0 commit comments