feat(dataplex): save taxonomy display name for policy tag

raystack · Oct 17, 2023 · e36286c · e36286c
1 parent 442a06e
commit e36286c
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 9 deletions.
diff --git a/plugins/providers/dataplex/client.go b/plugins/providers/dataplex/client.go
@@ -55,9 +55,10 @@ func (p *policyTagClient) GetPolicies(ctx context.Context) ([]*Policy, error) {
 				break
 			}
 			policyTags = append(policyTags, &Policy{
-				Name:        tag.Name,
-				DisplayName: tag.DisplayName,
-				Description: tag.Description,
+				Name:                tag.Name,
+				DisplayName:         tag.DisplayName,
+				Description:         tag.Description,
+				TaxonomyDisplayName: taxonomy.DisplayName,
 			})
 		}
 	}

diff --git a/plugins/providers/dataplex/model.go b/plugins/providers/dataplex/model.go
@@ -13,9 +13,10 @@ const (
 
 // Policy is a reference to a Dataplex Policy Tag
 type Policy struct {
-	Name        string
-	DisplayName string
-	Description string
+	Name                string
+	DisplayName         string
+	Description         string
+	TaxonomyDisplayName string
 }
 
 func (p *Policy) ToDomain() *domain.Resource {
@@ -25,6 +26,7 @@ func (p *Policy) ToDomain() *domain.Resource {
 		URN:  p.Name,
 		Details: map[string]interface{}{
 			"description": p.Description,
+			"taxonomy":    p.TaxonomyDisplayName,
 		},
 	}
 }

diff --git a/plugins/providers/dataplex/provider_test.go b/plugins/providers/dataplex/provider_test.go
@@ -282,9 +282,10 @@ func TestGetResources(t *testing.T) {
 		}
 		expectedPolicies := []*dataplex.Policy{
 			{
-				Name:        "p_name",
-				DisplayName: "p_displayname",
-				Description: "p_description",
+				Name:                "p_name",
+				DisplayName:         "p_displayname",
+				Description:         "p_description",
+				TaxonomyDisplayName: "p_taxonomydisplayname",
 			},
 		}
 		client.On("GetPolicies", mock.Anything).Return(expectedPolicies, nil).Once()
@@ -298,6 +299,7 @@ func TestGetResources(t *testing.T) {
 				URN:          "p_name",
 				Details: map[string]interface{}{
 					"description": "p_description",
+					"taxonomy":    "p_taxonomydisplayname",
 				},
 			},
 		}