Add unit test for graph rag interfaces

test/agentchat/contrib/graph_rag/test_graph_rag_basic.py

@@ -0,0 +1,17 @@
+from unittest.mock import Mock
+
+from autogen.agentchat.contrib.graph_rag.graph_query_engine import GraphQueryEngine
+from autogen.agentchat.contrib.graph_rag.graph_rag_capability import GraphRagCapability
+from autogen.agentchat.conversable_agent import ConversableAgent
+
+
+def test_dry_run():
+    """Dry run for basic graph rag objects."""
+    mock_graph_query_engine = Mock(spec=GraphQueryEngine)
+
+    graph_rag_agent = ConversableAgent(
+        name="graph_rag_agent",
+        max_consecutive_auto_reply=3,
+    )
+    graph_rag_capability = GraphRagCapability(mock_graph_query_engine)
+    graph_rag_capability.add_to_agent(graph_rag_agent)