Skip to content

Commit b8bcc3a

Browse files
committed
Update managedidentity_test.go
Removed some unused test variables
1 parent 6e1211e commit b8bcc3a

File tree

1 file changed

+29
-93
lines changed

1 file changed

+29
-93
lines changed

apps/managedidentity/managedidentity_test.go

Lines changed: 29 additions & 93 deletions
Original file line numberDiff line numberDiff line change
@@ -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

9798
func getMockFilePath(t *testing.T) string {
@@ -144,64 +145,52 @@ func setCustomAzureArcFilePath(t *testing.T, path string) {
144145

145146
func 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

249238
func 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

Comments
 (0)