feat(modelarmor): code refactor

tirthrajsinh-zala-crest · tirthrajsinh-zala-crest · commit a9f0e31b568d · 2025-04-11T18:46:35.000+05:30
diff --git a/modelarmor/modelarmor_test.go b/modelarmor/modelarmor_test.go
@@ -393,12 +393,12 @@ MCBSCi9JbmZvIDEgMCBSPj4Kc3RhcnR4cmVmCjE5ODQ0CiUlRU9GCg=="`
 	templateID := fmt.Sprintf("test-model-armor-%s", uuid.New().String())
 
 	var b bytes.Buffer
-	if _, err := createModelArmorTemplate(&b, tc.ProjectID, "us-central1", templateID); err != nil {
+	if _, err := createModelArmorTemplate(&b, tc.ProjectID, testLocation(t), templateID); err != nil {
 		t.Fatal(err)
 	}
-	defer testCleanupTemplate(t, fmt.Sprintf("projects/%s/locations/%s/templates/%s", tc.ProjectID, "us-central1", templateID))
+	defer testCleanupTemplate(t, fmt.Sprintf("projects/%s/locations/%s/templates/%s", tc.ProjectID, testLocation(t), templateID))
 
-	if _, err := screenPDFFile(&b, tc.ProjectID, "us-central1", templateID, pdfContentBase); err != nil {
+	if _, err := screenPDFFile(&b, tc.ProjectID, testLocation(t), templateID, pdfContentBase); err != nil {
 		t.Fatal(err)
 	}
 

Original file line number	Diff line number	Diff line change
@@ -393,12 +393,12 @@ MCBSCi9JbmZvIDEgMCBSPj4Kc3RhcnR4cmVmCjE5ODQ0CiUlRU9GCg=="`
`393`	`393`	`templateID := fmt.Sprintf("test-model-armor-%s", uuid.New().String())`
`394`	`394`
`395`	`395`	`var b bytes.Buffer`
`396`		`- if _, err := createModelArmorTemplate(&b, tc.ProjectID, "us-central1", templateID); err != nil {`
	`396`	`+ if _, err := createModelArmorTemplate(&b, tc.ProjectID, testLocation(t), templateID); err != nil {`
`397`	`397`	`t.Fatal(err)`
`398`	`398`	`}`
`399`		`- defer testCleanupTemplate(t, fmt.Sprintf("projects/%s/locations/%s/templates/%s", tc.ProjectID, "us-central1", templateID))`
	`399`	`+ defer testCleanupTemplate(t, fmt.Sprintf("projects/%s/locations/%s/templates/%s", tc.ProjectID, testLocation(t), templateID))`
`400`	`400`
`401`		`- if _, err := screenPDFFile(&b, tc.ProjectID, "us-central1", templateID, pdfContentBase); err != nil {`
	`401`	`+ if _, err := screenPDFFile(&b, tc.ProjectID, testLocation(t), templateID, pdfContentBase); err != nil {`
`402`	`402`	`t.Fatal(err)`
`403`	`403`	`}`
`404`	`404`