use crate::types::describe::{FieldDescribe, FieldType, SObjectDescribe};
use std::collections::HashMap;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct FieldChange {
pub name: String,
pub old_type: FieldType,
pub new_type: FieldType,
}
#[derive(Debug, Clone, PartialEq, Default)]
pub struct SchemaDiffResult {
pub added: Vec<FieldDescribe>,
pub removed: Vec<FieldDescribe>,
pub changed: Vec<FieldChange>,
}
impl SchemaDiffResult {
#[must_use]
pub fn is_empty(&self) -> bool {
self.added.is_empty() && self.removed.is_empty() && self.changed.is_empty()
}
}
#[must_use]
pub fn compare_schemas(
old_schema: &SObjectDescribe,
new_schema: &SObjectDescribe,
) -> SchemaDiffResult {
let mut result = SchemaDiffResult::default();
let mut old_fields: HashMap<&str, &FieldDescribe> =
HashMap::with_capacity(old_schema.fields.len());
for field in &old_schema.fields {
old_fields.insert(field.name.as_str(), field);
}
for new_field in &new_schema.fields {
if let Some(old_field) = old_fields.remove(new_field.name.as_str()) {
if old_field.type_ != new_field.type_ {
result.changed.push(FieldChange {
name: new_field.name.clone(),
old_type: old_field.type_.clone(),
new_type: new_field.type_.clone(),
});
}
} else {
result.added.push(new_field.clone());
}
}
result.removed.extend(old_fields.into_values().cloned());
result
.added
.sort_by(|a, b| crate::schema::cmp_field_names(&a.name, &b.name));
result
.removed
.sort_by(|a, b| crate::schema::cmp_field_names(&a.name, &b.name));
result
.changed
.sort_by(|a, b| crate::schema::cmp_field_names(&a.name, &b.name));
result
}
#[cfg(test)]
mod tests {
use super::*;
use crate::test_support::Must;
use serde_json::json;
fn create_mock_describe(fields_json: &serde_json::Value) -> SObjectDescribe {
let describe_json = json!({
"name": "Account",
"label": "Account",
"custom": false,
"queryable": true,
"activateable": false, "createable": true, "customSetting": false, "deletable": true,
"deprecatedAndHidden": false, "feedEnabled": true, "hasSubtypes": false,
"isSubtype": false, "keyPrefix": "001", "labelPlural": "Accounts", "layoutable": true,
"mergeable": true, "mruEnabled": true, "replicateable": true, "retrieveable": true,
"searchable": true, "triggerable": true, "undeletable": true, "updateable": true,
"urls": {}, "childRelationships": [], "recordTypeInfos": [],
"fields": fields_json.clone()
});
serde_json::from_value(describe_json).must()
}
fn mock_field(name: &str, field_type: &str) -> serde_json::Value {
json!({
"name": name,
"type": field_type,
"label": format!("{} Label", name),
"referenceTo": [],
"aggregatable": true, "autoNumber": false, "byteLength": 18, "calculated": false,
"cascadeDelete": false, "caseSensitive": false, "createable": false, "custom": false,
"defaultedOnCreate": true, "dependentPicklist": false, "deprecatedAndHidden": false,
"digits": 0, "displayLocationInDecimal": false, "encrypted": false, "externalId": false,
"filterable": true, "groupable": true, "highScaleNumber": false, "htmlFormatted": false,
"idLookup": true, "length": 18, "nameField": false, "namePointing": false, "nillable": false,
"permissionable": false, "polymorphicForeignKey": false, "precision": 0, "queryByDistance": false,
"restrictedDelete": false, "restrictedPicklist": false, "scale": 0, "soapType": "tns:ID",
"sortable": true, "unique": false, "updateable": false, "writeRequiresMasterRead": false
})
}
#[test]
fn test_schema_diff_no_changes() {
let old_schema = create_mock_describe(&json!([
mock_field("Id", "id"),
mock_field("Name", "string")
]));
let new_schema = create_mock_describe(&json!([
mock_field("Id", "id"),
mock_field("Name", "string")
]));
let diff = compare_schemas(&old_schema, &new_schema);
assert!(diff.is_empty());
assert_eq!(diff.added.len(), 0);
assert_eq!(diff.removed.len(), 0);
assert_eq!(diff.changed.len(), 0);
}
#[test]
fn test_schema_diff_added_fields() {
let old_schema = create_mock_describe(&json!([mock_field("Id", "id")]));
let new_schema = create_mock_describe(&json!([
mock_field("Id", "id"),
mock_field("Name", "string"),
mock_field("Website", "url")
]));
let diff = compare_schemas(&old_schema, &new_schema);
assert!(!diff.is_empty());
assert_eq!(diff.added.len(), 2);
assert_eq!(diff.removed.len(), 0);
assert_eq!(diff.changed.len(), 0);
assert_eq!(diff.added[0].name, "Name");
assert_eq!(diff.added[1].name, "Website");
}
#[test]
fn test_schema_diff_removed_fields() {
let old_schema = create_mock_describe(&json!([
mock_field("Id", "id"),
mock_field("Name", "string"),
mock_field("Website", "url")
]));
let new_schema = create_mock_describe(&json!([mock_field("Id", "id")]));
let diff = compare_schemas(&old_schema, &new_schema);
assert!(!diff.is_empty());
assert_eq!(diff.added.len(), 0);
assert_eq!(diff.removed.len(), 2);
assert_eq!(diff.changed.len(), 0);
assert_eq!(diff.removed[0].name, "Name");
assert_eq!(diff.removed[1].name, "Website");
}
#[test]
fn test_schema_diff_changed_fields() {
let old_schema =
create_mock_describe(&json!([mock_field("Id", "id"), mock_field("Age", "int")]));
let new_schema = create_mock_describe(&json!([
mock_field("Id", "id"),
mock_field("Age", "double")
]));
let diff = compare_schemas(&old_schema, &new_schema);
assert!(!diff.is_empty());
assert_eq!(diff.added.len(), 0);
assert_eq!(diff.removed.len(), 0);
assert_eq!(diff.changed.len(), 1);
assert_eq!(diff.changed[0].name, "Age");
assert_eq!(diff.changed[0].old_type, FieldType::Int);
assert_eq!(diff.changed[0].new_type, FieldType::Double);
}
}