@@ -92,6 +92,7 @@ func createMockFile(t *testing.T, path string, size int64) {
9292 if _ , err := f .WriteString ("secret file data" ); err != nil {
9393 t .Fatalf ("failed to write to file: %v" , err )
9494 }
95+ t .Cleanup (func () { os .Remove (path ) })
9596}
9697
9798func getMockFilePath (t * testing.T ) string {
@@ -144,64 +145,52 @@ func setCustomAzureArcFilePath(t *testing.T, path string) {
144145
145146func TestGetSource (t * testing.T ) {
146147 testCases := []struct {
147- name string
148- source Source
149- expectedSource Source
148+ name string
149+ source Source
150150 }{
151151 {
152- name : "testAzureArcSystemAssigned" ,
153- source : AzureArc ,
154- expectedSource : AzureArc ,
152+ name : "testAzureArcSystemAssigned" ,
153+ source : AzureArc ,
155154 },
156155 {
157- name : "testAzureArcUserClientAssigned" ,
158- source : AzureArc ,
159- expectedSource : AzureArc ,
156+ name : "testAzureArcUserClientAssigned" ,
157+ source : AzureArc ,
160158 },
161159 {
162- name : "testAzureArcUserResourceAssigned" ,
163- source : AzureArc ,
164- expectedSource : AzureArc ,
160+ name : "testAzureArcUserResourceAssigned" ,
161+ source : AzureArc ,
165162 },
166163 {
167- name : "testAzureArcUserObjectAssigned" ,
168- source : AzureArc ,
169- expectedSource : AzureArc ,
164+ name : "testAzureArcUserObjectAssigned" ,
165+ source : AzureArc ,
170166 },
171167 {
172- name : "testDefaultToImds" ,
173- source : DefaultToIMDS ,
174- expectedSource : DefaultToIMDS ,
168+ name : "testDefaultToImds" ,
169+ source : DefaultToIMDS ,
175170 },
176171 {
177- name : "testDefaultToImdsClientAssigned" ,
178- source : DefaultToIMDS ,
179- expectedSource : DefaultToIMDS ,
172+ name : "testDefaultToImdsClientAssigned" ,
173+ source : DefaultToIMDS ,
180174 },
181175 {
182- name : "testDefaultToImdsResourceAssigned" ,
183- source : DefaultToIMDS ,
184- expectedSource : DefaultToIMDS ,
176+ name : "testDefaultToImdsResourceAssigned" ,
177+ source : DefaultToIMDS ,
185178 },
186179 {
187- name : "testDefaultToImdsObjectAssigned" ,
188- source : DefaultToIMDS ,
189- expectedSource : DefaultToIMDS ,
180+ name : "testDefaultToImdsObjectAssigned" ,
181+ source : DefaultToIMDS ,
190182 },
191183 {
192- name : "testDefaultToImdsEmptyEndpoint" ,
193- source : DefaultToIMDS ,
194- expectedSource : DefaultToIMDS ,
184+ name : "testDefaultToImdsEmptyEndpoint" ,
185+ source : DefaultToIMDS ,
195186 },
196187 {
197- name : "testDefaultToImdsLinux" ,
198- source : DefaultToIMDS ,
199- expectedSource : DefaultToIMDS ,
188+ name : "testDefaultToImdsLinux" ,
189+ source : DefaultToIMDS ,
200190 },
201191 {
202- name : "testDefaultToImdsEmptyEndpointLinux" ,
203- source : DefaultToIMDS ,
204- expectedSource : DefaultToIMDS ,
192+ name : "testDefaultToImdsEmptyEndpointLinux" ,
193+ source : DefaultToIMDS ,
205194 },
206195 }
207196 for _ , testCase := range testCases {
@@ -214,8 +203,8 @@ func TestGetSource(t *testing.T) {
214203 if err != nil {
215204 t .Fatalf ("error while getting source: %s" , err .Error ())
216205 }
217- if actualSource != testCase .expectedSource {
218- t .Errorf (errorExpectedButGot , testCase .expectedSource , actualSource )
206+ if actualSource != testCase .source {
207+ t .Errorf (errorExpectedButGot , testCase .source , actualSource )
219208 }
220209 })
221210 }
@@ -248,9 +237,8 @@ func TestAzureArcReturnsWhenHimdsFound(t *testing.T) {
248237
249238func TestIMDSAcquireTokenReturnsTokenSuccess (t * testing.T ) {
250239 testCases := []struct {
251- resource string
252- miType ID
253- apiVersion string
240+ resource string
241+ miType ID
254242 }{
255243 {resource : resource , miType : SystemAssigned ()},
256244 {resource : resourceDefaultSuffix , miType : SystemAssigned ()},
@@ -459,7 +447,6 @@ func TestAzureArcAcquireTokenReturnsToken(t *testing.T) {
459447 createMockFile (t , mockFilePath , 0 )
460448 }
461449
462- t .Cleanup (func () { os .Remove (mockFilePath ) })
463450 headers := http.Header {}
464451 for k , v := range testCase .request .headers {
465452 headers .Set (k , v )
@@ -691,54 +678,3 @@ func TestAzureArcUserAssignedFailure(t *testing.T) {
691678 })
692679 }
693680}
694-
695- func TestValidateAzureArcEnvironment (t * testing.T ) {
696- testCases := []struct {
697- name string
698- identityEndpoint string
699- imdsEndpoint string
700- platform string
701- expectedResult bool
702- }{
703- {
704- name : "Both endpoints provided" ,
705- identityEndpoint : "endpoint" ,
706- imdsEndpoint : "endpoint" ,
707- platform : runtime .GOOS ,
708- expectedResult : true ,
709- },
710- {
711- name : "Only identityEndpoint provided" ,
712- identityEndpoint : "endpoint" ,
713- platform : runtime .GOOS ,
714- expectedResult : false ,
715- },
716- {
717- name : "Only imdsEndpoint provided" ,
718- imdsEndpoint : "endpoint" ,
719- platform : runtime .GOOS ,
720- expectedResult : false ,
721- },
722- {
723- name : "No endpoints provided" ,
724- platform : runtime .GOOS ,
725- expectedResult : false ,
726- },
727- {
728- name : "Platform not supported" ,
729- platform : "darwin" ,
730- expectedResult : false ,
731- },
732- }
733-
734- for _ , tc := range testCases {
735- t .Run (tc .name , func (t * testing.T ) {
736- setCustomAzureArcFilePath (t , fakeAzureArcFilePath )
737-
738- result := isAzureArcEnvironment (tc .identityEndpoint , tc .imdsEndpoint , tc .platform )
739- if result != tc .expectedResult {
740- t .Fatalf (errorExpectedButGot , tc .expectedResult , result )
741- }
742- })
743- }
744- }
0 commit comments