Fixing tests & linting errors

Signed-off-by: Amanda Crawley <[email protected]>

Author: bala-ceg

plugins/huggingface/huggingface-cli.go

@@ -9,9 +9,9 @@ import (
 
 func HuggingFaceCLI() schema.Executable {
 	return schema.Executable{
-		Name:      "HuggingFace CLI", 
-		Runs:      []string{"huggingface-cli"},
-		DocsURL:   sdk.URL("https://huggingface.co/docs/huggingface_hub/quick-start"),
+		Name:    "HuggingFace CLI",
+		Runs:    []string{"huggingface-cli"},
+		DocsURL: sdk.URL("https://huggingface.co/docs/huggingface_hub/quick-start"),
 		NeedsAuth: needsauth.IfAll(
 			needsauth.NotForHelpOrVersion(),
 			needsauth.NotWithoutArgs(),

plugins/huggingface/test-fixtures/token

@@ -0,0 +1 @@
+hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE

plugins/huggingface/user_access_token.go

@@ -14,16 +14,16 @@ import (
 func UserAccessToken() schema.CredentialType {
 	return schema.CredentialType{
 		Name:          credname.APIToken,
-		DocsURL:       sdk.URL("https://huggingface.co/docs/hub/security-tokens"), 
-		ManagementURL: sdk.URL("https://huggingface.co/settings/tokens"), 
+		DocsURL:       sdk.URL("https://huggingface.co/docs/hub/security-tokens"),
+		ManagementURL: sdk.URL("https://huggingface.co/settings/tokens"),
 		Fields: []schema.CredentialField{
 			{
 				Name:                fieldname.UserAccessToken,
 				MarkdownDescription: "Token used to authenticate to HuggingFace.",
 				Secret:              true,
 				Composition: &schema.ValueComposition{
 					Length: 37,
-					Prefix: "hf_", 
+					Prefix: "hf_",
 					Charset: schema.Charset{
 						Uppercase: true,
 						Lowercase: true,
@@ -64,16 +64,15 @@ func UserAccessToken() schema.CredentialType {
 		Importer: importer.TryAll(
 			importer.TryEnvVarPair(defaultEnvVarMapping),
 			TryHuggingFaceTokenFile(),
-		),}
+		)}
 }
 
 var defaultEnvVarMapping = map[string]sdk.FieldName{
-	"HUGGING_FACE_HUB_TOKEN": fieldname.UserAccessToken, 
-	"HF_ENDPOINT": fieldname.Endpoint,
-	"HF_INFERENCE_ENDPOINT": fieldname.APIUrl,
+	"HUGGING_FACE_HUB_TOKEN": fieldname.UserAccessToken,
+	"HF_ENDPOINT":            fieldname.Endpoint,
+	"HF_INFERENCE_ENDPOINT":  fieldname.APIUrl,
 }
 
-
 func TryHuggingFaceTokenFile() sdk.Importer {
 	return importer.TryFile("~/.cache/huggingface/token", func(ctx context.Context, contents importer.FileContents, in sdk.ImportInput, out *sdk.ImportAttempt) {
 		fileData := string(contents)

plugins/huggingface/user_access_token_test.go

@@ -2,68 +2,47 @@ package huggingface
 
 import (
 	"testing"
-	
+
 	"github.com/1Password/shell-plugins/sdk"
 	"github.com/1Password/shell-plugins/sdk/plugintest"
 	"github.com/1Password/shell-plugins/sdk/schema/fieldname"
 )
-	
-func TestAPITokenProvisioner(t *testing.T) {
-	plugintest.TestProvisioner(t, UserAccessToken().DefaultProvisioner, map[string]plugintest.ProvisionCase{
-		"default": {
-			ItemFields: map[sdk.FieldName]string{ 
-				fieldname.UserAccessToken: "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
-				fieldname.Endpoint: "https://huggingface.co",
-				fieldname.APIUrl: "https://api-inference.huggingface.com",
-			},
-			ExpectedOutput: sdk.ProvisionOutput{
-				Environment: map[string]string{
-					"HUGGINGFACE_TOKEN": "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
-					"HF_ENDPOINT": "https://huggingface.co",
-					"HF_INFERENCE_ENDPOINT": "https://api-inference.huggingface.com",
 
+func TestAPITokenProvisioner(t *testing.T) {
+	plugintest.TestProvisioner(
+		t, UserAccessToken().DefaultProvisioner, map[string]plugintest.ProvisionCase{
+			"default": {
+				ItemFields: map[sdk.FieldName]string{
+					fieldname.UserAccessToken: "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
+					fieldname.Endpoint:        "https://huggingface.co",
+					fieldname.APIUrl:          "https://api-inference.huggingface.com",
+				},
+				ExpectedOutput: sdk.ProvisionOutput{
+					Environment: map[string]string{
+						"HUGGING_FACE_HUB_TOKEN": "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
+						"HF_ENDPOINT":            "https://huggingface.co",
+						"HF_INFERENCE_ENDPOINT":  "https://api-inference.huggingface.com",
+					},
 				},
 			},
-		},
-	})
+		})
 }
 
 func TestAPITokenImporter(t *testing.T) {
-	plugintest.TestImporter(t, UserAccessToken().Importer, map[string]plugintest.ImportCase{
-		"environment": {
-			Environment: map[string]string{ 
-				"HUGGINGFACE_TOKEN": "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
-				"HF_ENDPOINT": "https://huggingface.co",
-				"HF_INFERENCE_ENDPOINT": "https://api-inference.huggingface.com",
-			},
-			ExpectedCandidates: []sdk.ImportCandidate{
-				{
-					Fields: map[sdk.FieldName]string{
-						fieldname.UserAccessToken: "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
-						fieldname.Endpoint: "https://huggingface.co",
-						fieldname.APIUrl: "https://api-inference.huggingface.com",
-					},
-				},
-			},
-		},
-		"token file": {
-			Files: map[string]string{
-				"~/.cache/huggingface/token": plugintest.LoadFixture(t, "token"),
-			},
-			ExpectedCandidates: []sdk.ImportCandidate{
-				{
-					NameHint: "[email protected]",
-					Fields: map[sdk.FieldName]string{
-						fieldname.UserAccessToken: "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
-					},
+	plugintest.TestImporter(
+		t, UserAccessToken().Importer, map[string]plugintest.ImportCase{
+			"config file (macOS)": {
+				Files: map[string]string{
+					"~/.cache/huggingface/token": plugintest.LoadFixture(t, "token"),
 				},
-				{
-					NameHint: "[email protected]",
-					Fields: map[sdk.FieldName]string{
-						fieldname.UserAccessToken: "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
+				ExpectedCandidates: []sdk.ImportCandidate{
+					{
+						Fields: map[sdk.FieldName]string{
+							fieldname.UserAccessToken: "hf_yVvZeburdKtnwkVCWPXimmNwaFuEXAMPLE",
+						},
 					},
 				},
 			},
 		},
-	})
+	)
 }