-
Notifications
You must be signed in to change notification settings - Fork 178
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add middleware for mock MSI headers (#3714)
* Add middleware * Add frontend * Add unit test * Add license and update go mod * vendor * Fix linting * lint * Remove unneeded comments
- Loading branch information
Showing
140 changed files
with
23,587 additions
and
627 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,24 @@ | ||
| package middleware | ||
|
|
||
| // Copyright (c) Microsoft Corporation. | ||
| // Licensed under the Apache License 2.0. | ||
|
|
||
| import ( | ||
| "net/http" | ||
| "os" | ||
|
|
||
| "github.com/Azure/msi-dataplane/pkg/dataplane" | ||
| ) | ||
|
|
||
| const ( | ||
| mockIdentityURL = "https://bogus.identity.azure-int.net/subscriptions/00000000-0000-0000-0000-000000000000/resourcegroups/rg/providers/Microsoft.ApiManagement/service/test/credentials?tid=00000000-0000-0000-0000-000000000000&oid=00000000-0000-0000-0000-000000000000&aid=00000000-0000-0000-0000-000000000000" | ||
| mockTenantIDEnvVar = "MOCK_MSI_TENANT_ID" | ||
| ) | ||
|
|
||
| func MockMSIMiddleware(h http.Handler) http.Handler { | ||
| return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { | ||
| r.Header.Set(dataplane.MsiIdentityURLHeader, mockIdentityURL) | ||
| r.Header.Set(dataplane.MsiTenantHeader, os.Getenv(mockTenantIDEnvVar)) | ||
| h.ServeHTTP(w, r) | ||
| }) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,32 @@ | ||
| package middleware | ||
|
|
||
| // Copyright (c) Microsoft Corporation. | ||
| // Licensed under the Apache License 2.0. | ||
|
|
||
| import ( | ||
| "net/http" | ||
| "net/http/httptest" | ||
| "testing" | ||
|
|
||
| "github.com/Azure/msi-dataplane/pkg/dataplane" | ||
| ) | ||
|
|
||
| func TestMockMSIMiddleware(t *testing.T) { | ||
| mockTenantID := "test-tenant-id" | ||
| t.Setenv(mockTenantIDEnvVar, mockTenantID) | ||
|
|
||
| mockHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {}) | ||
| handler := MockMSIMiddleware(mockHandler) | ||
|
|
||
| req := httptest.NewRequest(http.MethodGet, "http://example.com", nil) | ||
| rr := httptest.NewRecorder() | ||
|
|
||
| handler.ServeHTTP(rr, req) | ||
|
|
||
| if req.Header.Get(dataplane.MsiIdentityURLHeader) != mockIdentityURL { | ||
| t.Errorf("Expected %s, got %s", mockIdentityURL, req.Header.Get(dataplane.MsiIdentityURLHeader)) | ||
| } | ||
| if req.Header.Get(dataplane.MsiTenantHeader) != mockTenantID { | ||
| t.Errorf("Expected %s, got %s", mockTenantID, req.Header.Get(dataplane.MsiTenantHeader)) | ||
| } | ||
| } |
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
10 changes: 10 additions & 0 deletions
10
vendor/github.com/Azure/msi-dataplane/internal/swagger/README
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
28 changes: 28 additions & 0 deletions
28
vendor/github.com/Azure/msi-dataplane/internal/swagger/autorest.md
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
3 changes: 3 additions & 0 deletions
3
vendor/github.com/Azure/msi-dataplane/internal/swagger/build.go
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
9 changes: 9 additions & 0 deletions
9
vendor/github.com/Azure/msi-dataplane/internal/swagger/constructor.go
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
10 changes: 10 additions & 0 deletions
10
vendor/github.com/Azure/msi-dataplane/internal/swagger/zz_generated_constants.go
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
Oops, something went wrong.