feat(hostcatalogs): Classify all proto fields
This commit is contained in:
1 parent
5440195cce
commit
8fa3e8dee0
11 files changed
+446
-219
No files matched your search
@@ -0,0 +1,157 @@
|
||||
package hostcatalogs_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/hashicorp/boundary/sdk/pbs/controller/api"
|
||||
pb "github.com/hashicorp/boundary/sdk/pbs/controller/api/resources/hostcatalogs"
|
||||
plugins "github.com/hashicorp/boundary/sdk/pbs/controller/api/resources/plugins"
|
||||
"github.com/hashicorp/boundary/sdk/pbs/controller/api/resources/scopes"
|
||||
"github.com/hashicorp/boundary/sdk/wrapper"
|
||||
"github.com/hashicorp/eventlogger"
|
||||
"github.com/hashicorp/eventlogger/filters/encrypt"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"google.golang.org/protobuf/types/known/structpb"
|
||||
"google.golang.org/protobuf/types/known/timestamppb"
|
||||
"google.golang.org/protobuf/types/known/wrapperspb"
|
||||
)
|
||||
|
||||
func TestClassification(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
now := time.Now()
|
||||
wrapper := wrapper.TestWrapper(t)
|
||||
testEncryptingFilter := api.NewEncryptFilter(t, wrapper)
|
||||
testEncryptingFilter.FilterOperationOverrides = map[encrypt.DataClassification]encrypt.FilterOperation{
|
||||
// Use HMAC for sensitive fields for easy test comparisons
|
||||
encrypt.SensitiveClassification: encrypt.HmacSha256Operation,
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
testEvent *eventlogger.Event
|
||||
wantEvent *eventlogger.Event
|
||||
}{
|
||||
{
|
||||
name: "validate-response-filtering",
|
||||
testEvent: &eventlogger.Event{
|
||||
Type: "test",
|
||||
CreatedAt: now,
|
||||
Payload: &pb.HostCatalog{
|
||||
Id: "public-id",
|
||||
ScopeId: "public-scope_id",
|
||||
Scope: &scopes.ScopeInfo{
|
||||
Id: "public-scope_id",
|
||||
Type: "public-scope_type",
|
||||
Name: "public-scope_name",
|
||||
Description: "public-scope_description",
|
||||
ParentScopeId: "public-parent_scope_id",
|
||||
},
|
||||
PluginId: "public-plugin_id",
|
||||
Plugin: &plugins.PluginInfo{
|
||||
Id: "public-plugin_id",
|
||||
Name: "public-plugin_name",
|
||||
Description: "public-plugin_description",
|
||||
},
|
||||
Name: wrapperspb.String("public-name"),
|
||||
Description: wrapperspb.String("public-description"),
|
||||
CreatedTime: timestamppb.New(now),
|
||||
UpdatedTime: timestamppb.New(now),
|
||||
Version: 1,
|
||||
Type: "public-type",
|
||||
Attrs: &pb.HostCatalog_Attributes{
|
||||
Attributes: &structpb.Struct{
|
||||
Fields: map[string]*structpb.Value{
|
||||
"public-attr_key": structpb.NewStringValue("sensitive-attr_value"),
|
||||
},
|
||||
},
|
||||
},
|
||||
Secrets: &structpb.Struct{
|
||||
Fields: map[string]*structpb.Value{
|
||||
"public-secret_key": structpb.NewStringValue("public-secret_value"),
|
||||
},
|
||||
},
|
||||
SecretsHmac: "public-hmac",
|
||||
AuthorizedActions: []string{
|
||||
"public-action",
|
||||
},
|
||||
AuthorizedCollectionActions: map[string]*structpb.ListValue{
|
||||
"public-collection_action": {
|
||||
Values: []*structpb.Value{
|
||||
structpb.NewStringValue("public-collection_action_value"),
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
wantEvent: &eventlogger.Event{
|
||||
Type: "test",
|
||||
CreatedAt: now,
|
||||
Payload: &pb.HostCatalog{
|
||||
Id: "public-id",
|
||||
ScopeId: "public-scope_id",
|
||||
Scope: &scopes.ScopeInfo{
|
||||
Id: "public-scope_id",
|
||||
Type: "public-scope_type",
|
||||
Name: "public-scope_name",
|
||||
Description: "public-scope_description",
|
||||
ParentScopeId: "public-parent_scope_id",
|
||||
},
|
||||
PluginId: "public-plugin_id",
|
||||
Plugin: &plugins.PluginInfo{
|
||||
Id: "public-plugin_id",
|
||||
Name: "public-plugin_name",
|
||||
Description: "public-plugin_description",
|
||||
},
|
||||
Name: wrapperspb.String("public-name"),
|
||||
Description: wrapperspb.String("public-description"),
|
||||
CreatedTime: timestamppb.New(now),
|
||||
UpdatedTime: timestamppb.New(now),
|
||||
Version: 1,
|
||||
Type: "public-type",
|
||||
Attrs: &pb.HostCatalog_Attributes{
|
||||
Attributes: &structpb.Struct{
|
||||
Fields: map[string]*structpb.Value{
|
||||
"public-attr_key": structpb.NewStringValue(encrypt.RedactedData),
|
||||
},
|
||||
},
|
||||
},
|
||||
Secrets: &structpb.Struct{
|
||||
Fields: map[string]*structpb.Value{
|
||||
"public-secret_key": structpb.NewStringValue(encrypt.RedactedData),
|
||||
},
|
||||
},
|
||||
SecretsHmac: "public-hmac",
|
||||
AuthorizedActions: []string{
|
||||
"public-action",
|
||||
},
|
||||
AuthorizedCollectionActions: map[string]*structpb.ListValue{
|
||||
"public-collection_action": {
|
||||
Values: []*structpb.Value{
|
||||
structpb.NewStringValue("public-collection_action_value"),
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
assert, require := assert.New(t), require.New(t)
|
||||
got, err := testEncryptingFilter.Process(ctx, tt.testEvent)
|
||||
require.NoError(err)
|
||||
require.NotNil(got)
|
||||
actualJson, err := json.Marshal(got)
|
||||
require.NoError(err)
|
||||
t.Log(string(actualJson))
|
||||
|
||||
wantJson, err := json.Marshal(tt.wantEvent)
|
||||
require.NoError(err)
|
||||
assert.JSONEq(string(wantJson), string(actualJson))
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user