Function setupQueryExecutionGraph

packages/graph/src/test/queryExecution.test.ts:38–88

()

}

function setupQueryExecutionGraph(): Graph<GraphSchema> {
  // Create a simple schema
  const schema = {
    vertices: {
      User: {
        properties: {
          name: { type: makeType<string>("") },
          age: { type: makeType<number>(0) },
          active: { type: makeType<boolean>(false) },
        },
      },
    },
    edges: {
      follows: {
        properties: {},
      },
      likes: {
        properties: {},
      },
    },
  } as const satisfies GraphSchema;

  const graph = new Graph({ schema, storage: new InMemoryGraphStorage() });

  // Add some test data
  const alice = graph.addVertex("User", {
    name: "Alice",
    age: 25,
    active: true,
  });
  const bob = graph.addVertex("User", { name: "Bob", age: 30, active: true });
  const charlie = graph.addVertex("User", {
    name: "Charlie",
    age: 20,
    active: false,
  });
  const david = graph.addVertex("User", {
    name: "David",
    age: 35,
    active: true,
  });

  // Create some relationships
  graph.addEdge(alice.id, "follows", bob.id, {});
  graph.addEdge(alice.id, "follows", charlie.id, {});
  graph.addEdge(bob.id, "follows", david.id, {});
  graph.addEdge(charlie.id, "follows", alice.id, {});
  graph.addEdge(alice.id, "likes", bob.id, {});

  return graph;
}

test("Query Execution End-to-End - Simple vertex queries - should fetch all users", () => {
  const graph = setupQueryExecutionGraph();