@kiyeonjeon21/datacontext 0.3.2 → 0.4.0

package/tests/knowledge-store.test.ts

@@ -161,5 +161,135 @@ describe('KnowledgeStore', () => {
       expect(desc?.description).toBe('Persisted description');
     });
   });
+
+  describe('Table Relationships (Graph)', () => {
+    it('should add a relationship from FK', async () => {
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+
+      const relationships = store.getRelationships();
+      expect(relationships.length).toBe(1);
+      expect(relationships[0].from.table).toBe('orders');
+      expect(relationships[0].to.table).toBe('users');
+      expect(relationships[0].joinCondition).toBe('orders.user_id = users.id');
+      expect(relationships[0].relationshipType).toBe('foreign_key');
+    });
+
+    it('should get related tables (1-hop)', async () => {
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'product_id',
+        toTable: 'products',
+        toColumn: 'id',
+      });
+
+      const relatedToOrders = store.getRelatedTables('orders', 'public');
+      expect(relatedToOrders.length).toBe(2);
+
+      const relatedToUsers = store.getRelatedTables('users', 'public');
+      expect(relatedToUsers.length).toBe(1);
+    });
+
+    it('should find join path between tables', async () => {
+      // Setup: order_items → orders → users
+      await store.addRelationshipFromFK({
+        fromTable: 'order_items',
+        fromColumn: 'order_id',
+        toTable: 'orders',
+        toColumn: 'id',
+      });
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+
+      const path = store.findJoinPath('order_items', 'users', 'public');
+      expect(path.length).toBe(2);
+      expect(path[0].from.table).toBe('order_items');
+      expect(path[0].to.table).toBe('orders');
+      expect(path[1].from.table).toBe('orders');
+      expect(path[1].to.table).toBe('users');
+    });
+
+    it('should return empty path if no connection', async () => {
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+
+      const path = store.findJoinPath('orders', 'products', 'public');
+      expect(path.length).toBe(0);
+    });
+
+    it('should get graph summary', async () => {
+      await store.setTableDescription('users', 'Users table');
+      await store.setTableDescription('orders', 'Orders table');
+      await store.setTableDescription('products', 'Products table');
+      
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+
+      const summary = store.getGraphSummary();
+      expect(summary.nodeCount).toBe(3);
+      expect(summary.edgeCount).toBe(1);
+      expect(summary.tablesWithRelationships).toBe(2);
+      expect(summary.isolatedTables).toBe(1);
+      expect(summary.relationshipsByType.foreign_key).toBe(1);
+    });
+
+    it('should build context with relationships', async () => {
+      await store.setTableDescription('orders', 'Orders table');
+      await store.setTableDescription('users', 'Users table');
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+
+      const context = store.buildContextWithRelationships(['orders', 'users']);
+      
+      expect(context).toContain('Table Relationships');
+      expect(context).toContain('orders.user_id = users.id');
+      expect(context).toContain('many-to-one');
+    });
+
+    it('should not duplicate relationships', async () => {
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+      // Add same relationship again
+      await store.addRelationshipFromFK({
+        fromTable: 'orders',
+        fromColumn: 'user_id',
+        toTable: 'users',
+        toColumn: 'id',
+      });
+
+      const relationships = store.getRelationships();
+      expect(relationships.length).toBe(1);
+    });
+  });
 });