Function setupMultiLabelGraph

packages/graph/src/test/queryExecution.test.ts:256–296

()

// Tests for labels() function
function setupMultiLabelGraph(): Graph<GraphSchema> {
  const schema = {
    vertices: {
      User: {
        properties: {
          name: { type: makeType<string>("") },
        },
      },
      Post: {
        properties: {
          title: { type: makeType<string>("") },
        },
      },
      Comment: {
        properties: {
          text: { type: makeType<string>("") },
        },
      },
    },
    edges: {
      wrote: {
        properties: {},
      },
    },
  } as const satisfies GraphSchema;

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

  // Add vertices with different labels
  const alice = graph.addVertex("User", { name: "Alice" });
  const bob = graph.addVertex("User", { name: "Bob" });
  const post1 = graph.addVertex("Post", { title: "Hello World" });
  const post2 = graph.addVertex("Post", { title: "Second Post" });
  const comment1 = graph.addVertex("Comment", { text: "Nice post!" });

  graph.addEdge(alice.id, "wrote", post1.id, {});
  graph.addEdge(bob.id, "wrote", post2.id, {});
  graph.addEdge(alice.id, "wrote", comment1.id, {});

  return graph;
}

test("Query Execution End-to-End - labels() function - should return labels for a single vertex", () => {
  const graph = setupMultiLabelGraph();