Fix bundle init not working in Azure Government

NEXT_CHANGELOG.md

@@ -10,4 +10,7 @@
 
 ### Dependency updates
 
+### Bundles
+* Fix bundle init not working in Azure Government ([#4286](https://github.com/databricks/cli/pull/4286))
+
 ### API Changes

acceptance/bundle/templates/default-python/azure-government/input.json

@@ -0,0 +1,7 @@
+{
+    "project_name": "my_default_python",
+    "include_notebook": "yes",
+    "include_dlt": "yes",
+    "include_python": "yes",
+    "serverless": "no"
+}

acceptance/bundle/templates/default-python/azure-government/output.txt

@@ -0,0 +1,13 @@
+
+>>> [CLI] bundle init default-python --config-file ./input.json --output-dir output
+Welcome to the default Python template for Databricks Asset Bundles!
+
+Answer the following questions to customize your project.
+You can always change your configuration in the databricks.yml file later.
+
+Note that [DATABRICKS_URL] is used for initialization.
+(For information on how to change your profile, see https://docs.databricks.com/dev-tools/cli/profiles.html.)
+
+✨ Your new project has been created in the 'my_default_python' directory!
+
+To get started, refer to the project README.md file and the documentation at https://docs.databricks.com/dev-tools/bundles/index.html.

acceptance/bundle/templates/default-python/azure-government/script

@@ -0,0 +1,3 @@
+trace $CLI bundle init default-python --config-file ./input.json --output-dir output
+
+rm -rf output

acceptance/bundle/templates/default-python/azure-government/test.toml

@@ -0,0 +1,4 @@
+[[Server]]
+Pattern = "GET /api/2.1/unity-catalog/current-metastore-assignment"
+Response.StatusCode = 404
+Response.Body = "{ \"error\": \"Bad Target: /api/2.1/unity-catalog/current-metastore-assignment\" }"

libs/template/helpers.go

@@ -148,7 +148,7 @@ func loadHelpers(ctx context.Context) template.FuncMap {
 				metastore, err := w.Metastores.Current(ctx)
 				if err != nil {
 					var aerr *apierr.APIError
-					if errors.As(err, &aerr) && slices.Contains(metastoreDisabledErrorCodes, aerr.ErrorCode) {
+					if errors.As(err, &aerr) && (slices.Contains(metastoreDisabledErrorCodes, aerr.ErrorCode) || aerr.Message == "Bad Target: /api/2.1/unity-catalog/current-metastore-assignment") {
 						// Ignore: access denied or workspace doesn't have a metastore assigned
 						empty_default := ""
 						cachedCatalog = &empty_default