reifydb-catalog 0.4.13

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

use reifydb_core::{
	interface::catalog::flow::{FlowEdge, FlowEdgeId},
	internal,
};
use reifydb_transaction::transaction::Transaction;
use reifydb_type::error::Error;

use crate::{CatalogStore, Result};

impl CatalogStore {
	pub(crate) fn get_flow_edge(rx: &mut Transaction<'_>, edge_id: FlowEdgeId) -> Result<FlowEdge> {
		CatalogStore::find_flow_edge(rx, edge_id)?.ok_or_else(|| {
			Error(Box::new(internal!(
				"Flow edge with ID {:?} not found in catalog. This indicates a critical catalog inconsistency.",
				edge_id
			)))
		})
	}
}

#[cfg(test)]
pub mod tests {
	use reifydb_core::interface::catalog::flow::FlowEdgeId;
	use reifydb_engine::test_harness::create_test_admin_transaction;
	use reifydb_transaction::transaction::Transaction;

	use crate::{
		CatalogStore,
		test_utils::{create_flow_edge, create_flow_node, create_namespace, ensure_test_flow},
	};

	#[test]
	fn test_get_flow_edge_ok() {
		let mut txn = create_test_admin_transaction();
		let _namespace = create_namespace(&mut txn, "test_namespace");
		let flow = ensure_test_flow(&mut txn);

		let node1 = create_flow_node(&mut txn, flow.id, 1, &[0x01]);
		let node2 = create_flow_node(&mut txn, flow.id, 4, &[0x02]);
		let edge = create_flow_edge(&mut txn, flow.id, node1.id, node2.id);

		let result = CatalogStore::get_flow_edge(&mut Transaction::Admin(&mut txn), edge.id).unwrap();
		assert_eq!(result.id, edge.id);
		assert_eq!(result.flow, flow.id);
		assert_eq!(result.source, node1.id);
		assert_eq!(result.target, node2.id);
	}

	#[test]
	fn test_get_flow_edge_not_found() {
		let mut txn = create_test_admin_transaction();

		let err = CatalogStore::get_flow_edge(&mut Transaction::Admin(&mut txn), FlowEdgeId(999)).unwrap_err();
		assert_eq!(err.code, "INTERNAL_ERROR");
		assert!(err.message.contains("FlowEdgeId(999)"));
		assert!(err.message.contains("not found in catalog"));
	}
}