fix: fix the bugs in the unit tests

ratify-project · Oct 3, 2024 · aa6be1a · aa6be1a
1 parent 2c3e52a
commit aa6be1a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pkg/common/oras/authprovider/azure/azureidentity_test.go b/pkg/common/oras/authprovider/azure/azureidentity_test.go
@@ -191,6 +191,11 @@ func TestMIProvide_RefreshAAD(t *testing.T) {
 
 func TestMIProvide_Failure_InvalidHostName(t *testing.T) {
 	provider := &MIAuthProvider{
+		tenantID: "test_tenant",
+		clientID: "test_client",
+		identityToken: azcore.AccessToken{
+			Token: "test_token",
+		},
 		getRegistryHost: func(_ string) (string, error) {
 			return "", errors.New("invalid hostname")
 		},

diff --git a/pkg/common/oras/authprovider/azure/azureworkloadidentity_test.go b/pkg/common/oras/authprovider/azure/azureworkloadidentity_test.go
@@ -247,6 +247,11 @@ func TestWIProvide_RefreshAAD(t *testing.T) {
 
 func TestWIProvide_Failure_InvalidHostName(t *testing.T) {
 	provider := &WIAuthProvider{
+		aadToken: confidential.AuthResult{
+			AccessToken: "mockToken",
+			ExpiresOn:   time.Now(),
+		},
+		tenantID: "mockTenantID",
 		getRegistryHost: func(_ string) (string, error) {
 			return "", errors.New("invalid hostname")
 		},