test: update linked schema test

Author: omer-topal

internal/schema/linked_schema_test.go

@@ -1785,28 +1785,32 @@ var _ = Describe("linked schema", func() {
 				{
 					Kind: PathChainLinkedEntrance,
 					TargetEntrance: &base.Entrance{
-						Type:  "department",
-						Value: "dept_name",
+						Type:  "company",
+						Value: "company_id",
 					},
 					TupleSetRelation: "",
 					PathChain: []*base.RelationReference{
 						{
 							Type:     "employee",
 							Relation: "department",
 						},
+						{
+							Type:     "department",
+							Relation: "company",
+						},
 					},
 				},
 				{
 					Kind: PathChainLinkedEntrance,
 					TargetEntrance: &base.Entrance{
-						Type:  "company",
-						Value: "company_id",
+						Type:  "department",
+						Value: "dept_name",
 					},
 					TupleSetRelation: "",
 					PathChain: []*base.RelationReference{
 						{
-							Type:     "department",
-							Relation: "company",
+							Type:     "employee",
+							Relation: "department",
 						},
 					},
 				},
@@ -1821,6 +1825,65 @@ var _ = Describe("linked schema", func() {
 				},
 			}))
 		})
+
+		It("Case 32: Nested PathChain preserves tuple-set relation", func() {
+			sch, err := parser.NewParser(`
+			entity user {}
+
+			entity org {
+				attribute is_public boolean
+				permission view = is_public
+			}
+
+			entity folder {
+				relation parent @org
+				permission view = parent.view
+			}
+
+			entity resource {
+				relation parent @folder
+				relation alt @folder
+				permission view = parent.view
+			}
+			`).Parse()
+
+			Expect(err).ShouldNot(HaveOccurred())
+
+			c := compiler.NewCompiler(true, sch)
+			a, _, _ := c.Compile()
+
+			g := NewLinkedGraph(NewSchemaFromEntityAndRuleDefinitions(a, nil))
+
+			ent, err := g.LinkedEntrances(&base.Entrance{
+				Type:  "resource",
+				Value: "view",
+			}, &base.Entrance{
+				Type:  "user",
+				Value: "",
+			})
+
+			Expect(err).ShouldNot(HaveOccurred())
+			Expect(ent).Should(Equal([]*LinkedEntrance{
+				{
+					Kind: PathChainLinkedEntrance,
+					TargetEntrance: &base.Entrance{
+						Type:  "org",
+						Value: "is_public",
+					},
+					TupleSetRelation: "",
+					PathChain: []*base.RelationReference{
+						{
+							Type:     "resource",
+							Relation: "parent",
+						},
+						{
+							Type:     "folder",
+							Relation: "parent",
+						},
+					},
+				},
+			}))
+		})
 	})
 
 	Context("BuildRelationPathChain", func() {