fix: findRoots should get the defined primary key column (#6982)

* test: added failing test for github issue #6948

* fix: findRoots should get the actual primary key column, fixes #6948

src/repository/TreeRepository.ts

@@ -33,7 +33,7 @@ export class TreeRepository<Entity> extends Repository<Entity> {
         const escapeAlias = (alias: string) => this.manager.connection.driver.escape(alias);
         const escapeColumn = (column: string) => this.manager.connection.driver.escape(column);
         const parentPropertyName = this.manager.connection.namingStrategy.joinColumnName(
-          this.metadata.treeParentRelation!.propertyName, "id"
+          this.metadata.treeParentRelation!.propertyName, this.metadata.primaryColumns[0].propertyName
         );
 
         return this.createQueryBuilder("treeEntity")

test/github-issues/6948/entity/Category.ts

@@ -0,0 +1,17 @@
+import { Entity, PrimaryGeneratedColumn, Column, Tree, TreeParent, TreeChildren } from "../../../../src";
+
+@Entity()
+@Tree("materialized-path")
+export class Category {
+    @PrimaryGeneratedColumn()
+    cat_id: number;
+
+    @Column()
+    cat_name: string;
+
+    @TreeParent()
+    cat_parent: Category;
+
+    @TreeChildren({ cascade: true })
+    cat_children: Category[];
+}

test/github-issues/6948/issue-6948.ts

@@ -0,0 +1,38 @@
+import "reflect-metadata";
+import { Category } from "./entity/Category";
+import { Connection } from "../../../src/connection/Connection";
+import {
+    closeTestingConnections,
+    createTestingConnections,
+    reloadTestingDatabases,
+} from "../../../test/utils/test-utils";
+
+describe("github issues > #6948 TreeRepository's findRoots query incorrectly when using a custom primary key", () => {
+    let connections: Connection[];
+    before(
+        async () =>
+            (connections = await createTestingConnections({
+                entities: [Category],
+            }))
+    );
+    beforeEach(() => reloadTestingDatabases(connections));
+    after(() => closeTestingConnections(connections));
+
+    it("entity parent column should work with custom primary column names ", () =>
+        Promise.all(
+            connections.map(async (connection) => {
+                const categoryRepository = connection.getTreeRepository(
+                    Category
+                );
+                await categoryRepository.save(
+                    categoryRepository.create({
+                        cat_name: "Root node",
+                    })
+                );
+                const rootNodes = await categoryRepository.findRoots();
+                rootNodes[0].should.deep.include({
+                    cat_name: "Root node",
+                });
+            })
+        ));
+});