Compare commits
7 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 6ca00f27e9 | |||
| 520be66035 | |||
| c3146ca433 | |||
| b4d9628b05 | |||
| 635d31d723 | |||
| 08efcb92db | |||
| dad1216e1f |
13
.env
Normal file
13
.env
Normal file
@ -0,0 +1,13 @@
|
||||
ENVIRONMENT=local
|
||||
DATABASE_PASSWORD=QgSvstSjoc6fKphMzNgT3SliNY10eSRS
|
||||
DATABASE_ROLE=agreego_admin
|
||||
DATABASE_HOST=127.1.27.9
|
||||
DATABASE_PORT=5432
|
||||
POSTGRES_PASSWORD=xzIq5JT0xY3F+2m1GtnrKDdK29sNSXVVYZHPKJVh8pI=
|
||||
DATABASE_NAME=agreego
|
||||
DEV_DATABASE_NAME=agreego_dev
|
||||
GITEA_TOKEN=3d70c23673517330623a5122998fb304e3c73f0a
|
||||
MOOV_ACCOUNT_ID=69a0d2f6-77a2-4e26-934f-d869134f87d3
|
||||
MOOV_PUBLIC_KEY=9OMhK5qGnh7Tmk2Z
|
||||
MOOV_SECRET_KEY=DrRox7B-YWfO9IheiUUX7lGP8-7VY-Ni
|
||||
MOOV_DOMAIN=http://localhost
|
||||
2
flows
2
flows
Submodule flows updated: db55335254...9d758d581e
477
src/lib.rs
477
src/lib.rs
@ -4,399 +4,166 @@ pg_module_magic!();
|
||||
|
||||
use serde_json::{json, Value};
|
||||
use std::{collections::HashMap, sync::RwLock};
|
||||
use boon::{Compiler, Schemas, ValidationError, SchemaIndex};
|
||||
use boon::{Compiler, Schemas, ValidationError, SchemaIndex, CompileError};
|
||||
use lazy_static::lazy_static;
|
||||
|
||||
struct BoonCache {
|
||||
schemas: Schemas,
|
||||
id_to_index: HashMap<String, SchemaIndex>,
|
||||
schemas: Schemas,
|
||||
id_to_index: HashMap<String, SchemaIndex>,
|
||||
}
|
||||
|
||||
lazy_static! {
|
||||
static ref SCHEMA_CACHE: RwLock<BoonCache> = RwLock::new(BoonCache {
|
||||
schemas: Schemas::new(),
|
||||
id_to_index: HashMap::new()
|
||||
});
|
||||
static ref SCHEMA_CACHE: RwLock<BoonCache> = RwLock::new(BoonCache {
|
||||
schemas: Schemas::new(),
|
||||
id_to_index: HashMap::new(),
|
||||
});
|
||||
}
|
||||
|
||||
#[pg_extern(strict)]
|
||||
fn cache_json_schema(schema_id: &str, schema: JsonB) -> JsonB {
|
||||
let mut cache = SCHEMA_CACHE.write().unwrap();
|
||||
let schema_value: Value = schema.0;
|
||||
let mut cache = SCHEMA_CACHE.write().unwrap();
|
||||
let schema_value: Value = schema.0;
|
||||
let schema_path = format!("urn:{}", schema_id);
|
||||
|
||||
let mut compiler = Compiler::new();
|
||||
compiler.enable_format_assertions();
|
||||
let mut compiler = Compiler::new();
|
||||
compiler.enable_format_assertions();
|
||||
|
||||
let schema_url = format!("urn:jspg:{}", schema_id);
|
||||
// Use schema_path when adding the resource
|
||||
if let Err(e) = compiler.add_resource(&schema_path, schema_value.clone()) {
|
||||
return JsonB(json!({
|
||||
"success": false,
|
||||
"error": {
|
||||
"message": format!("Failed to add schema resource '{}': {}", schema_id, e),
|
||||
"schema_path": schema_path
|
||||
}
|
||||
}));
|
||||
}
|
||||
|
||||
if let Err(e) = compiler.add_resource(&schema_url, schema_value) {
|
||||
return JsonB(json!({
|
||||
"success": false,
|
||||
"error": format!("Failed to add schema resource '{}': {}", schema_id, e)
|
||||
}));
|
||||
// Use schema_path when compiling
|
||||
match compiler.compile(&schema_path, &mut cache.schemas) {
|
||||
Ok(sch_index) => {
|
||||
// Store the index using the original schema_id as the key
|
||||
cache.id_to_index.insert(schema_id.to_string(), sch_index);
|
||||
JsonB(json!({ "success": true }))
|
||||
}
|
||||
|
||||
match compiler.compile(&schema_url, &mut cache.schemas) {
|
||||
Ok(sch_index) => {
|
||||
cache.id_to_index.insert(schema_id.to_string(), sch_index);
|
||||
JsonB(json!({
|
||||
"success": true,
|
||||
"schema_id": schema_id,
|
||||
"message": "Schema cached successfully."
|
||||
}))
|
||||
Err(e) => {
|
||||
let error = match &e {
|
||||
CompileError::ValidationError { url: _url, src } => { // Prefix url with _
|
||||
json!({
|
||||
"message": format!("Schema '{}' failed validation against its metaschema: {}", schema_id, src),
|
||||
"schema_path": schema_path,
|
||||
"error": format!("{:?}", src),
|
||||
})
|
||||
}
|
||||
Err(e) => JsonB(json!({
|
||||
"success": false,
|
||||
"schema_id": schema_id,
|
||||
"error": format!("Schema compilation failed: {}", e)
|
||||
})),
|
||||
_ => {
|
||||
let _error_type = format!("{:?}", e).split('(').next().unwrap_or("Unknown").to_string(); // Prefix error_type with _
|
||||
json!({
|
||||
"message": format!("Schema '{}' compilation failed: {}", schema_id, e),
|
||||
"schema_path": schema_path,
|
||||
"error": format!("{:?}", e),
|
||||
})
|
||||
}
|
||||
};
|
||||
JsonB(json!({
|
||||
"success": false,
|
||||
"error": error
|
||||
}))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[pg_extern(strict, parallel_safe)]
|
||||
fn validate_json_schema(schema_id: &str, instance: JsonB) -> JsonB {
|
||||
let cache = SCHEMA_CACHE.read().unwrap();
|
||||
let cache = SCHEMA_CACHE.read().unwrap();
|
||||
|
||||
match cache.id_to_index.get(schema_id) {
|
||||
None => JsonB(json!({
|
||||
// Lookup uses the original schema_id
|
||||
match cache.id_to_index.get(schema_id) {
|
||||
None => JsonB(json!({
|
||||
"success": false,
|
||||
"error": {
|
||||
"message": format!("Schema with id '{}' not found in cache", schema_id)
|
||||
}
|
||||
})),
|
||||
Some(sch_index) => {
|
||||
let instance_value: Value = instance.0;
|
||||
match cache.schemas.validate(&instance_value, *sch_index) {
|
||||
Ok(_) => JsonB(json!({ "success": true })),
|
||||
Err(validation_error) => {
|
||||
let error = format_validation_error(&validation_error);
|
||||
JsonB(json!({
|
||||
"success": false,
|
||||
"errors": [{
|
||||
"kind": "SchemaNotFound",
|
||||
"message": format!("Schema with id '{}' not found in cache", schema_id)
|
||||
}]
|
||||
})),
|
||||
Some(sch_index) => {
|
||||
let instance_value: Value = instance.0;
|
||||
match cache.schemas.validate(&instance_value, *sch_index) {
|
||||
Ok(_) => JsonB(json!({ "success": true })),
|
||||
Err(validation_error) => {
|
||||
let error_details = format_boon_errors(&validation_error);
|
||||
JsonB(json!({
|
||||
"success": false,
|
||||
"errors": [error_details]
|
||||
}))
|
||||
}
|
||||
}
|
||||
"error": error
|
||||
}))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn format_boon_errors(error: &ValidationError) -> Value {
|
||||
json!({
|
||||
"instance_path": error.instance_location.to_string(),
|
||||
"schema_path": error.schema_url.to_string(),
|
||||
"kind": format!("{:?}", error.kind),
|
||||
"message": format!("{}", error),
|
||||
"causes": error
|
||||
.causes
|
||||
.iter()
|
||||
.map(format_boon_errors)
|
||||
.collect::<Vec<_>>()
|
||||
})
|
||||
fn format_validation_error(error: &ValidationError) -> Value {
|
||||
let nested_errors: Vec<Value> = error.causes.iter().map(format_validation_error).collect();
|
||||
|
||||
// Use specific message for leaf errors, generic for containers
|
||||
let message = if error.causes.is_empty() {
|
||||
let default_message = format!("{}", error); // Use boon's default message for specific errors
|
||||
// Try to strip the "at '<path>': " prefix
|
||||
if let Some(start_index) = default_message.find("': ") {
|
||||
default_message[start_index + 3..].to_string()
|
||||
} else {
|
||||
default_message // Fallback if pattern not found
|
||||
}
|
||||
} else {
|
||||
"Validation failed due to nested errors".to_string() // Generic message for container errors
|
||||
};
|
||||
|
||||
json!({
|
||||
"message": message,
|
||||
"schema_path": error.schema_url.to_string(), // Use schema_url directly
|
||||
"instance_path": error.instance_location.to_string(),
|
||||
"error": nested_errors // Recursively format nested errors
|
||||
})
|
||||
}
|
||||
|
||||
#[pg_extern(strict, parallel_safe)]
|
||||
fn json_schema_cached(schema_id: &str) -> bool {
|
||||
let cache = SCHEMA_CACHE.read().unwrap();
|
||||
cache.id_to_index.contains_key(schema_id)
|
||||
let cache = SCHEMA_CACHE.read().unwrap();
|
||||
cache.id_to_index.contains_key(schema_id)
|
||||
}
|
||||
|
||||
#[pg_extern(strict)]
|
||||
fn clear_json_schemas() -> JsonB {
|
||||
let mut cache = SCHEMA_CACHE.write().unwrap();
|
||||
*cache = BoonCache {
|
||||
schemas: Schemas::new(),
|
||||
id_to_index: HashMap::new()
|
||||
};
|
||||
JsonB(json!({
|
||||
"success": true,
|
||||
"message": "Schema cache cleared."
|
||||
}))
|
||||
fn clear_json_schemas() {
|
||||
let mut cache = SCHEMA_CACHE.write().unwrap();
|
||||
*cache = BoonCache {
|
||||
schemas: Schemas::new(),
|
||||
id_to_index: HashMap::new(),
|
||||
};
|
||||
}
|
||||
|
||||
#[pg_extern(strict, parallel_safe)]
|
||||
fn show_json_schemas() -> JsonB {
|
||||
let cache = SCHEMA_CACHE.read().unwrap();
|
||||
let ids: Vec<&String> = cache.id_to_index.keys().collect();
|
||||
JsonB(json!({
|
||||
"cached_schema_ids": ids
|
||||
}))
|
||||
}
|
||||
|
||||
#[pg_schema]
|
||||
#[cfg(any(test, feature = "pg_test"))]
|
||||
mod tests {
|
||||
use pgrx::*;
|
||||
use pgrx::pg_test;
|
||||
use super::*;
|
||||
use serde_json::json;
|
||||
|
||||
fn jsonb(val: Value) -> JsonB {
|
||||
JsonB(val)
|
||||
}
|
||||
|
||||
fn setup_test() {
|
||||
clear_json_schemas();
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_cache_and_validate_json_schema() {
|
||||
setup_test();
|
||||
let schema_id = "my_schema";
|
||||
let schema = json!({
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"name": { "type": "string" },
|
||||
"age": { "type": "integer", "minimum": 0 }
|
||||
},
|
||||
"required": ["name", "age"]
|
||||
});
|
||||
let valid_instance = json!({ "name": "Alice", "age": 30 });
|
||||
let invalid_instance_type = json!({ "name": "Bob", "age": -5 });
|
||||
let invalid_instance_missing = json!({ "name": "Charlie" });
|
||||
|
||||
let cache_result = cache_json_schema(schema_id, jsonb(schema.clone()));
|
||||
assert!(cache_result.0["success"].as_bool().unwrap());
|
||||
|
||||
let valid_result = validate_json_schema(schema_id, jsonb(valid_instance));
|
||||
assert!(valid_result.0["success"].as_bool().unwrap());
|
||||
|
||||
let invalid_result_type = validate_json_schema(schema_id, jsonb(invalid_instance_type));
|
||||
assert!(!invalid_result_type.0["success"].as_bool().unwrap());
|
||||
|
||||
// --- Assertions for invalid_result_type ---
|
||||
|
||||
// Get top-level errors
|
||||
let top_level_errors = invalid_result_type.0["errors"].as_array().expect("Top-level 'errors' should be an array");
|
||||
assert_eq!(top_level_errors.len(), 1, "Should have exactly one top-level error for invalid type");
|
||||
|
||||
// Get the first (and only) top-level error
|
||||
let top_level_error = top_level_errors.get(0).expect("Should get the first top-level error");
|
||||
|
||||
// Check top-level error kind
|
||||
assert!(top_level_error.get("kind").and_then(Value::as_str).map_or(false, |k| k.starts_with("Schema { url:")),
|
||||
"Incorrect kind for top-level error. Expected 'Schema {{ url:'. Error: {:?}. All errors: {:?}", top_level_error, top_level_errors);
|
||||
|
||||
// Get the 'causes' array from the top-level error
|
||||
let causes_age = top_level_error.get("causes").and_then(Value::as_array).expect("Top-level error 'causes' should be an array");
|
||||
assert_eq!(causes_age.len(), 1, "Should have one cause for the age error");
|
||||
|
||||
// Get the actual age error from the 'causes' array
|
||||
let age_error = causes_age.get(0).expect("Should have an error object in 'causes'");
|
||||
assert_eq!(age_error.get("instance_path").and_then(Value::as_str), Some("/age"),
|
||||
"Incorrect instance_path for age error. Error: {:?}. All errors: {:?}", age_error, top_level_errors);
|
||||
|
||||
assert!(age_error.get("kind").and_then(Value::as_str).map_or(false, |k| k.starts_with("Minimum { got:")),
|
||||
"Incorrect kind prefix for age error. Expected 'Minimum {{ got:'. Error: {:?}. All errors: {:?}", age_error, top_level_errors);
|
||||
|
||||
let expected_prefix = "at '/age': must be >=0";
|
||||
assert!(age_error.get("message")
|
||||
.and_then(Value::as_str)
|
||||
.map_or(false, |m| m.starts_with(expected_prefix)),
|
||||
"Incorrect message prefix for age error. Expected prefix '{}'. Error: {:?}. All errors: {:?}",
|
||||
expected_prefix, age_error, top_level_errors);
|
||||
|
||||
let invalid_result_missing = validate_json_schema(schema_id, jsonb(invalid_instance_missing));
|
||||
assert!(!invalid_result_missing.0["success"].as_bool().unwrap(), "Validation should fail for missing required field");
|
||||
|
||||
// --- Assertions for invalid_result_missing ---
|
||||
|
||||
// Get top-level errors
|
||||
let top_level_errors_missing = invalid_result_missing.0["errors"].as_array().expect("Errors should be an array for missing field");
|
||||
assert_eq!(top_level_errors_missing.len(), 1, "Should have one top-level error for missing field");
|
||||
|
||||
// Get the first (and only) top-level error
|
||||
let top_error_missing = top_level_errors_missing.get(0).expect("Should get the first top-level missing field error");
|
||||
|
||||
// Check top-level error kind
|
||||
assert!(top_error_missing.get("kind").and_then(Value::as_str).map_or(false, |k| k.starts_with("Schema { url:")),
|
||||
"Incorrect kind for missing field top-level error. Error: {:?}. All errors: {:?}", top_error_missing, top_level_errors_missing);
|
||||
|
||||
// Get the 'causes' array from the top-level error
|
||||
let causes_missing = top_error_missing.get("causes").and_then(Value::as_array).expect("Causes should be an array for missing field error");
|
||||
assert_eq!(causes_missing.len(), 1, "Should have one cause for missing field");
|
||||
|
||||
// Get the actual missing field error from the 'causes' array
|
||||
let missing_error = causes_missing.get(0).expect("Should have missing field error object in 'causes'");
|
||||
|
||||
// Assertions on the specific missing field error
|
||||
assert_eq!(missing_error.get("instance_path").and_then(Value::as_str), Some(""),
|
||||
"Incorrect instance_path for missing age error: {:?}", missing_error);
|
||||
assert!(missing_error.get("kind").and_then(Value::as_str).map_or(false, |k| k.starts_with("Required { want: [\"age\"]")),
|
||||
"Incorrect kind for missing age error. Expected prefix 'Required {{ want: [\"age\"] }}'. Error: {:?}", missing_error);
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_validate_json_schema_not_cached() {
|
||||
setup_test();
|
||||
let instance = json!({ "foo": "bar" });
|
||||
let result = validate_json_schema("non_existent_schema", jsonb(instance));
|
||||
assert!(!result.0["success"].as_bool().unwrap());
|
||||
let errors = result.0["errors"].as_array().unwrap();
|
||||
assert_eq!(errors.len(), 1);
|
||||
assert_eq!(errors[0]["kind"], json!("SchemaNotFound"));
|
||||
assert!(errors[0]["message"].as_str().unwrap().contains("non_existent_schema"));
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_cache_invalid_json_schema() {
|
||||
setup_test();
|
||||
let schema_id = "invalid_schema";
|
||||
let invalid_schema_json = "{\"type\": \"string\" \"maxLength\": 5}";
|
||||
let invalid_schema_value: Result<Value, _> = serde_json::from_str(invalid_schema_json);
|
||||
assert!(invalid_schema_value.is_err(), "Test setup assumes invalid JSON string");
|
||||
|
||||
let schema_representing_invalid = json!({
|
||||
"type": 123
|
||||
});
|
||||
|
||||
let result = cache_json_schema(schema_id, jsonb(schema_representing_invalid.clone()));
|
||||
assert!(!result.0["success"].as_bool().unwrap());
|
||||
assert!(result.0["error"].as_str().unwrap().contains("Schema compilation failed"));
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_validate_json_schema_detailed_validation_errors() {
|
||||
setup_test();
|
||||
let schema_id = "detailed_schema";
|
||||
let schema = json!({
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"address": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"street": { "type": "string" },
|
||||
"city": { "type": "string", "maxLength": 10 }
|
||||
},
|
||||
"required": ["street", "city"]
|
||||
}
|
||||
},
|
||||
"required": ["address"]
|
||||
});
|
||||
let invalid_instance = json!({
|
||||
"address": {
|
||||
"city": "San Francisco Bay Area"
|
||||
}
|
||||
});
|
||||
|
||||
assert!(cache_json_schema(schema_id, jsonb(schema.clone())).0["success"].as_bool().unwrap());
|
||||
let result = validate_json_schema(schema_id, jsonb(invalid_instance));
|
||||
assert!(!result.0["success"].as_bool().unwrap());
|
||||
|
||||
let errors = result.0["errors"].as_array().expect("Errors should be an array");
|
||||
let top_error = errors.get(0).expect("Expected at least one top-level error object");
|
||||
let causes = top_error.get("causes").and_then(Value::as_array).expect("Expected causes array");
|
||||
|
||||
let has_required_street_error = causes.iter().any(|e|
|
||||
e.get("instance_path").and_then(Value::as_str) == Some("/address") && // Check path inside cause
|
||||
e.get("kind").and_then(Value::as_str).unwrap_or("").starts_with("Required { want:") && // Check kind prefix
|
||||
e.get("kind").and_then(Value::as_str).unwrap_or("").contains("street") // Ensure 'street' is mentioned
|
||||
);
|
||||
assert!(has_required_street_error, "Missing required 'street' error within causes. Actual errors: {:?}", errors);
|
||||
|
||||
let has_maxlength_city_error = causes.iter().any(|e| // Check within causes
|
||||
e.get("instance_path").and_then(Value::as_str) == Some("/address/city") &&
|
||||
e.get("kind").and_then(Value::as_str).unwrap_or("").starts_with("MaxLength { got:") // Check kind prefix
|
||||
);
|
||||
assert!(has_maxlength_city_error, "Missing maxLength 'city' error within causes. Actual errors: {:?}", errors);
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_validate_json_schema_oneof_validation_errors() {
|
||||
setup_test();
|
||||
let schema_id = "oneof_schema";
|
||||
let schema = json!({
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"value": {
|
||||
"oneOf": [
|
||||
{ "type": "string", "minLength": 5 },
|
||||
{ "type": "number", "minimum": 10 }
|
||||
]
|
||||
}
|
||||
},
|
||||
"required": ["value"]
|
||||
});
|
||||
assert!(cache_json_schema(schema_id, jsonb(schema.clone())).0["success"].as_bool().unwrap());
|
||||
|
||||
let invalid_instance = json!({ "value": "abc" });
|
||||
let result = validate_json_schema(schema_id, jsonb(invalid_instance));
|
||||
|
||||
assert!(!result.0["success"].as_bool().unwrap());
|
||||
|
||||
let errors_val = result.0["errors"].as_array().expect("Errors should be an array");
|
||||
let top_schema_error = errors_val.get(0).expect("Expected at least one top-level Schema error object");
|
||||
let schema_error_causes = top_schema_error.get("causes").and_then(Value::as_array).expect("Expected causes array for Schema error");
|
||||
|
||||
let oneof_error = schema_error_causes.iter().find(|e| {
|
||||
e.get("kind").and_then(Value::as_str) == Some("OneOf(None)") &&
|
||||
e.get("instance_path").and_then(Value::as_str) == Some("/value")
|
||||
}).expect("Could not find the OneOf(None) error for /value within Schema causes");
|
||||
|
||||
let oneof_causes = oneof_error.get("causes").and_then(Value::as_array)
|
||||
.expect("Expected causes array for OneOf error");
|
||||
|
||||
let has_minlength_error = oneof_causes.iter().any(|e| // Check within OneOf causes
|
||||
e.get("instance_path").and_then(Value::as_str) == Some("/value") &&
|
||||
e.get("kind").and_then(Value::as_str).unwrap_or("").starts_with("MinLength { got:") // Check kind prefix
|
||||
);
|
||||
assert!(has_minlength_error, "Missing MinLength error within OneOf causes. Actual errors: {:?}", errors_val);
|
||||
|
||||
let has_type_error = oneof_causes.iter().any(|e| // Check within OneOf causes
|
||||
e.get("instance_path").and_then(Value::as_str) == Some("/value") &&
|
||||
e.get("kind").and_then(Value::as_str).unwrap_or("").starts_with("Type { got: String, want: Types") // More specific kind check
|
||||
);
|
||||
assert!(has_type_error, "Missing Type error within OneOf causes. Actual errors: {:?}", errors_val);
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_clear_json_schemas() {
|
||||
setup_test();
|
||||
let schema_id = "schema_to_clear";
|
||||
let schema = json!({ "type": "string" });
|
||||
cache_json_schema(schema_id, jsonb(schema.clone()));
|
||||
|
||||
let show_result1 = show_json_schemas();
|
||||
assert!(show_result1.0["cached_schema_ids"].as_array().unwrap().iter().any(|id| id.as_str() == Some(schema_id)));
|
||||
|
||||
let clear_result = clear_json_schemas();
|
||||
assert!(clear_result.0["success"].as_bool().unwrap());
|
||||
|
||||
let show_result2 = show_json_schemas();
|
||||
assert!(show_result2.0["cached_schema_ids"].as_array().unwrap().is_empty());
|
||||
|
||||
let instance = json!("test");
|
||||
let validate_result = validate_json_schema(schema_id, jsonb(instance));
|
||||
assert!(!validate_result.0["success"].as_bool().unwrap());
|
||||
assert_eq!(validate_result.0["errors"].as_array().unwrap()[0]["kind"], json!("SchemaNotFound"));
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_show_json_schemas() {
|
||||
setup_test();
|
||||
let schema_id1 = "schema1";
|
||||
let schema_id2 = "schema2";
|
||||
let schema = json!({ "type": "boolean" });
|
||||
|
||||
cache_json_schema(schema_id1, jsonb(schema.clone()));
|
||||
cache_json_schema(schema_id2, jsonb(schema.clone()));
|
||||
|
||||
let result = show_json_schemas();
|
||||
let ids = result.0["cached_schema_ids"].as_array().unwrap();
|
||||
assert_eq!(ids.len(), 2);
|
||||
assert!(ids.contains(&json!(schema_id1)));
|
||||
assert!(ids.contains(&json!(schema_id2)));
|
||||
}
|
||||
fn show_json_schemas() -> Vec<String> {
|
||||
let cache = SCHEMA_CACHE.read().unwrap();
|
||||
let ids: Vec<String> = cache.id_to_index.keys().cloned().collect();
|
||||
ids
|
||||
}
|
||||
|
||||
/// This module is required by `cargo pgrx test` invocations.
|
||||
/// It must be visible at the root of your extension crate.
|
||||
#[cfg(test)]
|
||||
pub mod pg_test {
|
||||
pub fn setup(_options: Vec<&str>) {
|
||||
// perform one-off initialization when the pg_test framework starts
|
||||
}
|
||||
pub fn setup(_options: Vec<&str>) {
|
||||
// perform one-off initialization when the pg_test framework starts
|
||||
}
|
||||
|
||||
pub fn postgresql_conf_options() -> Vec<&'static str> {
|
||||
// return any postgresql.conf settings that are required for your tests
|
||||
vec![]
|
||||
}
|
||||
}
|
||||
#[must_use]
|
||||
pub fn postgresql_conf_options() -> Vec<&'static str> {
|
||||
// return any postgresql.conf settings that are required for your tests
|
||||
vec![]
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[cfg(any(test, feature = "pg_test"))]
|
||||
#[pg_schema]
|
||||
mod tests {
|
||||
include!("tests.rs");
|
||||
}
|
||||
|
||||
264
src/tests.rs
Normal file
264
src/tests.rs
Normal file
@ -0,0 +1,264 @@
|
||||
use crate::*;
|
||||
use serde_json::{json, Value};
|
||||
|
||||
// Helper macro for asserting success with pretty JSON output on failure
|
||||
macro_rules! assert_success_with_json {
|
||||
($result_jsonb:expr, $fmt:literal $(, $($args:tt)*)?) => {
|
||||
let condition_result: Option<bool> = $result_jsonb.0.get("success").and_then(Value::as_bool);
|
||||
if condition_result != Some(true) {
|
||||
let base_msg = format!($fmt $(, $($args)*)?);
|
||||
let pretty_json = serde_json::to_string_pretty(&$result_jsonb.0)
|
||||
.unwrap_or_else(|_| format!("(Failed to pretty-print JSON: {:?})", $result_jsonb.0));
|
||||
// Construct the full message string first
|
||||
let panic_msg = format!("Assertion Failed (expected success): {}\nResult JSON:\n{}", base_msg, pretty_json);
|
||||
panic!("{}", panic_msg); // Use the constructed string
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
// Helper macro for asserting failure with pretty JSON output on failure
|
||||
macro_rules! assert_failure_with_json {
|
||||
($result_jsonb:expr, $fmt:literal $(, $($args:tt)*)?) => {
|
||||
let condition_result: Option<bool> = $result_jsonb.0.get("success").and_then(Value::as_bool);
|
||||
if condition_result != Some(false) {
|
||||
let base_msg = format!($fmt $(, $($args)*)?);
|
||||
let pretty_json = serde_json::to_string_pretty(&$result_jsonb.0)
|
||||
.unwrap_or_else(|_| format!("(Failed to pretty-print JSON: {:?})", $result_jsonb.0));
|
||||
// Construct the full message string first
|
||||
let panic_msg = format!("Assertion Failed (expected failure): {}\nResult JSON:\n{}", base_msg, pretty_json);
|
||||
panic!("{}", panic_msg); // Use the constructed string
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
fn jsonb(val: Value) -> JsonB {
|
||||
JsonB(val)
|
||||
}
|
||||
|
||||
fn setup_test() {
|
||||
clear_json_schemas();
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_cache_and_validate_json_schema() {
|
||||
setup_test();
|
||||
let schema_id = "my_schema";
|
||||
let schema = json!({
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"name": { "type": "string" },
|
||||
"age": { "type": "integer", "minimum": 0 }
|
||||
},
|
||||
"required": ["name", "age"]
|
||||
});
|
||||
let valid_instance = json!({ "name": "Alice", "age": 30 });
|
||||
let invalid_instance_type = json!({ "name": "Bob", "age": -5 });
|
||||
let invalid_instance_missing = json!({ "name": "Charlie" });
|
||||
|
||||
let cache_result = cache_json_schema(schema_id, jsonb(schema.clone()));
|
||||
assert_success_with_json!(cache_result, "Cache operation should succeed.");
|
||||
|
||||
let valid_result = validate_json_schema(schema_id, jsonb(valid_instance));
|
||||
assert_success_with_json!(valid_result, "Validation of valid instance should succeed.");
|
||||
|
||||
let invalid_result_type = validate_json_schema(schema_id, jsonb(invalid_instance_type));
|
||||
assert_failure_with_json!(invalid_result_type, "Validation with invalid type should fail.");
|
||||
|
||||
let error_obj_type = invalid_result_type.0.get("error").expect("Expected top-level 'error' object");
|
||||
let causes_age = error_obj_type.get("error").and_then(Value::as_array).expect("Expected nested 'error' array (causes)");
|
||||
assert!(!causes_age.is_empty(), "Expected causes for invalid age");
|
||||
let age_error = causes_age.iter().find(|e| e["instance_path"].as_str() == Some("/age")).expect("Missing cause for /age instance path");
|
||||
let age_error_message = age_error["message"].as_str().expect("Expected string message for age error");
|
||||
assert_eq!(age_error_message, "must be >=0, but got -5", "Age error message mismatch: '{}'", age_error_message);
|
||||
|
||||
let invalid_result_missing = validate_json_schema(schema_id, jsonb(invalid_instance_missing));
|
||||
assert_failure_with_json!(invalid_result_missing, "Validation with missing field should fail.");
|
||||
let error_obj_missing = invalid_result_missing.0.get("error").expect("Expected 'error' object for missing field");
|
||||
let causes_missing = error_obj_missing.get("error").and_then(Value::as_array).expect("Expected nested 'error' array (causes)");
|
||||
assert!(!causes_missing.is_empty(), "Expected causes for missing field");
|
||||
let missing_prop_error = causes_missing.iter().find(|e| e["instance_path"].as_str() == Some("")).expect("Missing cause for root instance path");
|
||||
let expected_missing_message = "missing properties 'age'";
|
||||
assert_eq!(missing_prop_error["message"].as_str().expect("Expected string message for missing prop error"), expected_missing_message, "Missing property error message mismatch: '{}'", missing_prop_error["message"].as_str().unwrap_or("null"));
|
||||
|
||||
let non_existent_id = "non_existent_schema";
|
||||
let invalid_schema_result = validate_json_schema(non_existent_id, jsonb(json!({})));
|
||||
assert_failure_with_json!(invalid_schema_result, "Validation with non-existent schema should fail.");
|
||||
let error_obj = invalid_schema_result.0.get("error").expect("Expected top-level 'error' object");
|
||||
assert_eq!(error_obj["message"].as_str().expect("Expected message for cleared schema"), "Schema with id 'non_existent_schema' not found in cache");
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_validate_json_schema_not_cached() {
|
||||
setup_test();
|
||||
let instance = json!({ "foo": "bar" });
|
||||
let result = validate_json_schema("non_existent_schema", jsonb(instance));
|
||||
assert_failure_with_json!(result, "Validation with non-existent schema should fail.");
|
||||
assert_eq!(result.0.get("success"), Some(&Value::Bool(false)), "Expected 'success': false for non-existent schema");
|
||||
let error_obj = result.0.get("error").expect("Expected nested 'error' object for non-existent schema failure");
|
||||
let error_message = error_obj["message"].as_str().expect("Expected message string for non-existent schema");
|
||||
assert!(error_message.contains("non_existent_schema") && error_message.contains("not found"), "Error message mismatch for non-existent schema: '{}'", error_message);
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_cache_invalid_json_schema() {
|
||||
setup_test();
|
||||
let schema_id = "invalid_schema";
|
||||
let invalid_schema = json!({ "type": "invalid" });
|
||||
let cache_result = cache_json_schema(schema_id, jsonb(invalid_schema));
|
||||
assert_failure_with_json!(cache_result, "Caching invalid schema should fail.");
|
||||
|
||||
let error_obj = cache_result.0.get("error").expect("Expected 'error' object for failed cache");
|
||||
let message = error_obj.get("message").and_then(Value::as_str).expect("Message field missing");
|
||||
let expected_message_start = &format!("Schema '{}' failed validation against its metaschema: ", schema_id);
|
||||
assert!(message.starts_with(expected_message_start), "Compile error message start mismatch: {}", message);
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_validate_json_schema_detailed_validation_errors() {
|
||||
setup_test();
|
||||
let schema_id = "detailed_errors";
|
||||
let schema = json!({
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"address": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"street": { "type": "string" },
|
||||
"city": { "type": "string", "maxLength": 10 }
|
||||
},
|
||||
"required": ["street", "city"]
|
||||
}
|
||||
},
|
||||
"required": ["address"]
|
||||
});
|
||||
let _ = cache_json_schema(schema_id, jsonb(schema));
|
||||
|
||||
let invalid_instance = json!({
|
||||
"address": {
|
||||
"street": 123,
|
||||
"city": "Supercalifragilisticexpialidocious"
|
||||
}
|
||||
});
|
||||
|
||||
let result = validate_json_schema(schema_id, jsonb(invalid_instance));
|
||||
assert_failure_with_json!(result, "Validation should fail for detailed errors test.");
|
||||
|
||||
let error_obj = result.0.get("error").expect("Expected top-level 'error' object");
|
||||
assert_eq!(error_obj["message"].as_str(), Some("Validation failed due to nested errors"));
|
||||
|
||||
let causes_array = error_obj.get("error").and_then(Value::as_array).expect("Expected nested error array");
|
||||
assert_eq!(causes_array.len(), 2, "Expected exactly 2 top-level errors (street type, city length)");
|
||||
|
||||
let street_error = causes_array.iter().find(|e| e["instance_path"].as_str() == Some("/address/street")).expect("Missing street error");
|
||||
assert_eq!(street_error["message"].as_str().expect("Expected string message for street error"), "want string, but got number", "Street error message mismatch");
|
||||
|
||||
let city_error = causes_array.iter().find(|e| e["instance_path"].as_str() == Some("/address/city")).expect("Missing city error");
|
||||
let expected_city_error_message = "length must be <=10, but got 34";
|
||||
assert_eq!(city_error["message"].as_str().expect("Expected string message for city error"), expected_city_error_message, "City error message mismatch");
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_validate_json_schema_oneof_validation_errors() {
|
||||
setup_test();
|
||||
let schema_id = "oneof_schema";
|
||||
let schema = json!({
|
||||
"oneOf": [
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"string_prop": { "type": "string", "maxLength": 5 }
|
||||
},
|
||||
"required": ["string_prop"]
|
||||
},
|
||||
{
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"number_prop": { "type": "number", "minimum": 10 }
|
||||
},
|
||||
"required": ["number_prop"]
|
||||
}
|
||||
]
|
||||
});
|
||||
|
||||
let _ = cache_json_schema(schema_id, jsonb(schema));
|
||||
|
||||
let expected_urn = format!("urn:{}#", schema_id);
|
||||
|
||||
// --- Test case 1: String length failure ---
|
||||
let invalid_string_instance = json!({ "string_prop": "toolongstring" });
|
||||
let result_invalid_string = validate_json_schema(schema_id, jsonb(invalid_string_instance));
|
||||
assert_failure_with_json!(result_invalid_string, "Validation with invalid string length should fail");
|
||||
let error_obj_string = result_invalid_string.0.get("error").expect("Expected error object (string case)");
|
||||
|
||||
// Check the top-level error object directly: Input matches Branch 0 type, fails maxLength validation within it.
|
||||
assert_eq!(error_obj_string["schema_path"].as_str(), Some(expected_urn.as_str()), "Schema path mismatch (string case)");
|
||||
assert_eq!(error_obj_string["instance_path"].as_str(), Some(""), "Incorrect instance_path for string fail (string case)");
|
||||
let actual_string_schema_fail_message = error_obj_string["message"].as_str().expect("Expected string message for string schema fail");
|
||||
let expected_string_schema_fail_message = "Validation failed due to nested errors";
|
||||
assert_eq!(actual_string_schema_fail_message, expected_string_schema_fail_message, "String schema fail message mismatch: '{}'", actual_string_schema_fail_message);
|
||||
|
||||
// --- Test case 2: Number type failure ---
|
||||
let invalid_number_instance = json!({ "number_prop": 5 });
|
||||
let result_invalid_number = validate_json_schema(schema_id, jsonb(invalid_number_instance));
|
||||
assert_failure_with_json!(result_invalid_number, "Validation with invalid number should fail");
|
||||
let error_obj_number = result_invalid_number.0.get("error").expect("Expected error object (number case)");
|
||||
|
||||
// Check the top-level error object directly: Input matches Branch 1 type, fails minimum validation within it.
|
||||
assert_eq!(error_obj_number["schema_path"].as_str(), Some(expected_urn.as_str()), "Schema path mismatch (number case)");
|
||||
assert_eq!(error_obj_number["instance_path"].as_str(), Some(""), "Incorrect instance_path for number fail (number case)");
|
||||
let actual_number_schema_fail_message_num = error_obj_number["message"].as_str().expect("Expected string message for number schema fail (number case)");
|
||||
let expected_number_schema_fail_message_num = "Validation failed due to nested errors";
|
||||
assert_eq!(actual_number_schema_fail_message_num, expected_number_schema_fail_message_num, "Number schema fail message mismatch (number case): '{}'", actual_number_schema_fail_message_num);
|
||||
|
||||
// --- Test case 3: Bool type failure ---
|
||||
let invalid_bool_instance = json!({ "bool_prop": 123 });
|
||||
let result_invalid_bool = validate_json_schema(schema_id, jsonb(invalid_bool_instance));
|
||||
assert_failure_with_json!(result_invalid_bool, "Validation with invalid bool should fail");
|
||||
let error_obj_bool = result_invalid_bool.0.get("error").expect("Expected error object (bool case)");
|
||||
|
||||
// Boon reports the failure for the first branch checked (Branch 0).
|
||||
assert_eq!(error_obj_bool["schema_path"].as_str(), Some(expected_urn.as_str()), "Schema path mismatch (bool case)");
|
||||
// Instance path is empty because the failure is at the root of the object being checked against Branch 0
|
||||
assert_eq!(error_obj_bool["instance_path"].as_str(), Some(""), "Incorrect instance_path for bool fail");
|
||||
let actual_bool_fail_message_0 = error_obj_bool["message"].as_str().expect("Expected string message for branch 0 type fail");
|
||||
let expected_bool_fail_message_0 = "Validation failed due to nested errors";
|
||||
assert_eq!(actual_bool_fail_message_0, expected_bool_fail_message_0, "Branch 0 type fail message mismatch: '{}'", actual_bool_fail_message_0);
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_clear_json_schemas() {
|
||||
setup_test();
|
||||
let schema_id = "schema_to_clear";
|
||||
let schema = json!({ "type": "string" });
|
||||
cache_json_schema(schema_id, jsonb(schema.clone()));
|
||||
|
||||
let show_result1 = show_json_schemas();
|
||||
assert!(show_result1.contains(&schema_id.to_string()));
|
||||
|
||||
clear_json_schemas();
|
||||
|
||||
let show_result2 = show_json_schemas();
|
||||
assert!(show_result2.is_empty());
|
||||
|
||||
let instance = json!("test");
|
||||
let validate_result = validate_json_schema(schema_id, jsonb(instance));
|
||||
assert_failure_with_json!(validate_result, "Validation should fail after clearing schemas.");
|
||||
|
||||
let error_obj = validate_result.0.get("error").expect("Expected top-level 'error' object");
|
||||
assert_eq!(error_obj["message"].as_str().expect("Expected message for cleared schema"), "Schema with id 'schema_to_clear' not found in cache");
|
||||
}
|
||||
|
||||
#[pg_test]
|
||||
fn test_show_json_schemas() {
|
||||
setup_test();
|
||||
let schema_id1 = "schema1";
|
||||
let schema_id2 = "schema2";
|
||||
let schema = json!({ "type": "boolean" });
|
||||
|
||||
cache_json_schema(schema_id1, jsonb(schema.clone()));
|
||||
cache_json_schema(schema_id2, jsonb(schema.clone()));
|
||||
|
||||
let result = show_json_schemas();
|
||||
assert!(result.contains(&schema_id1.to_string()));
|
||||
assert!(result.contains(&schema_id2.to_string()));
|
||||
}
|
||||
Reference in New Issue
Block a user