fix(policy): enhance proto validation across policy requests

service/policy/attributes/attributes.proto

@@ -13,13 +13,21 @@ import "policy/selectors.proto";
 */
 
 message AttributeKeyAccessServer {
-  string attribute_id = 1;
-  string key_access_server_id = 2;
+  string attribute_id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
+  string key_access_server_id = 2 [
+    (buf.validate.field).string.uuid = true
+  ];
 }
 
 message ValueKeyAccessServer {
-  string value_id = 1;
-  string key_access_server_id = 2;
+  string value_id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
+  string key_access_server_id = 2 [
+    (buf.validate.field).string.uuid = true
+  ];
 }
 
 /*
@@ -37,15 +45,19 @@ message ListAttributesResponse {
 }
 
 message GetAttributeRequest {
-  string id = 1 [(buf.validate.field).required = true];
+  string id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
 }
 message GetAttributeResponse {
   policy.Attribute attribute = 1;
 }
 
 message CreateAttributeRequest {
   // Required
-  string namespace_id = 1 [(buf.validate.field).required = true];
+  string namespace_id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
   string name = 2 [
     (buf.validate.field).required = true,
     (buf.validate.field).string.max_len = 253,
@@ -84,7 +96,9 @@ message CreateAttributeResponse {
 
 message UpdateAttributeRequest {
   // Required
-  string id = 1 [(buf.validate.field).required = true];
+  string id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
 
   // Optional
   common.MetadataMutable metadata = 100;
@@ -95,7 +109,9 @@ message UpdateAttributeResponse {
 }
 
 message DeactivateAttributeRequest {
-  string id = 1 [(buf.validate.field).required = true];
+  string id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
 }
 message DeactivateAttributeResponse {
   policy.Attribute attribute = 1;
@@ -105,14 +121,18 @@ message DeactivateAttributeResponse {
 /// Value RPC messages
 ///
 message GetAttributeValueRequest {
-  string id = 1 [(buf.validate.field).required = true];
+  string id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
 }
 message GetAttributeValueResponse {
   policy.Value value = 1;
 }
 
 message ListAttributeValuesRequest {
-  string attribute_id = 1 [(buf.validate.field).required = true];
+  string attribute_id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
   // ACTIVE by default when not specified
   common.ActiveStateEnum state = 2;
 }
@@ -122,7 +142,9 @@ message ListAttributeValuesResponse {
 
 message CreateAttributeValueRequest {
   // Required
-  string attribute_id = 1 [(buf.validate.field).required = true];
+  string attribute_id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
   string value = 2     [
     (buf.validate.field).required = true,
     (buf.validate.field).string.max_len = 253,
@@ -145,7 +167,9 @@ message CreateAttributeValueResponse {
 }
 
 message UpdateAttributeValueRequest {
-  string id = 1 [(buf.validate.field).required = true];
+  string id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
 
   // Deprecated
   reserved "members";
@@ -160,7 +184,9 @@ message UpdateAttributeValueResponse {
 }
 
 message DeactivateAttributeValueRequest {
-  string id = 1 [(buf.validate.field).required = true];
+  string id = 1 [
+    (buf.validate.field).string.uuid = true
+  ];
 }
 message DeactivateAttributeValueResponse {
   policy.Value value = 1;

service/policy/attributes/attributes_test.go

@@ -121,7 +121,7 @@ func TestCreateAttribute_NamespaceIdMissing_Fails(t *testing.T) {
 
 	require.Error(t, err)
 	require.Contains(t, err.Error(), "namespace_id")
-	require.Contains(t, err.Error(), "[required]")
+	require.Contains(t, err.Error(), "[string.uuid_empty]")
 }
 
 func TestCreateAttribute_RuleMissing_Fails(t *testing.T) {
@@ -187,6 +187,86 @@ func TestCreateAttribute_ValueInvalid_Fails(t *testing.T) {
 	require.Contains(t, err.Error(), "[string.pattern]")
 }
 
+func TestAttributeKeyAccessServer_Succeeds(t *testing.T) {
+	validAttrKAS := &attributes.AttributeKeyAccessServer{
+		AttributeId:       validUUID,
+		KeyAccessServerId: validUUID,
+	}
+
+	err := getValidator().Validate(validAttrKAS)
+	require.NoError(t, err)
+}
+
+func TestAttributeKeyAccessServer_Fails(t *testing.T) {
+	bad := []struct {
+		attrID string
+		kasID  string
+	}{
+		{
+			"",
+			validUUID,
+		},
+		{
+			validUUID,
+			"",
+		},
+		{
+			"",
+			"",
+		},
+		{},
+	}
+
+	for _, test := range bad {
+		invalidAttrKAS := &attributes.AttributeKeyAccessServer{
+			AttributeId:       test.attrID,
+			KeyAccessServerId: test.kasID,
+		}
+		err := getValidator().Validate(invalidAttrKAS)
+		require.Error(t, err)
+		require.Contains(t, err.Error(), "string.uuid")
+	}
+}
+
+func TestGetAttributeRequest(t *testing.T) {
+	req := &attributes.GetAttributeRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.GetAttributeRequest{
+		Id: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}
+
+func TestUpdateAttributeRequest(t *testing.T) {
+	req := &attributes.UpdateAttributeRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.UpdateAttributeRequest{
+		Id: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}
+
+func TestDeactivateAttributeRequest(t *testing.T) {
+	req := &attributes.DeactivateAttributeRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.DeactivateAttributeRequest{
+		Id: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}
+
 // Create Attribute Values
 
 func TestCreateAttributeValue_Valid_Succeeds(t *testing.T) {
@@ -259,7 +339,7 @@ func TestCreateAttributeValue_AttributeIdMissing_Fails(t *testing.T) {
 
 	require.Error(t, err)
 	require.Contains(t, err.Error(), "attribute_id")
-	require.Contains(t, err.Error(), "[required]")
+	require.Contains(t, err.Error(), "[string.uuid_empty]")
 }
 
 func TestCreateAttributeValue_ValueMissing_Fails(t *testing.T) {
@@ -274,3 +354,96 @@ func TestCreateAttributeValue_ValueMissing_Fails(t *testing.T) {
 	require.Contains(t, err.Error(), "value")
 	require.Contains(t, err.Error(), "[required]")
 }
+
+func TestValueKeyAccessServer_Succeeds(t *testing.T) {
+	validValueKAS := &attributes.ValueKeyAccessServer{
+		ValueId:           validUUID,
+		KeyAccessServerId: validUUID,
+	}
+
+	err := getValidator().Validate(validValueKAS)
+	require.NoError(t, err)
+}
+
+func TestValueKeyAccessServer_Fails(t *testing.T) {
+	bad := []struct {
+		valID string
+		kasID string
+	}{
+		{
+			"",
+			validUUID,
+		},
+		{
+			validUUID,
+			"",
+		},
+		{
+			"",
+			"",
+		},
+		{},
+	}
+
+	for _, test := range bad {
+		invalidValKAS := &attributes.ValueKeyAccessServer{
+			ValueId:           test.valID,
+			KeyAccessServerId: test.kasID,
+		}
+		err := getValidator().Validate(invalidValKAS)
+		require.Error(t, err)
+		require.Contains(t, err.Error(), "string.uuid")
+	}
+}
+
+func TestGetAttributeValueRequest(t *testing.T) {
+	req := &attributes.GetAttributeValueRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.GetAttributeValueRequest{
+		Id: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}
+
+func TestListAttributeValuesRequest(t *testing.T) {
+	req := &attributes.ListAttributeValuesRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.ListAttributeValuesRequest{
+		AttributeId: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}
+
+func TestUpdateAttributeValueRequest(t *testing.T) {
+	req := &attributes.UpdateAttributeValueRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.UpdateAttributeValueRequest{
+		Id: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}
+
+func TestDeactivateAttributeValueRequest(t *testing.T) {
+	req := &attributes.DeactivateAttributeValueRequest{}
+	err := getValidator().Validate(req)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "string.uuid")
+
+	req = &attributes.DeactivateAttributeValueRequest{
+		Id: validUUID,
+	}
+	err = getValidator().Validate(req)
+	require.NoError(t, err)
+}