generic_setting_test.go

package settings

import (
	"testing"

	"github.com/databricks/databricks-sdk-go/apierr"
	"github.com/databricks/databricks-sdk-go/experimental/mocks"
	"github.com/databricks/databricks-sdk-go/service/settings"
	"github.com/databricks/terraform-provider-databricks/qa"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
)

// Choose an arbitrary setting to test.
var testSetting = AllSettingsResources()["default_namespace"]

func TestQueryCreateDefaultNameSetting(t *testing.T) {
	qa.ResourceFixture{
		MockWorkspaceClientFunc: func(w *mocks.MockWorkspaceClient) {
			e := w.GetMockDefaultNamespaceAPI().EXPECT()
			e.Update(mock.Anything, settings.UpdateDefaultNamespaceSettingRequest{
				AllowMissing: true,
				FieldMask:    "namespace.value",
				Setting: settings.DefaultNamespaceSetting{
					Etag: "",
					Namespace: settings.StringMessage{
						Value: "namespace_value",
					},
					SettingName: "default",
				},
			}).Return(nil, &apierr.APIError{
				ErrorCode:  "NOT_FOUND",
				StatusCode: 404,
				Message:    "SomeMessage",
				Details: []apierr.ErrorDetail{{
					Type: "type.googleapis.com/google.rpc.ErrorInfo",
					Metadata: map[string]string{
						etagAttrName: "etag1",
					},
				}},
			})
			e.Update(mock.Anything, settings.UpdateDefaultNamespaceSettingRequest{
				AllowMissing: true,
				FieldMask:    "namespace.value",
				Setting: settings.DefaultNamespaceSetting{
					Etag: "etag1",
					Namespace: settings.StringMessage{
						Value: "namespace_value",
					},
					SettingName: "default",
				},
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag2",
				Namespace: settings.StringMessage{
					Value: "namespace_value",
				},
				SettingName: "default",
			}, nil)
			e.Get(mock.Anything, settings.GetDefaultNamespaceSettingRequest{
				Etag: "etag2",
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag2",
				Namespace: settings.StringMessage{
					Value: "namespace_value",
				},
				SettingName: "default",
			}, nil)
		},
		Resource: testSetting,
		Create:   true,
		HCL: `
			namespace {
				value = "namespace_value"
			}
		`,
	}.ApplyAndExpectData(t, map[string]any{
		"id":                defaultSettingId,
		etagAttrName:        "etag2",
		"namespace.0.value": "namespace_value",
	})
}

func TestQueryReadDefaultNameSetting(t *testing.T) {
	qa.ResourceFixture{
		MockWorkspaceClientFunc: func(w *mocks.MockWorkspaceClient) {
			w.GetMockDefaultNamespaceAPI().EXPECT().Get(mock.Anything, settings.GetDefaultNamespaceSettingRequest{
				Etag: "etag1",
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag2",
				Namespace: settings.StringMessage{
					Value: "namespace_value",
				},
				SettingName: "default",
			}, nil)
		},
		Resource: testSetting,
		Read:     true,
		HCL: `
			namespace {
				value = "namespace_value"
			}
			etag = "etag1"
		`,
		ID: defaultSettingId,
	}.ApplyAndExpectData(t, map[string]any{
		"id":                defaultSettingId,
		etagAttrName:        "etag2",
		"namespace.0.value": "namespace_value",
	})
}

func TestQueryUpdateDefaultNameSetting(t *testing.T) {
	qa.ResourceFixture{
		MockWorkspaceClientFunc: func(w *mocks.MockWorkspaceClient) {
			e := w.GetMockDefaultNamespaceAPI().EXPECT()
			e.Update(mock.Anything, settings.UpdateDefaultNamespaceSettingRequest{
				AllowMissing: true,
				FieldMask:    "namespace.value",
				Setting: settings.DefaultNamespaceSetting{
					Etag: "etag1",
					Namespace: settings.StringMessage{
						Value: "new_namespace_value",
					},
					SettingName: "default",
				},
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag2",
				Namespace: settings.StringMessage{
					Value: "new_namespace_value",
				},
				SettingName: "default",
			}, nil)
			e.Get(mock.Anything, settings.GetDefaultNamespaceSettingRequest{
				Etag: "etag2",
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag2",
				Namespace: settings.StringMessage{
					Value: "new_namespace_value",
				},
				SettingName: "default",
			}, nil)
		},
		Resource: testSetting,
		Update:   true,
		HCL: `
			namespace {
				value = "new_namespace_value"
			}
			etag = "etag1"
		`,
		ID: defaultSettingId,
	}.ApplyAndExpectData(t, map[string]any{
		"id":                defaultSettingId,
		etagAttrName:        "etag2",
		"namespace.0.value": "new_namespace_value",
	})
}

func TestQueryUpdateDefaultNameSettingWithConflict(t *testing.T) {
	qa.ResourceFixture{
		MockWorkspaceClientFunc: func(w *mocks.MockWorkspaceClient) {
			e := w.GetMockDefaultNamespaceAPI().EXPECT()
			e.Update(mock.Anything, settings.UpdateDefaultNamespaceSettingRequest{
				AllowMissing: true,
				FieldMask:    "namespace.value",
				Setting: settings.DefaultNamespaceSetting{
					Etag: "etag1",
					Namespace: settings.StringMessage{
						Value: "new_namespace_value",
					},
					SettingName: "default",
				},
			}).Return(nil, &apierr.APIError{
				ErrorCode:  "RESOURCE_CONFLICT",
				StatusCode: 409,
				Message:    "SomeMessage",
				Details: []apierr.ErrorDetail{{
					Type: "type.googleapis.com/google.rpc.ErrorInfo",
					Metadata: map[string]string{
						etagAttrName: "etag2",
					},
				}},
			})
			e.Update(mock.Anything, settings.UpdateDefaultNamespaceSettingRequest{
				AllowMissing: true,
				FieldMask:    "namespace.value",
				Setting: settings.DefaultNamespaceSetting{
					Etag: "etag2",
					Namespace: settings.StringMessage{
						Value: "new_namespace_value",
					},
					SettingName: "default",
				},
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag3",
				Namespace: settings.StringMessage{
					Value: "new_namespace_value",
				},
				SettingName: "default",
			}, nil)
			e.Get(mock.Anything, settings.GetDefaultNamespaceSettingRequest{
				Etag: "etag3",
			}).Return(&settings.DefaultNamespaceSetting{
				Etag: "etag3",
				Namespace: settings.StringMessage{
					Value: "new_namespace_value",
				},
				SettingName: "default",
			}, nil)
		},
		Resource: testSetting,
		Update:   true,
		HCL: `
			namespace {
				value = "new_namespace_value"
			}
			etag = "etag1"
		`,
		ID: defaultSettingId,
	}.ApplyAndExpectData(t, map[string]any{
		"id":                defaultSettingId,
		etagAttrName:        "etag3",
		"namespace.0.value": "new_namespace_value",
	})
}

func TestQueryDeleteDefaultNameSetting(t *testing.T) {
	d, err := qa.ResourceFixture{
		MockWorkspaceClientFunc: func(w *mocks.MockWorkspaceClient) {
			w.GetMockDefaultNamespaceAPI().EXPECT().Delete(mock.Anything, settings.DeleteDefaultNamespaceSettingRequest{
				Etag: "etag1",
			}).Return(&settings.DeleteDefaultNamespaceSettingResponse{
				Etag: "etag2",
			}, nil)
		},
		Resource: testSetting,
		Delete:   true,
		HCL: `
			namespace {
				value = "new_namespace_value"
			}
			etag = "etag1"
		`,
		ID: defaultSettingId,
	}.Apply(t)

	assert.NoError(t, err)
	assert.Equal(t, "etag2", d.Get(etagAttrName).(string))
}

func TestQueryDeleteDefaultNameSettingWithConflict(t *testing.T) {
	d, err := qa.ResourceFixture{
		MockWorkspaceClientFunc: func(w *mocks.MockWorkspaceClient) {
			w.GetMockDefaultNamespaceAPI().EXPECT().Delete(mock.Anything, settings.DeleteDefaultNamespaceSettingRequest{
				Etag: "etag1",
			}).Return(nil, &apierr.APIError{
				ErrorCode:  "RESOURCE_CONFLICT",
				StatusCode: 409,
				Message:    "SomeMessage",
				Details: []apierr.ErrorDetail{{
					Type: "type.googleapis.com/google.rpc.ErrorInfo",
					Metadata: map[string]string{
						etagAttrName: "etag2",
					},
				}},
			})
			w.GetMockDefaultNamespaceAPI().EXPECT().Delete(mock.Anything, settings.DeleteDefaultNamespaceSettingRequest{
				Etag: "etag2",
			}).Return(&settings.DeleteDefaultNamespaceSettingResponse{
				Etag: "etag3",
			}, nil)
		},
		Resource: testSetting,
		Delete:   true,
		HCL: `
			namespace {
				value = "new_namespace_value"
			}
			etag = "etag1"
		`,
		ID: defaultSettingId,
	}.Apply(t)

	assert.NoError(t, err)
	assert.Equal(t, "etag3", d.Get(etagAttrName).(string))
}